Add weather commands to a new cog

cogs/weather.py

@@ -0,0 +1,102 @@
+# Imports
+import discord
+import json
+import requests
+from discord import ApplicationContext, option
+from discord.ext import commands
+
+# Variables
+api_key = "21cab08deb7b27f4c2b55f3e2df28ea4"
+with open("database/weather.json", 'r', encoding="utf-8") as f: user_db = json.load(f)
+
+# Functions
+def save():
+    """Dumps all cached databases to storage."""
+    with open("database/weather.json", 'w+', encoding="utf-8") as f: json.dump(user_db, f, indent=4)
+
+# Commands
+class Weather(commands.Cog):
+    def __init__(self, bot):
+        self.bot = bot
+
+    @commands.slash_command(
+        name="weather_set_location",
+        description="Set your default location for the /weather command."
+    )
+    @option(name="location", description="What location do you want to set?", type=str)
+    async def weather_set_location(self, ctx: ApplicationContext, location: str):
+        if str(ctx.author.id) not in user_db: user_db[str(ctx.author.id)] = {"location": None, "scale": "Celsius"}
+        test_ping = requests.get(f"https://api.openweathermap.org/data/2.5/weather?q={location}&appid={api_key}").content
+        test_ping_json = json.loads(test_ping)
+        if test_ping_json["cod"] == 404: return await ctx.respond(":warning: This location does not exist.", ephemeral=True)
+        else:
+            user_db[str(ctx.author.id)]["location"] = location.lower()
+            save()
+            localembed = discord.Embed(description="Your default location has been updated.", color=discord.Color.green())
+            await ctx.respond(embed=localembed)
+
+    @commands.slash_command(
+        name="weather_set_scale",
+        description="Set your preferred unit scale for temperature for the /weather command."
+    )
+    @option(name="scale", description="Which scale do you want to use?", type=str, choices=["Celsius", "Fahrenheit", "Kelvin"])
+    async def weather_set_scale(self, ctx: ApplicationContext, scale: str):
+        if str(ctx.author.id) not in user_db: user_db[str(ctx.author.id)] = {"location": None, "scale": "Celsius"}
+        if scale not in ["Celsius", "Fahrenheit", "Kelvin"]: return 1
+        user_db[str(ctx.author.id)]["scale"] = scale
+        save()
+        localembed = discord.Embed(description="Your preferred unit scale has been updated.", color=discord.Color.green())
+        await ctx.respond(embed=localembed)
+
+    @commands.slash_command(
+        name="weather",
+        description="See the current weather conditions of your set location, or another location."
+    )
+    @option(name="location", description="The location you want weather info about (leave empty for set location)", type=str, default=None)
+    async def weather(self, ctx: ApplicationContext, location: str = None):
+        if str(ctx.author.id) not in user_db: user_db[str(ctx.author.id)] = {"location": None, "scale": "Celsius"}
+        if location == None:
+            if user_db[str(ctx.author.id)]["location"] == None: return await ctx.respond("You do not have a default location set yet.\nEnter a location name and try again.", ephemeral=True)
+            else: location = user_db[str(ctx.author.id)]["location"]
+        api_request = requests.get(f"https://api.openweathermap.org/data/2.5/weather?q={location}&appid={api_key}").content
+        req: dict = json.loads(api_request)
+        print(req)
+        if req["cod"] == 404: return await ctx.respond(":x: This location was not found. Check your spelling or try another location instead.", ephemeral=True)
+        elif req["cod"] != 200: return await ctx.respond("A slight problem occured when trying to get information. This error has been automatically reported to the devs.", ephemeral=True)
+        else: pass
+
+        # Stripped API request data
+        loc_name = req["name"]
+        if user_db[str(ctx.author.id)]["scale"] == "Celsius":
+            temp = round(req["main"]["temp"] - 273)
+            temp_max = round(req["main"]["temp_max"] - 273)
+            temp_min = round(req["main"]["temp_min"] - 273)
+        elif user_db[str(ctx.author.id)]["scale"] == "Fahrenheit":
+            temp = round(((req["main"]["temp"] - 273) * 9/5) + 32)
+            temp_max = round(((req["main"]["temp_max"] - 273) * 9/5) + 32)
+            temp_min = round(((req["main"]["temp_min"] - 273) * 9/5) + 32)
+        else:
+            temp = round(req["main"]["temp"])
+            temp_max = round(req["main"]["temp_max"])
+            temp_min = round(req["main"]["temp_min"])
+        humidity = req["main"]["humidity"]
+        sunset = req["sys"]["sunrise"]
+        sunrise = req["sys"]["sunset"]
+        forcast = req["weather"][0]["main"]
+        forcast_description = req["weather"][0]["description"]
+
+        localembed = discord.Embed(
+            title=f"Weather for {loc_name} :flag_{req['sys']['country'].lower()}:",
+            description=f"**{forcast}**\n{forcast_description}",
+            color=discord.Color.blue()
+        )
+        if user_db[str(ctx.author.id)]["scale"] == "Celsius": localembed.add_field(name="Temperature", value=f"**{temp}C** (max: {temp_max}C,  min: {temp_min}C)")
+        elif user_db[str(ctx.author.id)]["scale"] == "Fahrenheit": localembed.add_field(name="Temperature", value=f"**{temp}F** (max: {temp_max}F,  min: {temp_min}F)")
+        else: localembed.add_field(name="Temperature", value=f"**{temp}K** (max: {temp_max}K,  min: {temp_min}K)")
+        localembed.add_field(name="Humidity", value=f"{humidity}%")
+        localembed.add_field(name="Sunrise", value=f"<t:{sunrise}:f>", inline=False)
+        localembed.add_field(name="Sunset", value=f"<t:{sunset}:f>", inline=True)
+        await ctx.respond(embed=localembed)
+
+# Initialization
+def setup(bot): bot.add_cog(Weather(bot))

config/commands.json

@@ -658,5 +658,35 @@
 	"usable_by": "everyone",
 	"disabled": false,
 	"bugged": false
+  },
+  "weather": {
+	"name": "Weather",
+	"description": "See the current weather conditions of your set location, or another location.",
+	"type": "weather",
+	"cooldown": null,
+	"args": ["location (optional, configurable by /weather_set_location)"],
+	"usable_by": "everyone",
+	"disabled": false,
+	"bugged": false
+  },
+  "weather_set_location": {
+	"name": "Set weather default location",
+	"description": "Set your default location for the `/weather` command.",
+	"type": "weather",
+	"cooldown": null,
+	"args": ["location"],
+	"usable_by": "everyone",
+	"disabled": false,
+	"bugged": false
+  },
+  "weather_set_scale": {
+	"name": "Set weather temperature unit scale",
+	"description": "Set your preferred unit scale for temperature for the `/weather` command.",
+	"type": "weather",
+	"cooldown": null,
+	"args": ["scale"],
+	"usable_by": "everyone",
+	"disabled": false,
+	"bugged": false
   }
 }

database/weather.json

@@ -0,0 +1 @@
+{}

main.py

@@ -296,7 +296,8 @@ async def reload(ctx: ApplicationContext, cog: str):
     "fun",
     "utils",
     "afk",
-    "osu"
+    "osu",
+    "weather"
 ]
 i = 1
 cog_errors = 0