def create_an_assignation(
data,
workshift_db=None,
day_off_assignation_db=None
):
assignation_data = data.get('assignation')
starting_date_str = assignation_data.get('starting_date')
assignation_data['starting_date'] = DateUtil.str_to_date(starting_date_str)
ending_date_str = assignation_data.get('ending_date')
assignation_data['ending_date'] = DateUtil.str_to_date(ending_date_str)
assignation = DumbAssignation(**assignation_data)
if workshift_db:
workshift_proxy = workshift_db.get_by_id(assignation.workshift_id)
assignation.workshift_proxy = workshift_proxy
if day_off_assignation_db:
day_off_assignations = day_off_assignation_db.get_by_person_id(
assignation.person_id
)
assignation.day_off_assignations = day_off_assignations
person_data = data.get('person', {})
assignation.person = DumbPerson(**person_data)
return ProxyFactory.create_assignation_proxy(assignation)
def create_proxy_day_off_assignation(day_off_assignations_data):
dumb_day_off_assignations = []
for day_off_assignation_data in day_off_assignations_data:
starting_date_str = day_off_assignation_data['starting_date']
day_off_assignation_data['starting_date'] = DateUtil.str_to_date(
starting_date_str
)
ending_date_str = day_off_assignation_data['ending_date']
day_off_assignation_data['ending_date'] = DateUtil.str_to_date(ending_date_str)
starting_time_str = day_off_assignation_data['starting_time']
day_off_assignation_data['starting_time'] = DateUtil.str_to_time(
starting_time_str
)
ending_time_str = day_off_assignation_data['ending_time']
day_off_assignation_data['ending_time'] = DateUtil.str_to_time(ending_time_str)
dumb_day_off_assignation = DumbDayOffAssignation(
**day_off_assignation_data
)
dumb_day_off_assignations.append(dumb_day_off_assignation)
return ProxyFactory.create_multiple_day_off_assignation_proxies(
dumb_day_off_assignations
)
def create_proxy_workshifts(workshifts_data):
dumb_workshifts = []
for workshift_data in workshifts_data:
days_data = workshift_data.pop('days', [])
days = []
for day_data in days_data:
starting_time_str = day_data.get('starting_time')
if starting_time_str:
day_data['starting_time'] = DateUtil.str_to_time(starting_time_str)
ending_time_str = day_data.get('ending_time')
if ending_time_str:
day_data['ending_time'] = DateUtil.str_to_time(ending_time_str)
date_str = day_data.get('date', None)
if date_str:
day_data['date'] = DateUtil.str_to_date(date_str)
day = DumbDay(**day_data)
days.append(day)
dumb_workshift = DumbWorkShift(**workshift_data)
dumb_workshift.days = days
dumb_workshifts.append(dumb_workshift)
return ProxyFactory.create_multiple_workshift_proxies(dumb_workshifts)
def test_str_to_date1(self):
str_date = "2019-10-1"
date_obj = DateUtil.str_to_date(str_date)
assert date_obj == datetime.strptime(str_date, "%Y-%m-%d").date()