(Space Cog): Created .apod command that support date parameter.

1 files changed, 27 insertions, 2 deletions

diff --git a/bot/seasons/evergreen/space.py b/bot/seasons/evergreen/space.py
index 7d471d47..2caeb6df 100644
--- a/bot/seasons/evergreen/space.py
+++ b/bot/seasons/evergreen/space.py
@@ -1,8 +1,10 @@
 import logging
-from typing import Any, Dict
+from datetime import datetime
+from typing import Any, Dict, Optional
 from urllib.parse import urlencode
 
-from discord.ext.commands import Cog
+from discord import Embed
+from discord.ext.commands import Cog, Context, command
 
 from bot.bot import SeasonalBot
 from bot.constants import Tokens
@@ -27,6 +29,29 @@ class Space(Cog):
         self.bot = bot
         self.http_session = bot.http_session
 
+    @command(name="apod")
+    async def apod(self, ctx: Context, date: Optional[str] = None) -> None:
+        """Get Astronomy Picture of Day from NASA API. Date is optional parameter, what formatting is YYYY-MM-DD."""
+        # Make copy of parameters
+        params = APOD_PARAMS.copy()
+        # Parse date to params, when provided. Show error message when invalid formatting
+        if date:
+            try:
+                params["date"] = datetime.strptime(date, "%Y-%m-%d").date().isoformat()
+            except ValueError:
+                await ctx.send(f"Invalid date {date}. Please make sure your date is in format YYYY-MM-DD.")
+                return
+
+        # Do request to NASA API
+        result = await self.fetch_from_nasa("planetary/apod", params)
+
+        # Create embed from result
+        embed = Embed(title=f"Astronomy Picture of Day in {result['date']}", description=result["explanation"])
+        embed.set_image(url=result["hdurl"])
+        embed.set_footer(text="Powered by NASA API")
+
+        await ctx.send(embed=embed)
+
     async def fetch_from_nasa(self, endpoint: str, params: Dict[str, Any]) -> Dict[str, Any]:
         """Fetch information from NASA API, return result."""
         # Generate request URL from base URL, endpoint and parsed params