Compare commits

...

3 commits

Author SHA1 Message Date
dece 0f0ab90062 journee_mondiale: plus option for jmcom 2022-09-13 18:31:49 +02:00
dece 725aa7d681 journee_mondiale: fix various issues 2022-09-13 18:23:54 +02:00
dece 37a75f051d plus: remove shortcut option 2022-09-13 18:23:39 +02:00
3 changed files with 29 additions and 24 deletions

View file

@ -152,11 +152,7 @@
"not_fresh_reply": "It's not from today but here it is: {url}" "not_fresh_reply": "It's not from today but here it is: {url}"
}, },
"plus": { "plus": {
"commands": ["plus"], "commands": ["plus"]
"aliases": {
"plus": "more"
},
"shortcut": "+"
}, },
"randomchoice": { "randomchoice": {
"commands": ["choose"], "commands": ["choose"],

View file

@ -1,4 +1,5 @@
import datetime import datetime
from typing import cast
# BS is optional and only for scrapping journee-mondiale.com, thus why we do not # BS is optional and only for scrapping journee-mondiale.com, thus why we do not
# mark the dependencies flag here. # mark the dependencies flag here.
@ -8,6 +9,7 @@ except ImportError:
BeautifulSoup = None BeautifulSoup = None
from edmond.plugin import Plugin from edmond.plugin import Plugin
from edmond.plugins.plus import PlusPlugin
from edmond.utils import http_get from edmond.utils import http_get
@ -31,6 +33,7 @@ class JourneeMondialePlugin(Plugin):
""" """
REQUIRED_CONFIGS = ["commands", "dates", "no_entry_reply"] REQUIRED_CONFIGS = ["commands", "dates", "no_entry_reply"]
JMCOM_URL = "https://www.journee-mondiale.com"
def __init__(self, bot): def __init__(self, bot):
super().__init__(bot) super().__init__(bot)
@ -45,7 +48,7 @@ class JourneeMondialePlugin(Plugin):
jmcom_reply = "" jmcom_reply = ""
if self.config.get("jmcom", False) is True: if self.config.get("jmcom", False) is True:
jmcom_reply = self.get_jmcom_days() jmcom_reply = self.get_jmcom_days(event.target)
if jmcom_reply: if jmcom_reply:
self.bot.say(event.target, jmcom_reply) self.bot.say(event.target, jmcom_reply)
@ -60,15 +63,16 @@ class JourneeMondialePlugin(Plugin):
today_obs = map( today_obs = map(
lambda line: line.split(maxsplit=1)[1], lambda line: line.split(maxsplit=1)[1],
filter( filter(
lambda line: line.startswith(date_tag), self.config["dates"] lambda line: line.startswith(date_tag),
self.config["dates"],
), ),
) )
days = ", ".join(today_obs) days = ", ".join(today_obs)
return reply return days
def get_jmcom_days(self) -> str: def get_jmcom_days(self, target) -> str:
"""Get international days from journee-mondiale.com.""" """Get international days from journee-mondiale.com."""
response = http_get(self.config["url"]) response = http_get(JourneeMondialePlugin.JMCOM_URL)
if not response: if not response:
return "" return ""
@ -77,13 +81,22 @@ class JourneeMondialePlugin(Plugin):
try: try:
items = soup.find("div", id="journeesDuJour").find_all("article") items = soup.find("div", id="journeesDuJour").find_all("article")
for item in items: for item in items:
entries.append({ entries.append(
"url": item.find("a").href, {
"url": item.find("a")["href"],
"title": item.find("h2").string, "title": item.find("h2").string,
}) }
)
except (ValueError, KeyError): except (ValueError, KeyError):
return "" return ""
# TODO add plus plugin support if plus_plugin := cast(PlusPlugin, self.bot.get_plugin("plus")):
return ", ".join(map(lambda i: i["title"], entries)) 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 = ", ".join(map(lambda i: i["title"], entries))
return days

View file

@ -20,16 +20,12 @@ class PlusPlugin(Plugin):
def __init__(self, bot): def __init__(self, bot):
super().__init__(bot) super().__init__(bot)
self.handlers = {} self.handlers = {}
self.shortcut = self.config.get("shortcut")
def on_pubmsg(self, event): def on_pubmsg(self, event):
if ( if not self.should_handle_command(event.arguments[0], no_content=True):
(self.shortcut and event.arguments[0].strip() == self.shortcut) return False
or self.should_handle_command(event.arguments[0], no_content=True)
):
self.process_handler(event) self.process_handler(event)
return True return True
return False
def process_handler(self, event): def process_handler(self, event):
if handler := self.handlers.pop(event.target, None): if handler := self.handlers.pop(event.target, None):