104 lines
3.4 KiB
Python
104 lines
3.4 KiB
Python
import datetime
|
|
from typing import cast
|
|
|
|
from bs4 import BeautifulSoup
|
|
|
|
from edmond.plugin import Plugin
|
|
from edmond.plugins.plus import PlusPlugin
|
|
from edmond.utils import http_get
|
|
|
|
|
|
class JourneeMondialePlugin(Plugin):
|
|
"""This plugin shows today's international observance.
|
|
|
|
It used to fetch data from the website journee-mondiale.com but it is
|
|
regularly broken so it is now loading a static list as a resource, and the
|
|
website is fetched only if the user adds the "jmcom" config value to True.
|
|
|
|
The local list uses the format "MM-DD Name", one per line, e.g.:
|
|
|
|
```
|
|
01-01 NYE
|
|
08-03 Something the 3rd of August
|
|
12-25 Christmas
|
|
```
|
|
|
|
A list can be found on the UN website but it has to be converted by hand:
|
|
https://www.un.org/en/observances/list-days-weeks
|
|
"""
|
|
|
|
REQUIRED_CONFIGS = ["commands", "dates", "no_entry_reply"]
|
|
JMCOM_URL = "https://www.journee-mondiale.com"
|
|
|
|
def __init__(self, bot):
|
|
super().__init__(bot)
|
|
|
|
def on_pubmsg(self, event):
|
|
if not self.should_handle_command(event.arguments[0], no_content=True):
|
|
return False
|
|
|
|
reg_days = self.get_registered_days()
|
|
if reg_days:
|
|
self.bot.say(event.target, ", ".join(reg_days))
|
|
|
|
jmcom_days = []
|
|
if self.config.get("jmcom", False) is True:
|
|
jmcom_days = self.get_jmcom_days(event.target)
|
|
if jmcom_days:
|
|
# Remove days already in reg_days.
|
|
lower_reg_days = [day.lower() for day in reg_days]
|
|
jmcom_days_filtered = [
|
|
day for day in jmcom_days
|
|
if day.lower() not in lower_reg_days
|
|
]
|
|
self.bot.say(event.target, ", ".join(jmcom_days_filtered))
|
|
|
|
if not (reg_days or jmcom_days):
|
|
self.bot.say(event.target, self.config["no_entry_reply"])
|
|
return True
|
|
|
|
def get_registered_days(self) -> list[str]:
|
|
"""Get international days for the local list."""
|
|
now = datetime.datetime.now()
|
|
date_tag = f"{now.month:02}-{now.day:02}"
|
|
today_items = map(
|
|
lambda line: line.split(maxsplit=1)[1],
|
|
filter(
|
|
lambda line: line.startswith(date_tag),
|
|
self.config["dates"],
|
|
),
|
|
)
|
|
return list(today_items)
|
|
|
|
def get_jmcom_days(self, target) -> list[str]:
|
|
"""Get international days from journee-mondiale.com."""
|
|
response = http_get(JourneeMondialePlugin.JMCOM_URL)
|
|
if not response:
|
|
return ""
|
|
|
|
soup = BeautifulSoup(response, "html.parser")
|
|
entries = []
|
|
try:
|
|
items = soup.find("div", id="journeesDuJour").find_all("article")
|
|
for item in items:
|
|
if (first_link := item.find("a")) is None:
|
|
continue
|
|
if (first_title := item.find("h2")) is None:
|
|
continue
|
|
link = first_link["href"]
|
|
title = first_title.string
|
|
entries.append({"url": link, "title": title})
|
|
except (ValueError, KeyError):
|
|
return ""
|
|
|
|
if plus_plugin := cast(PlusPlugin, self.bot.get_plugin("plus")):
|
|
|
|
def handler(plus_event):
|
|
urls = map(lambda i: i["url"], entries)
|
|
self.bot.say(plus_event.target, " — ".join(urls))
|
|
|
|
plus_plugin.add_handler(target, handler)
|
|
|
|
days = map(lambda i: i["title"], entries)
|
|
return list(days)
|