Administrator-py/db/Calendar.py

import re
from datetime import datetime, timezone, timedelta

from discord import Embed
from discord.ext.commands import Bot
import ics
import requests
from sqlalchemy.orm import relationship, Session

from db import Base
from sqlalchemy import Column, Integer, String, BigInteger, ForeignKey, Text, DateTime

name_re = re.compile(r"([A-Z]+ [A-Z]+)")


def url(resources: int, project_id: int, first_date: datetime, last_date: datetime):
    first_date = first_date.strftime("%Y-%m-%d")
    last_date = last_date.strftime("%Y-%m-%d")
    return "http://adelb.univ-lyon1.fr/jsp/custom/modules/plannings/anonymous_cal.jsp?" \
           f"resources={resources}&projectId={project_id}&calType=ical&firstDate={first_date}&lastDate={last_date}"


class Calendar(Base):
    __tablename__ = "calendars"
    id = Column(Integer, primary_key=True)
    name = Column(String, nullable=False)
    resources = Column(Integer, nullable=False)
    project_id = Column(Integer, nullable=False)
    server = Column(BigInteger, nullable=False)
    calendar = Column(Text)
    calendar_update = Column(DateTime)

    calendars_notify = relationship("CalendarNotify", backref="calendar", lazy="subquery")

    def __init__(self, name: str, resources: int, project_id: int, server: int):
        self.name = name
        self.resources = resources
        self.project_id = project_id
        self.server = server

    def cal(self) -> ics.Calendar:
        now = datetime.now()
        if not self.calendar or self.calendar_update <= now - timedelta(minutes=30):
            first_date = now - timedelta(days=now.weekday())
            last_date = now + timedelta(days=31)
            self.calendar = requests.get(url(self.resources, self.project_id, first_date, last_date)).text
            self.calendar_update = now

        return ics.Calendar(self.calendar)

    def events(self, first_date: datetime.date, last_date: datetime.date) -> [ics.Event]:
        events = []
        for e in sorted(list(self.cal().events), key=lambda x: x.begin):
            e.begin = e.begin.replace(tzinfo=timezone.utc).astimezone(tz=None)
            e.end = e.begin.replace(tzinfo=timezone.utc).astimezone(tz=None)
            e.organizer = name_re.findall(e.description)[0]
            events.append(e)
        return list(filter(lambda x: x.begin.date() >= first_date and x.end.date() <= last_date, events))

    async def notify(self, bot: Bot, event: ics.Event):
        for n in self.calendars_notify:
            bot.loop.create_task(n.notify(bot, event))


class CalendarNotify(Base):
    __tablename__ = "calendars_notify"
    channel = Column(BigInteger, primary_key=True)
    calendar_id = Column(Integer, ForeignKey("calendars.id"), primary_key=True)

    def __init__(self, channel: int, calender: int):
        self.channel = channel
        self.calendar_id = calender

    async def notify(self, bot: Bot, event: ics.Event):
        embed = Embed(title="Event is coming !")
        embed.add_field(name=f"{event.begin.strftime('%H:%M')} - {event.end.strftime('%H:%M')}",
                        value=f"{event.name} | {event.location} - {event.organizer}")
        await bot.get_channel(self.channel).send(embed=embed)