def read_worker_update_input(self): workerEntity = WorkerScheme() workerEntity.name = input("enter worker name: ") workerEntity.id = input("enter worker id: ") if CheckInput.check_worker_existence(workerEntity.id): workerEntity.is_still_working = input( "enter worker is still working (True/False): ") while workerEntity.is_still_working != "True" and workerEntity.is_still_working != "False": workerEntity.is_still_working = input( "Wrong input!\nTry again: ") workerEntity.department_id = input("enter department id: ") while not CheckInput.check_department_existence( workerEntity.department_id): input_str = input( "There is no such department!\nWanna add?\n(y/n): ") for case in switch(input_str): if case('y'): department_entity = self.read_department_input() workerEntity.department_id = department_entity.id db.insert_department(department_entity) break if case('n'): print("can't insert worker") return else: print("____________________") print("no such option, try again!") return workerEntity else: print("There is no such worker 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)
def read_department_update_input(self): department_entity = DepartmentScheme() department_entity.id = input("enter id: ") if CheckInput.check_department_existence(department_entity.id): department_entity.production = input("enter production: ") return department_entity else: print("There is no such department in db!") return None