convert.py

# -*- coding: utf8 -*-
__author__ = "Hoseong Son <me@sookcha.com>"

import datetime
import os
import xml.etree.ElementTree as ElementTree

from dateutil import parser
from icalendar import Calendar, Event


class Convert():
    def __init__(self, filename):
        self.filename = filename

    def get_subjects(self):
        result = []
        try:
            tree = ElementTree.parse(self.filename)
            root = tree.getroot()
        except:
            tree = ElementTree.fromstring(self.filename)
            root = tree

        for subject in root.iter('subject'):
            name = subject.find("name").get("value")
            single_subject = {"name": name, "professor": subject.find("professor").get("value"), "info": list(map(
                lambda x: {
                    "day": x.get("day"),
                    "place": x.get("place"),
                    "startAt": '{:02d}:{:02d}'.format(*divmod(int(x.get("starttime")) * 5, 60)),
                    "endAt": '{:02d}:{:02d}'.format(*divmod(int(x.get("endtime")) * 5, 60))
                }, subject.find("time").findall("data")
            )
            )}

            result.append(single_subject)

        return result

    def get_calendar(self, timetable, start_date, end_date, id):
        cal = Calendar()

        for item in timetable:
            for time in item["info"]:
                event = Event()
                event.add('summary', item["name"])
                event.add('dtstart',
                          parser.parse("%s %s" % (self.get_nearest_date(start_date, time["day"]), time["startAt"])))
                event.add('dtend',
                          parser.parse("%s %s" % (self.get_nearest_date(start_date, time["day"]), time["endAt"])))
                event.add('rrule', {'freq': 'WEEKLY', 'until': parser.parse(end_date)})
                if time["place"] != "":
                    event.add('location', time["place"])
                cal.add_component(event)

        if len(str(cal.to_ical())) <= 39:
            return None
        else:
            f = open(os.path.join('/', 'tmp', f'{id}.ics'), 'w+')
            f.write("BEGIN:VCALENDAR\nVERSION:2.0")
            f.close()

            f = open(os.path.join('/', 'tmp', f'{id}.ics'), 'ab')
            f.write(cal.to_ical()[15:])
            f.close()

            print("작업 완료!🙌")

    def get_nearest_date(self, start_date, weekday):
        start_date = parser.parse(start_date)
        weekday = int(weekday)

        if start_date.weekday() >= weekday:
            if start_date.weekday() > weekday: start_date += datetime.timedelta(days=7)
            start_date -= datetime.timedelta(start_date.weekday() - weekday)
        else:
            start_date += datetime.timedelta(weekday - start_date.weekday())

        return start_date