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: