mirror of https://github.com/stijndcl/didier
96 lines
2.9 KiB
Python
96 lines
2.9 KiB
Python
from datetime import datetime, timedelta
|
|
from discord import Embed, Colour
|
|
from functions.stringFormatters import leadingZero as lz
|
|
from functions.timeFormatters import intToWeekday
|
|
import pytz
|
|
import re
|
|
|
|
|
|
class UforaNotification:
|
|
def __init__(self, content: dict, course, notif_id, course_id):
|
|
self._content: dict = content
|
|
self._course = course
|
|
self._notif_id, self._course_id = notif_id, course_id
|
|
self._view_url = self._create_url()
|
|
self._title = self._clean_content(self._content["title"])
|
|
self._description = self._get_description()
|
|
self._published = self._get_published()
|
|
|
|
def to_embed(self):
|
|
embed = Embed(colour=Colour.from_rgb(30, 100, 200))
|
|
|
|
embed.set_author(name=self._course)
|
|
embed.title = self._title
|
|
embed.url = self._view_url
|
|
embed.description = self._description
|
|
embed.set_footer(text=self._published)
|
|
|
|
return embed
|
|
|
|
def get_id(self):
|
|
return int(self._notif_id) if self._notif_id is not None else self._content["id"]
|
|
|
|
def _create_url(self):
|
|
if self._notif_id is None or self._course_id is None:
|
|
return self._content["link"]
|
|
|
|
return "https://ufora.ugent.be/d2l/le/news/{0}/{1}/view?ou={0}".format(self._course_id, self._notif_id)
|
|
|
|
def _get_description(self):
|
|
desc = self._clean_content(self._content["summary"])
|
|
|
|
if len(desc) > 500:
|
|
return desc[:497] + "..."
|
|
|
|
return desc
|
|
|
|
def _clean_content(self, text: str):
|
|
# Dict with HTML & markdown tags to replace
|
|
html_table = {
|
|
# CHARACTERS:
|
|
"&": '&',
|
|
""": '"',
|
|
"apos;": "'",
|
|
">": ">",
|
|
"<": "<",
|
|
# MARKDOWN SUPPORT:
|
|
"<b>": "**",
|
|
"</b>": "**",
|
|
"<strong>": "**",
|
|
"</strong>": "**",
|
|
"<i>": "*",
|
|
"</i>": "*",
|
|
"<em>": "*",
|
|
"</em>": "*",
|
|
"<del>": "~~",
|
|
"</del>": "~~",
|
|
"<ins>": "__",
|
|
"</ins>": "__",
|
|
# Represent paragraphs with newlines
|
|
"</p>": "\n",
|
|
"<br>": "\n",
|
|
"<br/>": "\n",
|
|
"<br />": "\n"
|
|
}
|
|
|
|
# Unescape HTML
|
|
for key, value in html_table.items():
|
|
text = text.replace(key, value)
|
|
|
|
# Remove HTML tags
|
|
return re.sub(r"<[^>]*>", "", text)
|
|
|
|
def _get_published(self):
|
|
time_string = "%a, %d %b %Y %H:%M:%S %Z"
|
|
dt = datetime.strptime(self._content["published"], time_string)\
|
|
.astimezone(pytz.timezone("Europe/Brussels"))
|
|
|
|
# Apply timezone offset
|
|
dt = dt + timedelta(hours=dt.utcoffset().seconds//3600)
|
|
|
|
return "{} {}/{}/{} om {}:{}:{}".format(
|
|
intToWeekday(dt.weekday()),
|
|
lz(dt.day), lz(dt.month), lz(dt.year),
|
|
lz(dt.hour), lz(dt.minute), lz(dt.second)
|
|
)
|