Team 13

bot/cogs/snakes.py

@@ -1,7 +1,13 @@
 # coding=utf-8
+import json
 import logging
+import os
+import random
 from typing import Any, Dict
 
+import aiohttp
+
+import discord
 from discord.ext.commands import AutoShardedBot, Context, command
 
 log = logging.getLogger(__name__)
@@ -29,6 +35,29 @@ async def get_snek(self, name: str = None) -> Dict[str, Any]:
         :return: A dict containing information on a snake
         """
 
+    async def get_snek_qwant_json(self, snake_name):
+        """
+        Gets the json from Unsplash for a given snake query
+        :param snake_name: name of the snake
+        :return: the full JSON from the search API
+        """
+        url = "https://api.unsplash.com/search/photos?client_id" \
+              "={0}&query={1}+snake".format(os.environ.get("UNSPLASH_CLIENT_ID"), snake_name)
+        async with aiohttp.ClientSession() as session:
+            async with session.get(url) as response:
+                response = await response.read()
+                return json.loads(response.decode("utf-8"))
+
+    async def get_snek_image(self, name):
+        """
+        Gets the URL of a snake image
+        :param name: name of snake
+        :return: image url
+        """
+        json_response = await self.get_snek_qwant_json(name)
+        rand = random.randint(0, 9)  # prevents returning the same image every time
+        return json_response['results'][rand]['urls']['small']
+
     @command()
     async def get(self, ctx: Context, name: str = None):
         """
@@ -40,6 +69,28 @@ 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
         """
+        embed = discord.Embed(color=0x3E885B)
+        if name == "python":
+            # handle Python special case
+            embed.add_field(
+                name="Python (programming language)",
+                value="*Guido van Rossum*\n\n"
+                      "This language is neither dangerous nor venomous and be found in software globally",
+                inline=False
+            )
+            embed.set_image(url=await self.get_snek_image("python programming language"))
+        else:
+            url = await self.get_snek_image(name)  # not limited to snakes - user can search anything they like
+            embed.add_field(
+                name="common name",
+                value="*sci name*\n\nThis snake is (**not**) venomous and can be found in (...)",
+                inline=False
+            )
+            embed.set_image(url=url)
+        await ctx.channel.send(
+            content=ctx.message.author.mention + " Here's your snek!",
+            embed=embed
+        )
 
     # Any additional commands can be placed here. Be creative, but keep it to a reasonable amount!