Пример #1
0
    def firm_loop(self):
        gui = GUI()
        gui.print_firm()

        select_char = "s"
        while select_char != "q":
            select_char = input("\nSelect action:\n"
                                "\t1. Display of containers\n"
                                "\t2. Edit containers\n"
                                "\t3. Add a new container\n"
                                "\t4. Display orders\n"
                                "\t5. Exit\n").upper()
            if select_char not in "12345" or len(select_char) != 1:
                print("No such action")
                continue

            if select_char == '1':
                for i in range(len(self.container_list)):
                    self.container_list[i].print_info(i)

            elif select_char == '2':
                for i in range(len(self.container_list)):
                    self.container_list[i].print_info(i)
                container_id = int(input("Enter the container id: "))
                self.container_list[container_id].edit_container()

            elif select_char == '3':
                container = Container()
                container.price = int(input("Enter price of the container: "))
                container.size = int(input("Enter container size: "))
                self.container_list.append(container)

            if select_char == '4':
                for i in range(len(self.order)):
                    self.order[i].print_order(i)

                order_input = int(
                    input("Action:\n"
                          "1. Accept order\n"
                          "2. Delete order\n"
                          "3. Return\n"))

                if order_input == 1:
                    index = int(input("Enter ID orders to accept"))
                    if 0 <= index <= len(self.order):
                        self.order[index].paid = True

                if order_input == 2:
                    index = int(input("Enter ID orders to deleted"))
                    if 0 <= index <= len(self.order):
                        self.order.remove(self.order[index])

            if select_char == '5':
                gui = GUI()
                gui.clear(0)
                return