From 52801b4de1d63cd01191acf7fcee137977140ec0 Mon Sep 17 00:00:00 2001 From: lexicade Date: Fri, 27 Jan 2023 21:06:30 +0000 Subject: Project init --- cogs/SCP.py | 169 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 169 insertions(+) create mode 100644 cogs/SCP.py (limited to 'cogs/SCP.py') diff --git a/cogs/SCP.py b/cogs/SCP.py new file mode 100644 index 0000000..c44d9e2 --- /dev/null +++ b/cogs/SCP.py @@ -0,0 +1,169 @@ +from discord.ext import commands +import discord +import requests +import re +from bs4 import BeautifulSoup as soup +import importlib +import utils +importlib.reload(utils) + + +class SCP(commands.Cog): + def __init__(self, bot): + self.bot = bot + + @commands.defer(ephemeral=False) + @commands.command(aliases=[], application_command_meta=commands.ApplicationCommandMeta(options=[])) + async def scp(self, ctx, scp_id): + # Ensure url is given a three digit ID + wiki_url = f"http://www.scp-wiki.net/scp-{scp_id.rjust(3, '0')}" + + # Test if page exists + scp_html = requests.get(wiki_url) + if scp_html.status_code == 404: + scp_embed = discord.Embed(title=f"The SCP foundation has not published this file.", + description="") + await ctx.send(embed=scp_embed) + return + + # Collect data for the SCP + scp_url = await self.get_title_url(scp_id) + if scp_url is None: + scp_embed = discord.Embed(title=f"No information currently exists for SCP-{scp_id}.", + description="") + await ctx.send(embed=scp_embed) + return + + scp_class, scp_desc = await self.get_details(scp_html) + scp_name = await self.get_name(scp_id, scp_url) + + emb = await utils.embed(ctx, f"{scp_name}", "", url=wiki_url) + + try: + scp_classification_name = scp_class.split(": ")[0] + except Exception as e: + print(f"scp_class:{scp_class} --- e:{e}") + scp_classification_name = "Class: " + + try: + scp_classification_type = scp_class.split(": ")[1] + except Exception as e: + print(f"scp_class:{scp_class} --- e:{e}") + scp_classification_type = "Unknown" + + emb = await utils.field(emb, scp_classification_name, scp_classification_type) + emb = await utils.field(emb, "Description:", scp_desc) + + await ctx.send(embed=emb) + + async def get_details(self, scp_html): + scp_soup = soup(scp_html.text, 'html.parser') + # scp_soup = .replace('style="text-decoration: line-through;"', "~~") + scp_p = scp_soup.find_all("p") + scp_class = None + scp_desc = None + + html_purge = {'': '~~', + '': '~~', + '': '~~', + '

': '', + '

': '', + '': '', + '': ''} + + for i in scp_p: + + class_list = ["Object Class:", + "Object Class:", + "Anomaly Class:", + "Anomaly Class:", + "Library Class:", + "Library Class:"] + + # for class_item in class_list: + # print(i, class_item, class_item in i) + + if any(ext in str(i) for ext in class_list): + for tag_name in html_purge: + i = str(i).replace(tag_name, html_purge[tag_name]) + scp_class = str(i) + + elif "Description:" in str(i): + if "text-decoration: line-through" in str(i): + for tag_name in html_purge: + i = str(i).replace(tag_name, html_purge[tag_name]) + scp_desc = i.get_text().replace("Description: ", "") + if len(scp_desc) > 950: + scp_desc = scp_desc[:950] + "[...]" + + if scp_class is None: + scp_span = scp_soup.find_all("span") + for i in scp_span: + if "Object Class:" in str(i): + print(f"i{i}") + scp_class = str(i).replace("Object Class: ", "") + + elif "Anomaly Class:" in str(i): + print(f"i{i}") + scp_class = str(i).replace("Anomaly Class: ", "") + + elif "Library Class:" in str(i): + print(f"i{i}") + scp_class = str(i).replace("Library Class: ", "") + + scp_class = self.cleanhtml(scp_class) + scp_desc = self.cleanhtml(scp_desc) + return scp_class, scp_desc + + async def get_title_url(self, id): + if str(id).lower().endswith("-j"): + return "http://www.scp-wiki.net/joke-scps" + + elif int(id) <= 0: + return None + + elif int(id) <= 999: + return "http://www.scp-wiki.net/scp-series" + + elif int(id) <= 1999: + return "http://www.scp-wiki.net/scp-series-2" + + elif int(id) <= 2999: + return "http://www.scp-wiki.net/scp-series-3" + + elif int(id) <= 3999: + return "http://www.scp-wiki.net/scp-series-4" + + elif int(id) <= 4999: + return "http://www.scp-wiki.net/scp-series-5" + + else: + return None + + async def get_name(self, id, url): + scp_html = requests.get(url) + scp_soup = soup(scp_html.text, 'html.parser') + + scp_p = scp_soup.find_all("li") + scp_name = None + + for i in scp_p: + if str(id) in str(i.get_text()): + scp_name = i.get_text() + return scp_name + + def cleanhtml(self, raw_html: str): + cleanr = re.compile('<.*?>') + cleantext = re.sub(str(cleanr), '||', str(raw_html)) + cleantext = cleantext.replace("~~", "||") + return cleantext + + +def setup(bot): + print("INFO: Loading [SCP]... ", end="") + bot.add_cog(SCP(bot)) + print("Done!") + + +def teardown(bot): + print("INFO: Unloading [SCP]") -- cgit v1.2.3