Edm0nd/edmond/plugins/journee_mondiale.py

104 lines
3.4 KiB
Python
Raw Normal View History

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"]
2022-09-13 18:23:54 +02:00
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."""
2022-09-13 18:23:54 +02:00
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)