|
|
|
|
@ -1,5 +1,6 @@
|
|
|
|
|
import random
|
|
|
|
|
import urllib.parse
|
|
|
|
|
from typing import Optional
|
|
|
|
|
|
|
|
|
|
import requests
|
|
|
|
|
|
|
|
|
|
@ -9,7 +10,10 @@ BASE_URL = "https://taxref.mnhn.fr/api"
|
|
|
|
|
IMG_FETCH_HTML = """\
|
|
|
|
|
<!doctype html>
|
|
|
|
|
<html>
|
|
|
|
|
<head><meta charset="UTF-8"/></head>
|
|
|
|
|
<head>
|
|
|
|
|
<meta charset="UTF-8"/>
|
|
|
|
|
<style>img {{ display: block; max-width: 95%; }}</style>
|
|
|
|
|
</head>
|
|
|
|
|
<body></body>
|
|
|
|
|
<script>
|
|
|
|
|
const urls = [{}];
|
|
|
|
|
@ -52,7 +56,7 @@ class TaxrefPlugin(Plugin):
|
|
|
|
|
self.find_scientific_name(self.command.content, event.target)
|
|
|
|
|
return True
|
|
|
|
|
|
|
|
|
|
def search_by_name(self, name, target):
|
|
|
|
|
def search_by_name(self, name: str, target: str) -> None:
|
|
|
|
|
"""Get species data from a scientific name.
|
|
|
|
|
|
|
|
|
|
Try to disambiguate the results by focusing on species only and their
|
|
|
|
|
@ -116,7 +120,7 @@ class TaxrefPlugin(Plugin):
|
|
|
|
|
if images_reply := self.get_images_reply(item_to_use):
|
|
|
|
|
self.bot.say(target, images_reply)
|
|
|
|
|
|
|
|
|
|
def get_ambiguous_reply(self, items):
|
|
|
|
|
def get_ambiguous_reply(self, items) -> str:
|
|
|
|
|
"""Show a reply with potential species."""
|
|
|
|
|
reply = self.config["ambiguous_reply"]
|
|
|
|
|
append = ""
|
|
|
|
|
@ -128,7 +132,7 @@ class TaxrefPlugin(Plugin):
|
|
|
|
|
reply += append
|
|
|
|
|
return reply
|
|
|
|
|
|
|
|
|
|
def get_images_reply(self, item):
|
|
|
|
|
def get_images_reply(self, item) -> Optional[str]:
|
|
|
|
|
"""If there are media available, return one in a message.
|
|
|
|
|
|
|
|
|
|
If shrlok is available, return a link to an HTML page shared by shrlok.
|
|
|
|
|
@ -145,16 +149,19 @@ class TaxrefPlugin(Plugin):
|
|
|
|
|
"""
|
|
|
|
|
m_url = item.get("_links", {}).get("media", {}).get("href")
|
|
|
|
|
if not m_url:
|
|
|
|
|
self.bot.log_d("No media links.")
|
|
|
|
|
return None
|
|
|
|
|
response = requests.get(m_url)
|
|
|
|
|
if response.status_code != 200:
|
|
|
|
|
if (code := response.status_code) != 200:
|
|
|
|
|
self.bot.log_d(f"Failed to reach media link ({code}).")
|
|
|
|
|
return None
|
|
|
|
|
media_data = response.json()
|
|
|
|
|
items = media_data.get("_embedded", {}).get("media", [])
|
|
|
|
|
if not items:
|
|
|
|
|
self.bot.log_d("No media found in response.")
|
|
|
|
|
return None
|
|
|
|
|
|
|
|
|
|
def get_img_url(item):
|
|
|
|
|
def get_img_url(item) -> Optional[str]:
|
|
|
|
|
return item.get("_links", {}).get("file", {}).get("href")
|
|
|
|
|
|
|
|
|
|
if shrlok := self.bot.get_plugin("shrlok"):
|
|
|
|
|
@ -163,14 +170,19 @@ class TaxrefPlugin(Plugin):
|
|
|
|
|
urls = map(get_img_url, items)
|
|
|
|
|
urls_text = ",".join(map(lambda url: f'"{url}"', urls))
|
|
|
|
|
html = IMG_FETCH_HTML.format(urls_text)
|
|
|
|
|
link = shrlok.post_html(html)
|
|
|
|
|
link = shrlok.post_html(html) # type: ignore
|
|
|
|
|
if not link:
|
|
|
|
|
self.bot.log_d("shrlok plugin returned an empty string.")
|
|
|
|
|
else:
|
|
|
|
|
link = get_img_url(random.choice(items))
|
|
|
|
|
if not link:
|
|
|
|
|
self.bot.log_d("No link found.")
|
|
|
|
|
|
|
|
|
|
if link:
|
|
|
|
|
return "📷 " + link
|
|
|
|
|
return None
|
|
|
|
|
|
|
|
|
|
def find_scientific_name(self, name, target):
|
|
|
|
|
def find_scientific_name(self, name: str, target: str):
|
|
|
|
|
"""Find a corresponding scientific name for a vernacular name."""
|
|
|
|
|
name = name.lower()
|
|
|
|
|
enc_name = urllib.parse.quote(name)
|
|
|
|
|
@ -207,7 +219,7 @@ class TaxrefPlugin(Plugin):
|
|
|
|
|
)
|
|
|
|
|
+ "\n"
|
|
|
|
|
)
|
|
|
|
|
reply = shrlok.post_text(text)
|
|
|
|
|
reply = shrlok.post_text(text) # type: ignore
|
|
|
|
|
else:
|
|
|
|
|
reply = self.get_ambiguous_reply(items)
|
|
|
|
|
|
|
|
|
|
|
