def test_create_financial_req(self):
c = Controller(self.test_db)
req_id = c.create_financial_req(event_id='ev1234', req_amount=321, reason='reason1', req_dpt='service')
req_data = c.get_financial_req('id', req_id)[0]
self.assertEqual('ev1234', req_data['event_id'])
self.assertEqual(321, req_data['req_amount'])
self.assertEqual('reason1', req_data['reason'])
self.assertEqual('service', req_data['req_dpt'])
def test_update_financial_req(self):
c = Controller(self.test_db)
req_id = c.create_financial_req(event_id='ev1234', req_amount=321, reason='reason1', req_dpt='service')
updated_fin_req = {'id': req_id, 'event_id': 'ev321', 'req_amount': 3213, 'reason': 'reason2',
'req_dpt': 'production'}
c.update_financial_req(**updated_fin_req)
req_data = c.get_financial_req('id', req_id)[0]
self.assertEqual('ev321', req_data['event_id'])
self.assertEqual(3213, req_data['req_amount'])
self.assertEqual('reason2', req_data['reason'])
self.assertEqual('production', req_data['req_dpt'])
class Mediator:
base = None
login_form = None
roles = {'0': 'team_member',
'1': 'customer_service',
'2': 'senior_customer_service_officer',
'3': 'human_resources',
'4': 'administration',
'5': 'financial',
'6': 'production',
'7': 'service',
'8': 'vice_president'}
def __init__(self, db_name='db.json'):
self.c = Controller(db_name)
def check_credentials(self, username, password):
return self.c.login(username, password)
def login(self, username):
# Get that employee id, clients cant login...
user_id = self.c.get_user_id(username)
empl_data = self.get_employee('id', user_id, False)[0]
name = empl_data['name']
pos = empl_data['position']
self.base = Base(name, self.roles[pos].title().replace('_', ' '), username)
self.m = ManagerTabs(self.roles[pos], user_id, name)
self.base.set_central_widget(self.m)
def logout(self):
self.login_form.show()
def create_client(self, name, age, address, mail, phone):
return self.c.create_client(name, name=name, age=age, address=address, mail=mail, phone=phone, events=[])
def create_client_req(self, client_id, event_type, description, from_date, to_date, exp_no,
planned_budget, decorations, filming, poster,
food, music, computer, other, approved):
return self.c.create_client_req(client_id=client_id, event_type=event_type, description=description,
from_date=from_date,
to_date=to_date, exp_no=exp_no, planned_budget=planned_budget,
decorations=decorations,
filming=filming, poster=poster, food=food, music=music, computer=computer,
other=other, approved=approved)
def create_employee(self, name, age, address, mail, position):
return self.c.create_employee(name, name=name, age=age, address=address, mail=mail,
position=position)
def create_recruitment_req(self, type, years_exp, title, description, dpt_req):
return self.c.create_recruitment_req(type=type, years_exp=years_exp, title=title, description=description,
req_dpt=dpt_req)
def update_recruitment_req(self, recruitment_id, type, years_exp, title, description, dpt_req):
return self.c.update_recruitment_req(id=recruitment_id, type=type, years_exp=years_exp, title=title,
description=description,
req_dpt=dpt_req)
def create_task(self, sub_team, event_id, description, staff_id, priority):
return self.c.create_task(sub_team=sub_team, event_id=event_id, description=description, staff_id=staff_id,
priority=priority)
def update_task(self, task_id, sub_team, event_id, description, staff_id, priority):
self.c.update_task(id=task_id, sub_team=sub_team, event_id=event_id, description=description,
staff_id=staff_id, priority=priority)
def create_financial_req(self, event_id, req_amount, reason, req_dpt):
return self.c.create_financial_req(event_id=event_id, req_amount=req_amount, reason=reason, req_dpt=req_dpt)
def update_financial_req(self, fin_req_id, event_id, req_amount, reason, req_dpt):
self.c.update_financial_req(id=fin_req_id, event_id=event_id, req_amount=req_amount, reason=reason,
req_dpt=req_dpt)
def get_financial_req(self, col_name='', criteria='', all_data=True):
return self._get_data('financial_req', col_name, criteria, all_data)
def get_recruitment_req(self, col_name='', criteria='', all_data=True):
return self._get_data('recruitment_req', col_name, criteria, all_data)
def get_client(self, col_name='', criteria='', all_data=True):
return self._get_data('client', col_name, criteria, all_data)
def get_employee(self, col_name='', criteria='', all_data=True):
return self._get_data('employee', col_name, criteria, all_data)
def get_task(self, col_name='', criteria='', all_data=True):
return self._get_data('task', col_name, criteria, all_data)
def get_event(self, col_name='', criteria='', all_data=True):
return self._get_data('event', col_name, criteria, all_data)
def update_event(self, ev_id, client_id, event_type, description, from_date, to_date, exp_no,
planned_budget, decorations, filming, poster,
food, music, computer, other, approved):
return self.c.update_event(id=ev_id, client_id=client_id, event_type=event_type, description=description,
from_date=from_date,
to_date=to_date, exp_no=exp_no, planned_budget=planned_budget,
decorations=decorations,
filming=filming, poster=poster, food=food, music=music, computer=computer,
other=other, approved=approved)
def _get_data(self, table, col_name, criteria, all_data):
return getattr(self.c, "get_%s" % table)(col_name, criteria, all_data)