Team 11

.gitignore

@@ -9,6 +9,8 @@ __pycache__/
 # Distribution / packaging
 .Python
 env/
+.idea/
+.env
 build/
 develop-eggs/
 dist/

Pipfile

@@ -8,6 +8,8 @@ name = "pypi"
 aiodns = "*"
 aiohttp = "<2.3.0,>=2.0.0"
 websockets = ">=4.0,<5.0"
+"beautifulsoup4" = "*"
+wikipedia = "*"
 
 [dev-packages]
 "flake8" = "*"

bot/cogs/snakes.py

@@ -1,8 +1,10 @@
 # coding=utf-8
-import logging
+import logging, aiohttp, random, wikipedia
+from time import sleep
+from bs4 import BeautifulSoup
 from typing import Any, Dict
-
 from discord.ext.commands import AutoShardedBot, Context, command
+import asyncio
 
 log = logging.getLogger(__name__)
 
@@ -11,8 +13,22 @@ class Snakes:
     """
     Snake-related commands
     """
+    python_info = '''
+                    Python (Programming Language)
+                    \n
+                    Python is powerful... and fast;\n
+                    plays well with others;\n
+                    runs everywhere;\n
+                    is friendly & easy to learn;\n
+                    is Open.
+                    -------------------------------
+                    Created by: Guido Van Rossum \n
+                    Founded: 20th of February, 1991 \n
+                    Official website: https://python.org
+                '''
 
     def __init__(self, bot: AutoShardedBot):
+        self.inputs = []
         self.bot = bot
 
     async def get_snek(self, name: str = None) -> Dict[str, Any]:
@@ -28,9 +44,23 @@ async def get_snek(self, name: str = None) -> Dict[str, Any]:
         :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
         """
+        name = str(name)
+        site = 'https://en.wikipedia.org/wiki/' + name
+        async with aiohttp.ClientSession() as session:
+            async with session.get(site) as resp:
+                text = await resp.text()
+                soup = BeautifulSoup(text, 'lxml')
+
+
+        if name.lower() == 'python':
+            name = self.python_info
+
+        return name
+
+
 
     @command()
-    async def get(self, ctx: Context, name: str = None):
+    async def get(self, ctx: Context, name: str = None,):
         """
         Go online and fetch information about a snake
 
@@ -40,8 +70,93 @@ async def get(self, ctx: Context, name: str = None):
         :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(BeautifulSoup(text, 'lxml').find("title"))
+        name = str(name)
+        site = 'https://en.wikipedia.org/wiki/' + name
+        async with aiohttp.ClientSession() as session:
+            async with session.get(site) as resp:
+                text = await resp.text()
+                soup = BeautifulSoup(text, 'lxml')
+                title = soup.find('h1').text
+                description = soup.find('table').text
+
+                em = discord.Embed(title=title, description=description)
+
+        if name.lower() == 'python':
+            await ctx.send(await self.get_snek(name))
+        else:
+            await ctx.send(embed=em)
+        # await ctx.send(name)
+
+        # Any additional commands can be placed here. Be creative, but keep it to a reasonable amount!
+    @command()
+    async def snake(self, ctx: Context, x=50, y=30):
+        board = """"""
+        running = True
+        snake = []
+
+        head = [x//2, y//2]
+        snake.append(head)
+        userID = ctx.author.id
+
+        facing = 0
+
+        board += "```\n " + "#" * x + "##"
+        for yAxis in range(y):
+            board += "\n #"
+            for xAxis in range(x):
+                if head == [xAxis, yAxis]:
+                    board += "X"
+                else:
+                        board += "0"
+
+            board += "#"
+        board += "\n " + "#" * x + "##```"
+
+        snakeBoard = await ctx.send(board)
+
+        while running:
+            for mess in self.inputs:
+                if mess.author.id == userID:
+                    self.inputs = []
+                    if mess.content == "a":
+                        facing = (facing - 1) % 4
+                    if mess.content == "d":
+                        facing = (facing + 1) % 4
+                    break
+
+            if facing == 0:
+                head[1] -= 1
+            elif facing == 1:
+                head[0] += 1
+            elif facing == 2:
+                head[1] += 1
+            else:
+                head[0] -= 1
+
+            snake.pop(-1)
+            snake.append(head)
+
+            board = """"""
+            board += "```\n " + "#" * x + "##"
+            for yAxis in range(y):
+                board += "\n #"
+                for xAxis in range(x):
+                    if head == [xAxis, yAxis]:
+                        board += "X"
+                    else:
+                        board += "0"
+
+                board += "#"
+            board += "\n " + "#" * x + "##```"
+
+            await snakeBoard.edit(content=board)
+            await asyncio.sleep(0.8)
 
-    # Any additional commands can be placed here. Be creative, but keep it to a reasonable amount!
+    async def on_message(self, message):
+        if message.content in ("w", "a", "s", "d"):
+            self.inputs.append(message)
+            await message.delete()
 
 
 def setup(bot):