import requests
import sys
from rich import print
from rich.console import Console

def get_coords(city):
    # geocoding 用 nominatim（OpenStreetMap 無 KEY）
    r = requests.get(
        "https://geocode.maps.co/search",
        params={"q": city}
    )
    data = r.json()
    if not data:
        raise Exception("城市未找到")
    return data[0]["lat"], data[0]["lon"]

def get_weather(lat, lon):
    r = requests.get(
        "https://api.open-meteo.com/v1/forecast",
        params={"latitude": lat, "longitude": lon,
                "current_weather": True}
    )
    return r.json()["current_weather"]

def main():
    if len(sys.argv) < 2:
        print("[red]? 請(qǐng)?zhí)峁┏鞘忻?，例如：python weather.py 北京[/]")
        return

    city = sys.argv[1]
    try:
        lat, lon = get_coords(city)
        cw = get_weather(lat, lon)
        console = Console()
        console.print(f"?? [bold magenta]{city}[/]")
        console.print(f"?? 溫度：{cw['temperature']}°C，風(fēng)速：{cw['windspeed']}km/h，風(fēng)向：{cw['winddirection']}°")
    except Exception as e:
        console = Console()
        console.print(f"[red]? 錯(cuò)誤：{e}[/]")

if __name__ == "__main__":
    main()

API_KEY = "你的和風(fēng)天氣 key"

weather_icons = {
    "晴": "??", "多云": "?", "陰": "??", "小雨": "???", "中雨": "???", 
    "大雨": "???", "暴雨": "???", "雷陣雨": "??", "雪": "??"
}

import requests, sys
from config import API_KEY
from icons import weather_icons

def get_city_code(city):
    url = f"https://geoapi.qweather.com/v2/city/lookup?location={city}&key={API_KEY}"
    r = requests.get(url)
    data = r.json()
    if "location" in data:
        return data["location"][0]["id"]
    return None

def get_weather(city_id):
    url = f"https://devapi.qweather.com/v7/weather/now?location={city_id}&key={API_KEY}"
    r = requests.get(url)
    return r.json()

def display(city, weather):
    now = weather["now"]
    text = now["text"]
    icon = weather_icons.get(text, "")
    print(f"?? {city}")
    print(f"{icon} {text}   {now['temp']}°C")
    print(f"?? {now['windDir']} {now['windScale']}級(jí)   ?? 濕度 {now['humidity']}%")
    print(f"?? 更新時(shí)間：{weather['updateTime'][11:16]}")

if __name__ == "__main__":
    if len(sys.argv) < 2:
        print("?請(qǐng)輸入城市名：python weather.py 北京")
        sys.exit(1)

    city = sys.argv[1]
    city_id = get_city_code(city)
    if not city_id:
        print("? 城市不存在")
        sys.exit(1)

    weather = get_weather(city_id)
    display(city, weather)

python weather.py 上海