Team 20

Pipfile

@@ -6,8 +6,9 @@ name = "pypi"
 [packages]
 "72eb2aa" = {file = "https://github.com/Rapptz/discord.py/archive/rewrite.zip"}
 aiodns = "*"
-aiohttp = "<2.3.0,>=2.0.0"
+aiohttp = "*"
 websockets = ">=4.0,<5.0"
+google-api-python-client = "*"
 
 [dev-packages]
 "flake8" = "*"

bot/cogs/snakes.py

@@ -2,10 +2,33 @@
 import logging
 from typing import Any, Dict
 
+from googleapiclient.discovery import build
 from discord.ext.commands import AutoShardedBot, Context, command
+from discord import Embed
+
+import aiohttp
+import json
+import async_timeout
+import random
+import difflib
 
 log = logging.getLogger(__name__)
 
+# Probably should move these somewhere
+BASEURL = "https://en.wikipedia.org/w/api.php?format=json&action=query&prop=extracts|pageimages&exintro=&explaintext=&titles={}&redirects=1"
+PYTHON = {
+    "name": "Python",
+    "info": """Python is a species of programming language, \
+commonly used by coding beginners and experts alike. It was first discovered \
+in 1989 by Guido van Rossum in the Netherlands, and was released to the wild \
+two years later. Its use of dynamic typing is one of many distinct features, \
+alongside significant whitespace, heavy emphasis on readability, and, above all, \
+absolutely pain-free software distribution... *sigh*""",
+    "image": "https://www.python.org/static/community_logos/python-logo-master-v3-TM-flattened.png"
+}
+with open("bot/snakes.txt") as file:
+    SNAKES = list(map(lambda ln: ln.strip('\n'), file.readlines()))
+
 
 class Snakes:
     """
@@ -15,31 +38,67 @@ class Snakes:
     def __init__(self, bot: AutoShardedBot):
         self.bot = bot
 
+    @staticmethod
+    def snake_url(name) -> str:
+        """Get the URL of a snake"""
+
+        def encode_url(text):
+            """Encode a string to URL-friendly format"""
+            return BASEURL.format(text.replace(' ', '%20').replace("'", '%27'))
+
+        # Check if the snake name is known
+        if name.upper() in list(map(lambda n: n.upper(), SNAKES)):
+            return encode_url(name)
+
+        # Get a list of similar names if a match wasn't found
+        matches = difflib.get_close_matches(name, SNAKES, n=1)
+        return encode_url(matches[0] if matches != [] else random.choice(SNAKES))
+
+    @staticmethod
+    async def fetch(session, url):
+        """Fetch the contents of a URL as a json"""
+        async with async_timeout.timeout(10):
+            async with session.get(url) as response:
+                return await response.json()
+
     async def get_snek(self, name: str = None) -> Dict[str, Any]:
-        """
-        Go online and fetch information about a snake
+        """Get a snake with a given name, or otherwise randomly"""
+        if name is None:
+            name = random.choice([x.strip('\n') for x in SNAKES])
+
+        elif name.upper() == "PYTHON":
+            return PYTHON
 
-        The information includes the name of the snake, a picture of the snake, and various other pieces of info.
-        What information you get for the snake is up to you. Be creative!
+        # Get snake information
+        async with aiohttp.ClientSession() as session:
+            url = self.snake_url(name)
 
-        If "python" is given as the snake name, you should return information about the programming language, but with
-        all the information you'd provide for a real snake. Try to have some fun with this!
+            # Get the 
+            response = await self.fetch(session, url)
+            page = response["query"]["pages"]
+            content = next(iter(page.values()))
 
-        :param name: Optional, the name of the snake to get information for - omit for a random snake
-        :return: A dict containing information on a snake
-        """
+            # Parse the full-res image from the thumbnail
+            thumb = content.get("thumbnail", {}).get("source", "http://i.imgur.com/HtIPyLy.png/beep")
+            image = "/".join(thumb.replace("thumb/", "").split("/")[:-1])
+
+            # WHY WOULD YOU USE DICTIONARY LITERAL IN A RETURN STATEMENT but okay lol
+            return {
+                "name": content["title"],
+                "info": content.get("extract", "") or "I don't know about that snake!",
+                "image": image
+            }
 
     @command()
     async def get(self, ctx: Context, name: str = None):
-        """
-        Go online and fetch information about a snake
-
-        This should make use of your `get_snek` method, using it to get information about a snake. This information
-        should be sent back to Discord in an embed.
+        content = await self.get_snek(name)
+        # Just a temporary thing to make sure it's working
+        em = Embed()
+        em.title = content["name"]
+        em.description = content["info"][:1970] + "\n\nPS. If the image is a fucking map, blame wikipedia. -Somejuan"
+        em.set_image(url=content["image"])
 
-        :param ctx: Context object passed from discord.py
-        :param name: Optional, the name of the snake to get information for - omit for a random snake
-        """
+        await ctx.send(embed=em)
 
     # Any additional commands can be placed here. Be creative, but keep it to a reasonable amount!
 

run.py

@@ -35,6 +35,8 @@
 # Commands, etc
 bot.load_extension("bot.cogs.snakes")
 
-bot.run(os.environ.get("BOT_TOKEN"))
+#bot.run(os.environ.get("BOT_TOKEN"))
+# I SWEAR TO GOD, HONESTLY WATCH ME LEAVE THIS TOKEN IN LMAO
+bot.run('BOT TOKEN')
 
 bot.http_session.close()  # Close the aiohttp session when the bot finishes running