weather.py

import os
import requests
from dotenv import load_dotenv


load_dotenv()

# Add variables in .env file
# API Key from https://www.weatherapi.com/my/
API_KEY = os.environ.get("API_KEY")


def error(r):
    if r.get('error'):
        err = r['error']
        text ="Error ({error['code']}):- "+err['message']
        return text
    else:
        return False


def get_location_details(r):
    l_arr = []
    for i in r:
        if i=="lat":
            name = "Latitude"
        elif i=="lon":
            name = "Longitude"
        elif i=="tz_id":
            name = "Timezone ID"
        else:
            name = i.replace('_', ' ')
            if " " in name:
                words = name.split()
                name = " ".join(word.capitalize() for word in words)
        l_arr.append(f"<b>{name}:</b> {str(r[i])}")
    text = "\n".join(l_arr)
    return text


def uv(n):
    if n<2:
        return "Low"
    if n<5:
        return "Moderate"
    if n<7:
        return "High"
    if n<10:
        return "Very High"
    return "Extreme"


def get_aqi(r):
    aqi_arr = []
    for i in r:
        if i=="co":
            name = "Carbon Monoxide"
        elif i=="no2":
            name = "Nitrogen dioxide"
        elif i=="o3":
            name = "Ozone"
        elif i=="so2":
            name = "Sulfur dioxide"
        elif i=="pm2_5":
            name = "PM 2.5"
        elif i=="pm10":
            name = "PM 10"
        elif i=="us-epa-index":
            name = "US EPA Index"
        elif i=="gb-defra-index":
            name = "GB Defra Index"
        else:
            name = i.replace('_', ' ')
            if " " in name:
                words = name.split()
                name = " ".join(word.capitalize() for word in words)
        aqi_arr.append(f"<b>{name}:</b> {str(r[i])}")
    text = "\n".join(aqi_arr)
    return text


def get_current_details(r):
    curr_arr = []
    for i in r:
        if i in ["condition", "air_quality"]:
            pass
        else:
            value = str(r[i])
            if i=="is_day":
                name = "Day/Night"
                if value=="0":
                    value="Night"
                else:
                    value = "Day"
            elif i=="uv":
                name = "Ultraviolet Index"
                value = str(r[i])+f" ({uv(r[i])})"
            else:
                name = i.replace('_', ' ')
                if " " in name:
                    words = name.split()
                    name = " ".join(word.capitalize() for word in words)
            curr_arr.append(f"<b>{name}:</b> {value}")
    text = "\n".join(curr_arr)
    return text


# Weather checking module
def get_weather(query):
    api = f"http://api.weatherapi.com/v1/current.json?key={API_KEY}&q={query}&aqi=yes"
    r = requests.get(api).json()
    details = []
    
    if error(r):
        return [error(r)]
    loc = ""
    loc += "<b><u>Location Details</b></u>\n\n"
    loc += get_location_details(r['location'])
    details.append(loc)
    
    cw = ""
    cw += "<b><u>Current Weather Details</b></u>\n\n"
    cw += get_current_details(r['current'])
    details.append(cw)
    
    aqi = ""
    aqi += "<b><u>Air Quality Details</b></u>\n\n"
    aqi += get_aqi(r['current']['air_quality'])
    details.append(aqi)
    
    return details