Added more fields and responses.

1 files changed, 25 insertions, 5 deletions

diff --git a/bot/exts/info/pypi.py b/bot/exts/info/pypi.py
index 9567516c2..e4c90090d 100644
--- a/bot/exts/info/pypi.py
+++ b/bot/exts/info/pypi.py
@@ -1,10 +1,16 @@
+import logging
+from random import choice
+
 from discord import Embed
 from discord.ext.commands import Cog, Context, command
 
 from bot.bot import Bot
-from bot.constants import NEGATIVE_REPLIES
+from bot.constants import NEGATIVE_REPLIES, Colours
 
 URL = "https://pypi.org/pypi/{package}/json"
+FIELDS = ["author", "requires_python", "description", "license"]
+
+log = logging.getLogger(__name__)
 
 
 class PyPi(Cog):
@@ -16,16 +22,30 @@ class PyPi(Cog):
     @command(name="pypi", aliases=("package", "pack"))
     async def get_package_info(self, ctx: Context, package: str) -> None:
         """Getting information about a specific package."""
-        embed = Embed(title="PyPi package information")
+        embed = Embed(title=choice(NEGATIVE_REPLIES), colour=Colours.soft_red)
 
-        async with self.bot.http_session.get(URL.format(package_name=package)) as response:
+        async with self.bot.http_session.get(URL.format(package=package)) as response:
             if response.status == 404:
-                return await ctx.send(f"Package with name '{package}' could not be found.")
+                embed.description = f"Package could not be found."
+
             elif response.status == 200 and response.content_type == "application/json":
                 response_json = await response.json()
                 info = response_json["info"]
+
+                embed.title = "Python Package Index"
+                embed.colour = Colours.soft_green
+                embed.description = f"[{info['name']} v{info['version']}]({info['download_url']})\n"
+
+                for field in FIELDS:
+                    embed.add_field(
+                        name=field.replace("_", " ").title(),
+                        value=info[field],
+                        inline=False,
+                    )
+
             else:
-                return await ctx.send("There was an error when fetching your PyPi package.")
+                embed.description = "There was an error when fetching your PyPi package."
+                log.trace(f"Error when fetching PyPi package: {response.status}.")
 
         await ctx.send(embed=embed)