setup_weather.py

#!/usr/bin/env python3
"""
Dadboard - Weather Location Setup
Configure your location for weather data using BOM (Bureau of Meteorology) API
"""

import urllib.request
import urllib.parse
import json
from config_helper import set_setting, get_setting


def search_location(query):
    """Search for a location by name using BOM API"""
    encoded_query = urllib.parse.quote(query)
    url = f"https://api.weather.bom.gov.au/v1/locations?search={encoded_query}"

    req = urllib.request.Request(url, headers={'User-Agent': 'Mozilla/5.0'})
    with urllib.request.urlopen(req, timeout=10) as response:
        data = json.loads(response.read().decode())

    return data.get('data', [])


def test_weather(geohash):
    """Test fetching weather for the given geohash"""
    url = f"https://api.weather.bom.gov.au/v1/locations/{geohash}/forecasts/daily"

    req = urllib.request.Request(url, headers={'User-Agent': 'Mozilla/5.0'})
    with urllib.request.urlopen(req, timeout=10) as response:
        data = json.loads(response.read().decode())

    days = data.get('data', [])
    if not days:
        return None, None, None

    today = days[0]
    temp_max = today.get('temp_max')
    temp_min = today.get('temp_min')
    short_text = today.get('short_text', '')
    if short_text.endswith('.'):
        short_text = short_text[:-1]

    return temp_max, temp_min, short_text


def main():
    print("")
    print("Weather Location Setup")
    print("=" * 40)
    print("")
    print("This uses the BOM (Bureau of Meteorology) API")
    print("")

    # Check existing config
    existing_geohash = get_setting("weather", "geohash", "")
    existing_name = get_setting("weather", "location_name", "")

    if existing_geohash:
        print(f"Current location: {existing_name} (geohash: {existing_geohash})")
        choice = input("Keep existing location? (y/n): ").strip().lower()
        if choice == 'y':
            print("Keeping existing weather location.")
            return

    # Search for location
    while True:
        query = input("Enter your suburb name (or 'skip' to skip): ").strip()

        if query.lower() == 'skip':
            print("Skipping weather setup.")
            return

        if not query:
            continue

        print(f"\nSearching for '{query}'...")

        try:
            results = search_location(query)
        except Exception as e:
            print(f"Error searching: {e}")
            continue

        if not results:
            print("No locations found. Try a different search term.")
            continue

        print("\nSelect your location:")
        for i, loc in enumerate(results, 1):
            name = loc.get('name', 'Unknown')
            state = loc.get('state', '')
            display = name
            if state:
                display += f", {state}"
            print(f"  {i}. {display}")
        print(f"  0. Search again")

        try:
            choice = int(input("\nEnter number: ").strip())
        except ValueError:
            print("Invalid choice.")
            continue

        if choice == 0:
            continue

        if 1 <= choice <= len(results):
            selected = results[choice - 1]
            geohash = selected.get('geohash', '')
            name = selected.get('name', 'Unknown')
            state = selected.get('state', '')

            location_name = name
            if state:
                location_name += f", {state}"

            print(f"\nSelected: {location_name}")
            print(f"Geohash: {geohash}")

            # Test the weather API
            print("\nTesting weather API...")
            try:
                temp_max, temp_min, short_text = test_weather(geohash)
                if temp_max is not None:
                    print(f"Today's forecast: High {temp_max}°C, {short_text}, Low {temp_min}°C")
                else:
                    print("Forecast data not available yet")
            except Exception as e:
                print(f"Warning: Could not fetch weather: {e}")
                confirm = input("Save location anyway? (y/n): ").strip().lower()
                if confirm != 'y':
                    continue

            # Save to config
            set_setting("weather", "geohash", geohash)
            set_setting("weather", "location_name", location_name)

            print(f"\nWeather location saved: {location_name}")
            break


if __name__ == "__main__":
    main()