def hours_before_six_and_less_than_nine(shift): six_pm = TimeSheet.get_shift_end(shift).replace(hour=18, minute=0) break_start = TimeSheet.get_break_start(shift) break_end = TimeSheet.get_break_end(shift) # shifts where 6pm comes BEFORE 9 hours duty if (TimeSheet.get_shift_start(shift) + timedelta(hours=9)) > TimeSheet.get_shift_end(shift) \ .replace(hour=18, minute=0) < TimeSheet.get_shift_end(shift).replace(hour=12, minute=0): if break_start and break_end < six_pm: # break STARTS and ENDS BEFORE 6pm hours_before_six_pm = ( TimeSheet.get_shift_start(shift) + six_pm) - TimeSheet.get_break_duration(shift) return hours_before_six_pm elif break_start < six_pm < break_end: # break STARTS BEFORE and ENDS AFTER 6pm break_before_six_pm = six_pm - TimeSheet.get_break_start(shift) hours_before_six_pm = ( six_pm - TimeSheet.get_shift_start(shift)) - break_before_six_pm return hours_before_six_pm else: # no break spanning BEFORE 6pm hours_before_six_pm = TimeSheet.get_shift_end(shift).replace(hour=18, minute=0) - \ TimeSheet.get_shift_start(shift) return hours_before_six_pm else: # no break BEFORE 6pm up to 9hrs hours_before_six_pm = TimeSheet.get_shift_end(shift).replace(hour=18, minute=0) - \ TimeSheet.get_shift_start(shift) return hours_before_six_pm
def hours_after_six_and_less_than_nine(shift): six_pm = TimeSheet.get_shift_end(shift).replace(hour=18, minute=0) duty_beyond_twelve = Weekday.duty_beyond_twelve_hours(shift) duty_beyond_nine = Weekday.duty_beyond_nine_hours(shift) overtime_duty = duty_beyond_twelve + duty_beyond_nine duty_remainder = TimeSheet.get_shift_length(shift) - overtime_duty start_plus_remainder = TimeSheet.get_shift_start( shift) + duty_remainder break_start = TimeSheet.get_break_start(shift) break_end = TimeSheet.get_break_end(shift) if TimeSheet.get_shift_start(shift) + duty_remainder > six_pm: if break_start and break_end < six_pm: # break STARTS and ENDS BEFORE 6pm hours_after_six_pm = (start_plus_remainder - six_pm ) - TimeSheet.get_break_duration(shift) return hours_after_six_pm elif break_start < six_pm < break_end: # break STARTS BEFORE and ENDS AFTER 6pm break_after_six_pm = TimeSheet.get_break_end(shift) - six_pm hours_after_six_pm = (start_plus_remainder - six_pm) - break_after_six_pm return hours_after_six_pm else: # no break spanning AFTER 6pm hours_after_six_pm = timedelta(hours=0, minutes=0) return hours_after_six_pm else: # no break within the reminder of duty up to 9hrs hours_after_six_pm = timedelta(hours=0, minutes=0) return hours_after_six_pm