74 lines
2.3 KiB
Python
74 lines
2.3 KiB
Python
|
import urllib.parse
|
||
|
|
||
|
import requests
|
||
|
|
||
|
from edmond.plugin import Plugin
|
||
|
|
||
|
BASE_URL = "https://taxref.mnhn.fr/api"
|
||
|
|
||
|
|
||
|
class TaxrefPlugin(Plugin):
|
||
|
|
||
|
REQUIRED_CONFIGS = [
|
||
|
"commands", "not_found_reply", "reply", "ambiguous_reply"
|
||
|
]
|
||
|
|
||
|
def __init__(self, bot):
|
||
|
super().__init__(bot)
|
||
|
|
||
|
def on_pubmsg(self, event):
|
||
|
if not self.should_handle_command(event.arguments[0]):
|
||
|
return False
|
||
|
|
||
|
if self.command.ident == self.config["commands"][0]:
|
||
|
self.search_by_name(self.command.content, event.target)
|
||
|
return True
|
||
|
|
||
|
def search_by_name(self, name, target):
|
||
|
name = urllib.parse.quote(name.lower())
|
||
|
url = f"{BASE_URL}/taxa/search?scientificNames={name}&page=1&size=100"
|
||
|
response = requests.get(url)
|
||
|
if response.status_code != 200:
|
||
|
self.signal_failure(target)
|
||
|
return
|
||
|
data = response.json()
|
||
|
items = data.get("_embedded", {}).get("taxa", [])
|
||
|
|
||
|
if not items:
|
||
|
self.bot.say(target, self.config["not_found_reply"])
|
||
|
return
|
||
|
|
||
|
if len(items) == 1:
|
||
|
# Only one result: use it.
|
||
|
item_to_use = items[0]
|
||
|
else:
|
||
|
# More than one result: if the results contain a corresponding
|
||
|
# species, use it, else return names for sub-species etc.
|
||
|
species_items = []
|
||
|
for item in items:
|
||
|
if item["rankId"] == "ES":
|
||
|
species_items.append(item)
|
||
|
if len(species_items) == 1:
|
||
|
item_to_use = species_items[0]
|
||
|
else:
|
||
|
reply = self.config["ambiguous_reply"]
|
||
|
append = ""
|
||
|
if len(items) > 5:
|
||
|
append = f"… (+{len(items)})"
|
||
|
items = items[:5]
|
||
|
reply += ", ".join(
|
||
|
item["scientificName"] for item in items
|
||
|
)
|
||
|
if append:
|
||
|
reply += append
|
||
|
self.bot.say(target, reply)
|
||
|
return
|
||
|
reply = self.config["reply"].format(
|
||
|
sci_name=item_to_use["scientificName"],
|
||
|
fr_name=item_to_use["frenchVernacularName"],
|
||
|
family=item_to_use["familyName"],
|
||
|
cd_nom=item_to_use["id"],
|
||
|
cd_ref=item_to_use["referenceId"],
|
||
|
)
|
||
|
self.bot.say(target, reply)
|