From a3a62b630450193d2285e4d564ef9f87872a7953 Mon Sep 17 00:00:00 2001 From: Raphael Roberts Date: Wed, 9 Jan 2019 13:26:07 -0600 Subject: [PATCH] fixed rrule bug in output format --- body_create.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/body_create.py b/body_create.py index 971d858..6046e05 100644 --- a/body_create.py +++ b/body_create.py @@ -42,7 +42,7 @@ def rrule_former(class_obj): ret = rrule.rrule(freq=rrule.WEEKLY,dtstart=start,wkst=rrule.SU,until=end,byweekday=days) ret_str = str(ret).split('\n')[-1] ret_str=re.sub(r'(UNTIL=[^;]+)',r'\1Z',ret_str) - return 'RRULE:{}'.format(ret_str) + return '{}'.format(ret_str) def create_body(class_obj,is_lab = False): if not is_lab: @@ -73,10 +73,12 @@ def json_dump(obj): def create_event(class_list): for class_obj in class_list: - body,lab_body = create_body(class_obj) - yield body - if lab_body: - yield lab_body + out = create_body(class_obj) + if out: + body,lab_body = out + yield body + if lab_body: + yield lab_body if __name__ == "__main__": from scraper import get_classes with open('schedule.html') as file: