Strip exception dates

Don't fail for single exception dates.
pull/26/head
dlichtistw 2018-09-16 22:37:23 +02:00
rodzic 2f518cce81
commit 9803b5007d
1 zmienionych plików z 6 dodań i 2 usunięć

Wyświetl plik

@ -6,6 +6,7 @@ from random import randint
from datetime import datetime, timedelta, date from datetime import datetime, timedelta, date
from icalendar import Calendar from icalendar import Calendar
from icalendar.prop import vDDDLists
from pytz import utc from pytz import utc
@ -340,8 +341,11 @@ def extract_exdates(component):
exd_prop = component.get('exdate') exd_prop = component.get('exdate')
if exd_prop: if exd_prop:
for exd_list in exd_prop: if isinstance(exd_prop, list):
dates.extend(exd.dt for exd in exd_list.dts if (exd)) for exd_list in exd_prop:
dates.extend(exd.dt for exd in exd_list.dts)
elif isinstance(exd_prop, vDDDLists):
dates.extend(exd.dt for exd in exd_prop.dts)
return dates return dates