Initial commit

1 files changed, 112 insertions, 0 deletions

diff --git a/weather.py b/weather.py
new file mode 100644
index 0000000..50b947a
--- /dev/null
+++ b/weather.py
@@ -0,0 +1,112 @@
+import urllib.parse
+import os
+import requests
+from pituophis import Item
+from dotenv import load_dotenv
+from prettytable import PrettyTable
+from pyfiglet import Figlet
+
+load_dotenv()
+
+
+def get_code(code):
+    match code:
+        case 0:
+            return "Clear sky"
+        case 1 | 2 | 3:
+            return "Partly cloudy"
+        case 45 | 48:
+            return "Foggy"
+        case 51 | 53 | 55:
+            return "Drizzle"
+        case 56 | 57:
+            return "Freezing drizzle"
+        case 61 | 63 | 65:
+            return "Rain"
+        case 66 | 67:
+            return "Freezing rain"
+        case 71 | 73 | 675:
+            return "Snow"
+        case 77:
+            return "Snow grains"
+        case 80 | 81 | 82:
+            return "Rain showers"
+        case 85 | 86:
+            return "Snow showers"
+        case 95:
+            return "Thunderstorm"
+        case 96 | 99:
+            return "Thunderstorm with hail"
+        case _:
+            return "Unknown"
+
+
+def get_cities(query):
+    data = requests.get(
+        f"https://geocoding-api.open-meteo.com/v1/search?name={query}&count=10&language=en&format=json").json()
+    if "results" in data:
+        return [Item(
+            itype="0",
+            text=f"{i['name']} ({i['admin1'] + ', ' if 'admin1' in i else ''}{i['country']})",
+            path=f"/weathertxt?latlong={str(i['latitude'])}@{str(i['longitude'])}&city={i['name']}",
+            host=os.getenv("HOSTNAME")
+        ) for i in data['results']]
+    else:
+        return [Item(itype="3", text="City could not be found")]
+
+
+def get_weather(city):
+    result = []
+    query = city.split("?latlong=")[1]
+    latitude = query.split("&city=")[0].split("@")[0]
+    longitude = query.split("&city=")[0].split("@")[1]
+    place = query.split("&city=")[1]
+
+    print(f"https://api.open-meteo.com/v1/forecast?latitude={latitude}&longitude={longitude}&hourly"
+                        f"=temperature_2m,relativehumidity_2m,precipitation_probability,"
+                        f"weathercode,windspeed_10m&daily=weathercode,"
+                        f"temperature_2m_max,temperature_2m_min,sunrise,"
+                        f"sunset&current_weather=true&timezone=auto")
+
+    data = requests.get(f"https://api.open-meteo.com/v1/forecast?latitude={latitude}&longitude={longitude}&hourly"
+                        f"=temperature_2m,relativehumidity_2m,precipitation_probability,"
+                        f"weathercode,windspeed_10m&daily=weathercode,"
+                        f"temperature_2m_max,temperature_2m_min,sunrise,"
+                        f"sunset&current_weather=true&timezone=auto").json()
+
+    f = Figlet(font="big")
+    result += f.renderText("weather").split("\n")
+    f.setFont(font="small")
+
+    result.append("="*80)
+    result.append(f"{place}".center(80))
+    result.append("="*80)
+
+    result += f.renderText("hourly").split("\n")
+    table_hourly = PrettyTable()
+    table_hourly.field_names = ["Time", "Summary", "Temperature", "Humidity", "Precipitation Chance", "Wind Speed"]
+    for idx, i in enumerate(data['hourly']['time'][:23]):
+        table_hourly.add_row([
+            i[-5:],
+            get_code(data['hourly']['weathercode'][idx]),
+            f"{data['hourly']['temperature_2m'][idx]} C",
+            f"{data['hourly']['relativehumidity_2m'][idx]} %",
+            f"{data['hourly']['precipitation_probability'][idx]} %",
+            f"{data['hourly']['windspeed_10m'][idx]} km/h",
+        ])
+    result.append(table_hourly.get_string())
+    result += ["\n"] * 3
+
+    result += f.renderText("daily").split("\n")
+    table_daily = PrettyTable()
+    table_daily.field_names = ["Date", "Summary", "Temperature (Max/Min)", "Sunrise / Sunset"]
+    for idx, i in enumerate(data['daily']['time']):
+        table_daily.add_row([
+            i,
+            get_code(data['daily']['weathercode'][idx]),
+            f"{data['daily']['temperature_2m_max'][idx]} C / {data['daily']['temperature_2m_min'][idx]} C",
+            f"{data['daily']['sunrise'][idx][-5:]} / {data['daily']['sunset'][idx][-5:]}"
+        ])
+    result.append(table_daily.get_string())
+
+    return result