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
