Project initHEADmain

1 files changed, 127 insertions, 0 deletions

diff --git a/cogs/Weather.py b/cogs/Weather.py
new file mode 100644
index 0000000..84bcf20
--- /dev/null
+++ b/cogs/Weather.py
@@ -0,0 +1,127 @@
+from discord.ext import commands
+import discord
+import requests
+import json
+import importlib
+import utils
+import datetime
+importlib.reload(utils)
+
+
+class Weather(commands.Cog):
+    def __init__(self, bot):
+        self.bot = bot
+
+    @staticmethod
+    async def convert_time(time, format='%H:%M:%S'):
+        return datetime.datetime.utcfromtimestamp(time).strftime(format)
+
+    @staticmethod
+    async def convert_icon(name):
+        icons = {"Clear": "🌞",
+                 "Clouds": "☁",
+                 "Rain": "🌧️",
+                 "Thunderstorm": "⛈",
+                 "Mist": "🌁",
+                 "Snow": "❄"}
+        return icons[name]
+
+    @commands.command(
+        aliases=[],
+        application_command_meta=commands.ApplicationCommandMeta(
+            options=[
+                discord.ApplicationCommandOption(
+                    name="location",
+                    description="Enter a location tog et weather data from.",
+                    type=discord.ApplicationCommandOptionType.string,
+                    required=True,
+                ),
+            ],
+        )
+    )
+    async def weather(self, ctx, location):
+        """Get the weather of any area."""
+        location = location.replace(" ", "%20")
+        response = requests.get(f"https://api.openweathermap.org/data/2.5/weather?q={location}&appid=d83a0ac0ba2d9d59007724f54bd4f960&units=metric")
+        json_data = json.loads(response.text)
+
+        if json_data['cod'] == 200:
+            str_sunrise = await self.convert_time(int(json_data['sys']['sunrise'] + json_data['timezone']))
+            str_sunset = await self.convert_time(int(json_data['sys']['sunset'] + json_data['timezone']))
+            str_st = await self.convert_time(int(json_data['dt'] + json_data['timezone']))
+            icn_weather = await self.convert_icon(json_data['weather'][0]['main'])
+
+            emb = await utils.embed(ctx, f"🗺️ Weather for {json_data['name']} from {str_st}", "")
+            emb = await utils.field(emb, f"{icn_weather} Current forecast", f"{str(json_data['weather'][0]['description']).capitalize()}", inline=False)
+            emb = await utils.field(emb, f"🌡️ Temperature", f"{json_data['main']['temp_min']} to {json_data['main']['temp_max']}°С", inline=True)
+            emb = await utils.field(emb, f"💨 Wind", f"{json_data['wind']['speed']} m/s", inline=True)
+            emb = await utils.field(emb, f"☁ Clouds", f"{json_data['clouds']['all']}%", inline=True)
+            emb = await utils.field(emb, f"📏 Pressure", f"{json_data['main']['pressure']}hPa", inline=True)
+            emb = await utils.field(emb, f"💦 Humidity", f"{json_data['main']['humidity']}%", inline=True)
+            emb = await utils.field(emb, f"👁️ Visibility", f"{json_data['visibility']/1000}KM", inline=True)
+            emb = await utils.field(emb, f"🌅 Sunrise", f"{str_sunrise}", inline=True)
+            emb = await utils.field(emb, f"🌇 Sunset", f"{str_sunset}", inline=True)
+
+        else:
+            emb = await utils.embed(ctx, f"We can't find that area.", f"Check that ***`{location}`*** is spelt right and try again.")
+
+        await ctx.send(embed=emb)
+
+    @commands.command(
+        aliases=[],
+        application_command_meta=commands.ApplicationCommandMeta(
+            options=[
+                discord.ApplicationCommandOption(
+                    name="location",
+                    description="Enter a location to get forecast data from.",
+                    type=discord.ApplicationCommandOptionType.string,
+                    required=True,
+                ),
+            ],
+        )
+    )
+    async def forecast(self, ctx, location):
+        """Get a forecast of any area."""
+        location = location.replace(" ", "%20")
+        response = requests.get(f"https://api.openweathermap.org/data/2.5/weather?q={location}&appid=d83a0ac0ba2d9d59007724f54bd4f960&units=metric")
+        json_data1 = json.loads(response.text)
+
+        if json_data1['cod'] == 200:
+            # Get first request
+            str_name = json_data1['name']
+            response = requests.get(f"https://api.openweathermap.org/data/2.5/onecall?lat={json_data1['coord']['lat']}&lon={json_data1['coord']['lon']}&exclude=hourly,minutely,current&appid=d83a0ac0ba2d9d59007724f54bd4f960&units=metric")
+            json_data2 = json.loads(response.text)
+            int_timezone = int(json_data2['timezone_offset'])
+
+            emb = await utils.embed(ctx, f"🗺️ Forecast for {str_name}", "")
+
+            # Get second request for lat/lon based forecast
+            for day in json_data2['daily']:
+                str_day = await self.convert_time(int(day['dt']) + int_timezone, '%A %d %b')
+                icn_weather = await self.convert_icon(day['weather'][0]['main'])
+
+                day_details = f"{icn_weather} {day['weather'][0]['main']}\n" + \
+                              f"Chance of rain: {int(day['pop'] * 100)}%\n" + \
+                              f"UV Index: {day['uvi']}\n" + \
+                              f"Temp: {day['temp']['min']} to {day['temp']['max']}°С"
+
+                emb = await utils.field(emb, f"{str_day}", day_details, inline=True)
+
+        else:
+            emb = await utils.embed(ctx, f"We can't find the forecast for that area.", f"Check that ***`{location}`*** is spelt right and try again.")
+
+        await ctx.send(embed=emb)
+
+
+
+def setup(bot):
+    print("INFO: Loading [Weather]... ", end="")
+    bot.add_cog(Weather(bot))
+    print("Done!")
+
+
+def teardown(bot):
+    print("INFO: Unloading [Weather]")
+
+
+# Tuesday 11 Aug - Rain
\ No newline at end of file