diff options
Diffstat (limited to 'weather.py')
| -rw-r--r-- | weather.py | 112 |
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¤t_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¤t_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 |