deadcade/Matrix-Selfbot
1
1
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Reimplement XKCD Title lookup

Browse source
This commit is contained in:
deadcade 2022-06-12 12:31:51 +02:00
parent 7964fde51b
commit 83b7f55399
2 changed files with 31 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
data/settings.json
View file

@ -23,7 +23,7 @@
"help": "{prefix}help (command): Get information on commands", "help": "{prefix}help (command): Get information on commands",
"ping": "{prefix}ping {Host}: Ping the selfbot (or any host)", "ping": "{prefix}ping {Host}: Ping the selfbot (or any host)",
"roll": "{prefix}roll (Number): Roll a dice (with number sides)", "roll": "{prefix}roll (Number): Roll a dice (with number sides)",
"xkcd": "{prefix}xkcd (Number): Get an XKCD comic", "xkcd": "{prefix}xkcd (Number/Title): Get an XKCD comic",
"shrug": "{prefix}shrug {text}: Append ¯\\_(ツ)_/¯ to the text", "shrug": "{prefix}shrug {text}: Append ¯\\_(ツ)_/¯ to the text",
"emoji_size": "{prefix}emoji_size {size}: Set the default emoji size", "emoji_size": "{prefix}emoji_size {size}: Set the default emoji size",
"emoji_list": "{prefix}emoji_list: Get a list of usable custom emojis" "emoji_list": "{prefix}emoji_list: Get a list of usable custom emojis"

38
main.py
View file

@ -55,6 +55,14 @@ client = None
ratelimits = {} ratelimits = {}
# Common function definitions # Common function definitions
# Filter a title for local lookup with XKCD
def filter_xkcd_title(title):
filtered_title = ""
for char in title.lower().split("(")[0]:
if (char.isdecimal() or char.isalpha()) and (not char == " "):
filtered_title += char
return filtered_title
# Grab a list of all emojis on disk # Grab a list of all emojis on disk
def list_emojis(): def list_emojis():
global settings global settings
@ -221,15 +229,29 @@ async def xkcd(args, room, event):
comic = "" comic = ""
if len(args) == 1 and args[0].isdecimal(): if len(args) == 1 and args[0].isdecimal():
comic = args[0] + "/" comic = args[0] + "/"
try: elif len(args) > 0:
r = requests.get(f"https://xkcd.com/{comic}info.0.json") lookup = {}
r = requests.get("https://xkcd.com/archive/")
for line in r.text.split("\n"):
if "<a href=\"" in line and "\" title=\"2" in line:
num = line.split("/")[1]
title = filter_xkcd_title(line.split(">")[1].split("<")[0])
lookup[title] = num
user_title = filter_xkcd_title(" ".join(args))
if user_title in lookup.keys():
comic = lookup[user_title] + "/"
r = requests.get(f"https://xkcd.com/{comic}info.0.json")
if settings["debug"]:
rj = json.loads(r.text) rj = json.loads(r.text)
filename = download_file(rj["img"], settings["cache_path"] + "/" + str(rj["num"]) + "." + rj["img"].split(".")[-1]) else:
image = await send_file(filename) try:
await send_text(room.room_id, f"{rj['year']}/{rj['month']}/{rj['day']}, {str(rj['num'])}: <a href=\"https://xkcd.com/{str(rj['num'])}/\">{rj['safe_title']}</a>") rj = json.loads(r.text)
return await send_image(room.room_id, image, rj['alt']) except Exception:
except Exception: return await send_text(room.room_id, "Failed to get XKCD!")
return await send_text(room.room_id, "Failed to get XKCD!") filename = download_file(rj["img"], settings["cache_path"] + "/" + str(rj["num"]) + "." + rj["img"].split(".")[-1])
image = await send_file(filename)
await send_text(room.room_id, f"{rj['year']}/{rj['month']}/{rj['day']}, {str(rj['num'])}: <a href=\"https://xkcd.com/{str(rj['num'])}/\">{rj['safe_title']}</a>")
return await send_image(room.room_id, image, rj['alt'])
async def message_callback(room: nio.MatrixRoom, event: nio.RoomMessageText) -> None: async def message_callback(room: nio.MatrixRoom, event: nio.RoomMessageText) -> None:
global client, settings, emojis global client, settings, emojis