diff options
Diffstat (limited to 'hommaexceli_py/parser.py')
| -rw-r--r-- | hommaexceli_py/parser.py | 13 |
1 files changed, 10 insertions, 3 deletions
diff --git a/hommaexceli_py/parser.py b/hommaexceli_py/parser.py index b4abb46..1b779a6 100644 --- a/hommaexceli_py/parser.py +++ b/hommaexceli_py/parser.py @@ -38,8 +38,15 @@ def _process_row(row: SheetsRow): last_done_date = _timestamp_to_date(row.last_done) next_date = last_done_date + interval next_date_str = _date_to_timestamp(next_date) - return (row.last_done, row.interval, next_date_str, row.name) + return (next_date, row.name, row.interval) -def process_rows(rows: list[SheetsRow]): - return map(_process_row, rows) +def _is_this_week(processed_row): + next_date = processed_row[0] + now = datetime.datetime.now() + return (next_date - now) < datetime.timedelta(weeks=1) + + +def filter_only_this_weeks_rows(rows: list[SheetsRow]): + processed_rows = map(_process_row, rows) + return filter(_is_this_week, processed_rows) |
