Project initHEADmain

1 files changed, 169 insertions, 0 deletions

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 = {'<span style="text-decoration: line-through;">': '~~',
+                      '<span style="font-size:0%;">': '~~',
+                      '</span>': '~~',
+                      '<p>': '',
+                      '</p>': '',
+                      '<strong>': '',
+                      '</strong>': ''}
+
+        for i in scp_p:
+
+            class_list = ["<strong>Object Class:</strong>",
+                          "<strong>Object Class</strong>:",
+                          "<strong>Anomaly Class:</strong>",
+                          "<strong>Anomaly Class</strong>:",
+                          "<strong>Library Class:</strong>",
+                          "<strong>Library Class</strong>:"]
+
+            # 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 "<strong>Description:</strong>" 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 "<strong>Object Class:</strong>" in str(i):
+                    print(f"i{i}")
+                    scp_class = str(i).replace("Object Class: ", "")
+
+                elif "<strong>Anomaly Class:</strong>" in str(i):
+                    print(f"i{i}")
+                    scp_class = str(i).replace("Anomaly Class: ", "")
+
+                elif "<strong>Library Class:</strong>" 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]")