Merge pull request #86 from stijndcl/ufora_notifications

Fixes for Ufora Notifications
pull/87/head
Stijn De Clercq 2021-08-11 20:46:28 +02:00 committed by GitHub
commit cdb8a04cde
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 16 additions and 45 deletions

View File

@ -1,17 +1,17 @@
from datetime import datetime, timedelta from datetime import datetime
from discord import Embed, Colour from discord import Embed, Colour
from functions.stringFormatters import leadingZero as lz from functions.stringFormatters import leadingZero as lz
from functions.timeFormatters import intToWeekday from functions.timeFormatters import intToWeekday
from html import unescape from markdownify import markdownify as md
import pytz import pytz
import re
class UforaNotification: class UforaNotification:
def __init__(self, content: dict, course, notif_id, course_id): def __init__(self, content: dict, course, notif_id, course_id):
self._content: dict = content self._content: dict = content
self._course = course self._course = course
self._notif_id, self._course_id = notif_id, course_id self._notif_id = notif_id
self._course_id = course_id
self._view_url = self._create_url() self._view_url = self._create_url()
self._title = self._clean_content(self._content["title"]) self._title = self._clean_content(self._content["title"])
self._description = self._get_description() self._description = self._get_description()
@ -40,54 +40,24 @@ class UforaNotification:
def _get_description(self): def _get_description(self):
desc = self._clean_content(self._content["summary"]) desc = self._clean_content(self._content["summary"])
if len(desc) > 500: if len(desc) > 4096:
return desc[:497] + "..." return desc[:4093] + "..."
return desc return desc
def _clean_content(self, text: str): def _clean_content(self, text: str):
# Dict with HTML & markdown tags to replace return md(text)
html_table = {
# CHARACTERS:
"&": '&',
""": '"',
"apos;": "'",
">": ">",
"&lt;": "<",
# 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 unescape(re.sub(r"<[^>]*>", "", text))
def _get_published(self): def _get_published(self):
time_string = "%a, %d %b %Y %H:%M:%S %Z" # Datetime is unable to parse the timezone because it's useless
dt = datetime.strptime(self._content["published"], time_string)\ # We will hereby cut it out and pray the timezone will always be UTC+0
published = self._content["published"].rsplit(" ", 1)[0]
time_string = "%a, %d %b %Y %H:%M:%S"
dt = datetime.strptime(published, time_string)\
.astimezone(pytz.timezone("Europe/Brussels")) .astimezone(pytz.timezone("Europe/Brussels"))
# Apply timezone offset # Apply timezone offset in a hacky way
dt = dt + timedelta(hours=dt.utcoffset().seconds//3600) dt = dt + dt.utcoffset()
return "{} {}/{}/{} om {}:{}:{}".format( return "{} {}/{}/{} om {}:{}:{}".format(
intToWeekday(dt.weekday()), intToWeekday(dt.weekday()),

View File

@ -18,3 +18,4 @@ Quart-CORS==0.5.0
attrs~=21.2.0 attrs~=21.2.0
dacite~=1.6.0 dacite~=1.6.0
pytest==6.2.4 pytest==6.2.4
markdownify==0.9.2