cli.py

import pickle
import click
from click_shell import shell


def run():
    global our_dorm
    with open('saved_step.pickle', 'rb') as f:
        our_dorm = pickle.load(f)
    f.close()
    with open('saved_step_2.pickle', 'rb') as f:
        our_dorm.set_room_list(pickle.load(f))
    f.close()
    our_dorm.find_all_students()
    our_dorm.next_month()
    our_dorm.define_is_kitchen_okey()
    print(our_dorm.roach())
    our_dorm.noise_in_dorm()
    print(our_dorm.visit_every_room())
    our_dorm.change_duty()
    our_dorm.save_to_file()
    our_dorm.print_field()
    app()

@shell(prompt='> ', intro='Запуск CLI...')
def app():
    pass

    @app.command(help='Травить тараканов в комнате №[1..15]')
    @click.argument('number')
    def kill_roach(number):
        try:
            room = our_dorm.get_room_by_number(int(number))
            print(room.kill_roach())
            our_dorm.save_to_file()
        except IndexError:
            print('В общежитии 15 комнат')

    @app.command(help='Вывести информацию о комнате №[1..15]')
    @click.argument('number')
    def print_info_room(number):
        try:
            room = our_dorm.get_room_by_number(int(number))
            print(room.info())
        except IndexError:
            print('В общежитии 15 комнат')

    @app.command(help='Заселить студента в комнату №[1..15]')
    @click.argument('number')
    def check_in(number):
        try:
            room = our_dorm.get_room_by_number(int(number))
            print(our_dorm.check_in(room))
            our_dorm.save_to_file()
        except IndexError:
            print('В общежитии 15 комнат')

    @app.command()
    def print_info_dorm():
        print(our_dorm.info())

    @app.command()
    def print_map():
        our_dorm.print_field()