def read_journal_update_input(self): journal_entity = JournalScheme() journal_entity.id = input("enter id: ") if CheckInput.check_journal_existence(journal_entity.id): print("exit timestamp:") journal_entity.exit = self.get_date() print("entrance timestamp:") journal_entity.entrance = self.get_date() journal_entity.worker_id = input("worker id: ") while not CheckInput.check_worker_existence( journal_entity.worker_id): input_str = input( "There is no such worker!\nWanna add?\n(y/n): ") for case in switch(input_str): if case('y'): worker_entity = self.read_worker_input() journal_entity.worker_id = worker_entity.id db.insert_worker(worker_entity) break if case('n'): print("can't insert journal") return else: print("____________________") print("no such option, try again!") return journal_entity else: print("There is no such journal in db!") return None
def delete_entity(db_name): if db_name == "worker": worker_id = user_input.get_id() if not CheckInput.check_worker_existence(worker_id): print("There is no such worker in db!\n") else: db.delete_worker(worker_id) elif db_name == "journal": journal_id = user_input.get_id() if not CheckInput.check_journal_existence(journal_id): print("There is no such journal in db!\n") else: db.delete_journal(journal_id) elif db_name == "department": department_id = user_input.get_id() if not CheckInput.check_department_existence(department_id): print("There is no such department in db!\n") else: db.delete_department(department_id) elif db_name == "worker_card": worker_card_id = user_input.get_id() if not CheckInput.check_worker_card_existence(worker_card_id): print("There is no such worker card in db!\n") else: db.delete_worker_card(worker_card_id)