schedule_to_ics.py

import re
from uuid import uuid1
from datetime import datetime, date

import dateutil.relativedelta
from icalendar import Calendar
from icalendar import Event


def display(cal):
    return cal.to_ical().decode('utf-8').replace('\r\n', '\n').strip()


def get_ics(schedule):
    cal = Calendar()
    cal['version'] = '2.0'
    cal['prodid'] = '-//CQUT//Syllabus//CN'  # *mandatory elements* where the prodid can be changed, see RFC 5445
    start_monday = date(2017, 2, 20)  # 开学第一周星期一的时间
    dict_week = {'一': 0, '二': 1, '三': 2, '四': 3, '五': 4, '六': 5, '日': 6}
    # 设置课程时间
    dict_day = {1: dateutil.relativedelta.relativedelta(hours=8, minutes=0),
                2: dateutil.relativedelta.relativedelta(hours=8, minutes=50),
                3: dateutil.relativedelta.relativedelta(hours=9, minutes=50),
                4: dateutil.relativedelta.relativedelta(hours=10, minutes=40),
                5: dateutil.relativedelta.relativedelta(hours=11, minutes=30),
                6: dateutil.relativedelta.relativedelta(hours=11, minutes=30),
                7: dateutil.relativedelta.relativedelta(hours=14, minutes=55),
                8: dateutil.relativedelta.relativedelta(hours=15, minutes=45),
                9: dateutil.relativedelta.relativedelta(hours=16, minutes=25),
                10: dateutil.relativedelta.relativedelta(hours=17, minutes=25),
                11: dateutil.relativedelta.relativedelta(hours=18, minutes=30),
                12: dateutil.relativedelta.relativedelta(hours=19, minutes=20),
                13: dateutil.relativedelta.relativedelta(hours=20, minutes=10)}
    for line in schedule:
        event = Event()
        print(line)
        # line should be like this: ['汇编语言程序设计', '周三第7,8节', '第10-10周|双周', '第1实验楼B403-A', '刘小洋(刘小洋)']

        info_day = re.findall(r'周(.*?)第(\d+),(\d+)?,?(\d+)节', line[1], re.S | re.M)
        info_day = info_day[0]
        print(info_day)
        # info_day should be like this: ('三', '7', '8')

        info_week = re.findall(r'第(\d+)-(\d+)周', line[2], re.S | re.M)
        info_week = info_week[0]
        print(info_week)
        # info_week should be like this: ('10', '10')

        dtstart_date = start_monday + dateutil.relativedelta.relativedelta(
            weeks=(int(info_week[0]) - 1)) + dateutil.relativedelta.relativedelta(
            days=int(dict_week[info_day[0]]))
        dtstart_datetime = datetime.combine(dtstart_date, datetime.min.time())
        ss = dict_day[int(info_day[1])]
        print(ss)
        dtstart = dtstart_datetime + dict_day[int(info_day[1])]
        print('dtstart' + str(dtstart))
        # 有可能是三节课或者是两节课
        # 我们的课持续45分钟（中间有5分钟课间时间）
        dtend = dtstart + dateutil.relativedelta.relativedelta(hours=1, minutes=35)  # 45+45+5=95
        if info_day[2] != '':  # 三节课
            dtend += dateutil.relativedelta.relativedelta(hours=0, minutes=50)  # 95+45+5

        event.add('uid', str(uuid1()) + '@CQUT')
        event.add('summary', line[0])
        event.add('dtstamp', datetime.now())
        event.add('dtstart', dtstart)
        event.add('dtend', dtend)

        if line[2].find('|') == -1:
            interval = 1
            count = int(info_week[1]) - int(info_week[0]) + 1
        else:
            interval = 2
            count = int(info_week[1]) - int(info_week[0]) / 2 + 1
        # 如果有单双周的课 那么这些课隔一周上一次

        event.add('rrule',
                  {'freq': 'weekly', 'interval': interval,
                   'count': count})
        # 设定重复次数

        event.add('location', line[3])
        # 设定重复地点

        cal.add_component(event)
    return cal


def generate_ics(schedule):
    print("获取成功!")
    print("\n课表是...")
    for line in schedule:
        print(line)
    print("\n正在生成 ics 文件...")
    ics = get_ics(schedule)
    print(display(ics))
    print("生成成功!")

    file_name = r'outputs\output.ics'
    print("\n正在保存到..." + file_name)
    with open(file_name, 'wb') as f:
        f.write(ics.to_ical())
        if f:
            print('保存成功!')
        else:
            print('保存失败!')


if __name__ == '__main__':
    data = [['大学物理学【Ⅱ（2）】', '周日第1,2,3节', '第1-10周', '1教0516', '韦建卫']]
    generate_ics(schedule=data)