def test_update_event(self):
c = Controller(self.test_db)
# Create a client
cl_id = c.create_client('username', name='Foo Bar', age='23', events=[])
event_id = c.create_client_req(event_type='Mohawk fans', from_date='09-12-2042', client_id=cl_id)
updated_event = {'id': event_id, 'from_date': '08-12-2015'}
c.update_event(**updated_event)
event_data = c.get_event('id', event_id)[0]
self.assertEqual('Mohawk fans', event_data['event_type'])
self.assertEqual(cl_id, event_data['client_id'])
self.assertEqual('08-12-2015', event_data['from_date'])
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)