Skip to content
Prev Previous commit
Next Next commit
fix: add offset to init query daterange
  • Loading branch information
Angeluz-07 committed Jun 17, 2020
commit 82782017b5ee49e52283d7bb802846300fe1a510
5 changes: 2 additions & 3 deletions time_tracker_api/time_entries/time_entries_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -467,10 +467,9 @@ def get_worked_time(self, args: dict = {}):
)

conditions = {"owner_id": event_ctx.user_id}
date_range = worked_time.date_range(args.get('time_offset'))
time_entries = self.repository.find_all(
event_ctx,
conditions=conditions,
date_range=worked_time.date_range(),
event_ctx, conditions=conditions, date_range=date_range,
)
return worked_time.summary(
time_entries, time_offset=args.get('time_offset')
Expand Down
6 changes: 4 additions & 2 deletions utils/worked_time.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,8 +39,10 @@ def start(self):
)


def date_range():
dr = MonthDateRange(pytz.UTC)
def date_range(time_offset: int):
offset_in_minutes = time_offset if time_offset else 300
tz = timezone(timedelta(minutes=-offset_in_minutes))
dr = MonthDateRange(tz)
return {
"start_date": datetime_str(dr.start()),
"end_date": datetime_str(dr.end()),
Expand Down