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 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 read_worker_card_input(self):
worker_card_entity = WorkerCardScheme()
worker_card_entity.id = input("enter id: ")
worker_card_entity.worker_id = input("enter worker_id: ")
while not CheckInput.check_worker_existence(
worker_card_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()
worker_card_entity.worker_id = worker_entity.id
db.insert_worker(worker_entity)
break
if case('n'):
print("can't insert worker card ")
return
else:
print("____________________")
print("no such option, try again!")
worker_card_entity.is_active = input("enter is active: ")
while worker_card_entity.is_active != "True" and worker_card_entity.is_active != "False":
worker_card_entity.is_active = input("Wrong input!\nTry again: ")
return worker_card_entity
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)
