lukegb/depot
1
0
Fork 0
Code Issues 1 Pull requests Projects Packages Activity Actions

py/icalfilter: update app.py

Browse source
This commit is contained in:
Luke Granger-Brown 2022-05-17 19:34:19 +00:00
parent 28ac784503
commit 5b84094ab0
1 changed files with 10 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
py/icalfilter/app.py
View file

@ -3,8 +3,8 @@ import concurrent.futures
import dataclasses
import datetime
import json
from typing import Any, Dict, List, Optional, Set, Union
import os
from typing import Any, Dict, List, Optional, Set, Union
import aiohttp
import attrs
@ -31,7 +31,10 @@ def maybe_fromisoformat(val: Optional[str]) -> Optional[datetime.datetime]:
return None
if isinstance(val, datetime.datetime):
return val
return datetime.datetime.fromisoformat(val)
dt = datetime.datetime.fromisoformat(val)
if not dt.tzinfo:
dt = dt.replace(tzinfo=UTC)
return dt
@attrs.frozen
@ -97,15 +100,11 @@ def _all_occurrences_before_expensive(
seen_timezones: Dict[str, icalendar.Timezone],
) -> bool:
# Recurring events are... more complicated.
# Generate an iCal of just this event...
fake_cal = icalendar.Calendar()
for tz in seen_timezones.values():
fake_cal.add_component(tz)
fake_cal.add_component(event)
# and check to see if we get any instances that are after the cutoff.
instances = icalevents.icalparser.parse_events(fake_cal.to_ical(), start=cutoff)
return not instances
rrule_or_rruleset = icalevents.icalparser.parse_rrule(event)
try:
return not rrule_or_rruleset.after(cutoff)
except TypeError:
return not rrule_or_rruleset.after(cutoff.replace(tzinfo=None))
async def all_occurrences_before(