Initial commit

This commit is contained in:
2025-11-14 17:35:52 +01:00
commit 3fc38a20d9
106 changed files with 9123 additions and 0 deletions

544
UserScripts/Weather.py Executable file
View File

@@ -0,0 +1,544 @@
#!/usr/bin/env python3
# /* ---- 💫 https://github.com/JaKooLit 💫 ---- */ #
# Rewritten to use Open-Meteo APIs (worldwide, no API key) for robust weather data.
# Outputs Waybar-compatible JSON and a simple text cache.
import json
import os
import sys
import time
import html
from typing import Any, Dict, List, Optional, Tuple
import requests
# =============== Configuration ===============
# You can configure behavior via environment variables OR the constants below.
# Examples (zsh):
# # One-off run
# # WEATHER_UNITS can be "metric" or "imperial"
# WEATHER_UNITS=imperial WEATHER_PLACE="Concord, NH" python3 /home/dwilliams/Projects/Weather.py
#
# # Persist in current shell session
# export WEATHER_UNITS=imperial
# export WEATHER_LAT=43.2229
# export WEATHER_LON=-71.332
# export WEATHER_PLACE="Concord, NH"
# export WEATHER_TOOLTIP_MARKUP=1 # 1 to enable Pango markup, 0 to disable
# export WEATHER_LOC_ICON="📍" # or "*" for ASCII-only
#
CACHE_DIR = os.path.expanduser("~/.cache")
API_CACHE_PATH = os.path.join(CACHE_DIR, "open_meteo_cache.json")
SIMPLE_TEXT_CACHE_PATH = os.path.join(CACHE_DIR, ".weather_cache")
CACHE_TTL_SECONDS = int(os.getenv("WEATHER_CACHE_TTL", "600")) # default 10 minutes
# Units: metric or imperial (default metric)
UNITS = os.getenv("WEATHER_UNITS", "metric").strip().lower() # metric|imperial
# Optional manual coordinates
ENV_LAT = os.getenv("WEATHER_LAT")
ENV_LON = os.getenv("WEATHER_LON")
# Optional manual place override for tooltip
ENV_PLACE = os.getenv("WEATHER_PLACE")
# Manual place name set inside this file. If set (non-empty), this takes top priority.
# Example: MANUAL_PLACE = "Concord, NH, US"
MANUAL_PLACE: Optional[str] = None
# Location icon in tooltip (default to a standard emoji to avoid missing glyphs)
LOC_ICON = os.getenv("WEATHER_LOC_ICON", "📍")
# Enable/disable Pango markup in tooltip (1/0, true/false)
TOOLTIP_MARKUP = os.getenv("WEATHER_TOOLTIP_MARKUP", "1").lower() not in ("0", "false", "no")
# Optional debug logging to stderr (set WEATHER_DEBUG=1 to enable)
DEBUG = os.getenv("WEATHER_DEBUG", "0").lower() not in ("0", "false", "no")
# HTTP settings
UA = (
"Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 "
"(KHTML, like Gecko) Chrome/128.0 Safari/537.36"
)
TIMEOUT = 8
SESSION = requests.Session()
SESSION.headers.update({"User-Agent": UA})
# =============== Icon and status mapping ===============
# Reuse prior icon set for continuity
WEATHER_ICONS = {
"sunnyDay": "󰖙",
"clearNight": "󰖔",
"cloudyFoggyDay": "",
"cloudyFoggyNight": "",
"rainyDay": "",
"rainyNight": "",
"snowyIcyDay": "",
"snowyIcyNight": "",
"severe": "",
"default": "",
}
WMO_STATUS = {
0: "Clear sky",
1: "Mainly clear",
2: "Partly cloudy",
3: "Overcast",
45: "Fog",
48: "Depositing rime fog",
51: "Light drizzle",
53: "Moderate drizzle",
55: "Dense drizzle",
56: "Freezing drizzle",
57: "Freezing drizzle",
61: "Light rain",
63: "Moderate rain",
65: "Heavy rain",
66: "Freezing rain",
67: "Freezing rain",
71: "Slight snow",
73: "Moderate snow",
75: "Heavy snow",
77: "Snow grains",
80: "Rain showers",
81: "Rain showers",
82: "Violent rain showers",
85: "Snow showers",
86: "Heavy snow showers",
95: "Thunderstorm",
96: "Thunderstorm w/ hail",
99: "Thunderstorm w/ hail",
}
def wmo_to_icon(code: int, is_day: int) -> str:
day = bool(is_day)
if code == 0:
return WEATHER_ICONS["sunnyDay" if day else "clearNight"]
if code in (1, 2, 3, 45, 48):
return WEATHER_ICONS["cloudyFoggyDay" if day else "cloudyFoggyNight"]
if code in (51, 53, 55, 61, 63, 65, 80, 81, 82):
return WEATHER_ICONS["rainyDay" if day else "rainyNight"]
if code in (56, 57, 66, 67, 71, 73, 75, 77, 85, 86):
return WEATHER_ICONS["snowyIcyDay" if day else "snowyIcyNight"]
if code in (95, 96, 99):
return WEATHER_ICONS["severe"]
return WEATHER_ICONS["default"]
def wmo_to_status(code: int) -> str:
return WMO_STATUS.get(code, "Unknown")
# =============== Utilities ===============
def esc(s: Optional[str]) -> str:
return html.escape(s, quote=False) if s else ""
def log_debug(msg: str) -> None:
if DEBUG:
print(msg, file=sys.stderr)
def ensure_cache_dir() -> None:
try:
os.makedirs(CACHE_DIR, exist_ok=True)
except Exception as e:
print(f"Error creating cache dir: {e}", file=sys.stderr)
def read_api_cache() -> Optional[Dict[str, Any]]:
try:
if not os.path.exists(API_CACHE_PATH):
return None
with open(API_CACHE_PATH, "r", encoding="utf-8") as f:
data = json.load(f)
if (time.time() - data.get("timestamp", 0)) <= CACHE_TTL_SECONDS:
return data
return None
except Exception as e:
print(f"Error reading cache: {e}", file=sys.stderr)
return None
def write_api_cache(payload: Dict[str, Any]) -> None:
try:
ensure_cache_dir()
payload["timestamp"] = time.time()
with open(API_CACHE_PATH, "w", encoding="utf-8") as f:
json.dump(payload, f)
except Exception as e:
print(f"Error writing API cache: {e}", file=sys.stderr)
def write_simple_text_cache(text: str) -> None:
try:
ensure_cache_dir()
with open(SIMPLE_TEXT_CACHE_PATH, "w", encoding="utf-8") as f:
f.write(text)
except Exception as e:
print(f"Error writing simple cache: {e}", file=sys.stderr)
def get_coords() -> Tuple[float, float]:
# 1) Explicit env
if ENV_LAT and ENV_LON:
try:
return float(ENV_LAT), float(ENV_LON)
except ValueError:
print("Invalid WEATHER_LAT/WEATHER_LON; falling back to IP geolocation", file=sys.stderr)
# 2) Try cached coordinates from last successful forecast
try:
cached = read_api_cache()
if cached and isinstance(cached, dict):
fc = cached.get("forecast") or {}
lat = fc.get("latitude")
lon = fc.get("longitude")
if isinstance(lat, (int, float)) and isinstance(lon, (int, float)):
return float(lat), float(lon)
except Exception as e:
print(f"Reading cached coords failed: {e}", file=sys.stderr)
# 3) IP-based geolocation with multiple providers (prefer ipwho.is, ipapi.co; ipinfo.io as fallback)
# ipwho.is
try:
resp = SESSION.get("https://ipwho.is/", timeout=TIMEOUT)
resp.raise_for_status()
data = resp.json()
if data.get("success"):
lat = data.get("latitude")
lon = data.get("longitude")
if isinstance(lat, (int, float)) and isinstance(lon, (int, float)):
return float(lat), float(lon)
except Exception as e:
print(f"ipwho.is failed: {e}", file=sys.stderr)
# ipapi.co
try:
resp = SESSION.get("https://ipapi.co/json", timeout=TIMEOUT)
resp.raise_for_status()
data = resp.json()
lat = data.get("latitude")
lon = data.get("longitude")
if isinstance(lat, (int, float)) and isinstance(lon, (int, float)):
return float(lat), float(lon)
except Exception as e:
print(f"ipapi.co failed: {e}", file=sys.stderr)
# ipinfo.io (fallback)
try:
resp = SESSION.get("https://ipinfo.io/json", timeout=TIMEOUT)
resp.raise_for_status()
data = resp.json()
loc = data.get("loc")
if loc and "," in loc:
lat_s, lon_s = loc.split(",", 1)
return float(lat_s), float(lon_s)
except Exception as e:
print(f"ipinfo.io failed: {e}", file=sys.stderr)
# 4) Last resort
print("IP geolocation failed: no providers succeeded", file=sys.stderr)
return 0.0, 0.0
def units_params(units: str) -> Dict[str, str]:
if units == "imperial":
return {
"temperature_unit": "fahrenheit",
"wind_speed_unit": "mph",
"precipitation_unit": "inch",
}
# default metric
return {
"temperature_unit": "celsius",
"wind_speed_unit": "kmh",
"precipitation_unit": "mm",
}
def format_visibility(meters: Optional[float]) -> str:
if meters is None:
return ""
try:
if UNITS == "imperial":
miles = meters / 1609.344
return f"{miles:.1f} mi"
else:
km = meters / 1000.0
return f"{km:.1f} km"
except Exception:
return ""
# =============== API Fetching ===============
def fetch_open_meteo(lat: float, lon: float) -> Dict[str, Any]:
base = "https://api.open-meteo.com/v1/forecast"
params = {
"latitude": lat,
"longitude": lon,
"current": "temperature_2m,apparent_temperature,relative_humidity_2m,wind_speed_10m,wind_direction_10m,weather_code,visibility,precipitation,pressure_msl,is_day",
"hourly": "precipitation_probability",
"daily": "temperature_2m_max,temperature_2m_min",
"timezone": "auto",
}
params.update(units_params(UNITS))
resp = SESSION.get(base, params=params, timeout=TIMEOUT)
resp.raise_for_status()
return resp.json()
def fetch_aqi(lat: float, lon: float) -> Optional[Dict[str, Any]]:
try:
base = "https://air-quality-api.open-meteo.com/v1/air-quality"
params = {
"latitude": lat,
"longitude": lon,
"current": "european_aqi",
"timezone": "auto",
}
resp = SESSION.get(base, params=params, timeout=TIMEOUT)
resp.raise_for_status()
return resp.json()
except Exception as e:
print(f"AQI fetch failed: {e}", file=sys.stderr)
return None
def fetch_place(lat: float, lon: float) -> Optional[str]:
"""Reverse geocode lat/lon to an approximate place. Tries Nominatim first, then Open-Meteo."""
lang = os.getenv("WEATHER_LANG", "en")
# 1) Nominatim (OpenStreetMap)
try:
base = "https://nominatim.openstreetmap.org/reverse"
params = {
"lat": lat,
"lon": lon,
"format": "jsonv2",
"accept-language": lang,
}
headers = {"User-Agent": UA + " Weather.py/1.0"}
resp = SESSION.get(base, params=params, headers=headers, timeout=TIMEOUT)
resp.raise_for_status()
data = resp.json()
address = data.get("address", {})
name = data.get("name") or address.get("city") or address.get("town") or address.get("village") or address.get("hamlet")
admin1 = address.get("state")
country = address.get("country")
parts = [part for part in [name, admin1, country] if part]
if parts:
return ", ".join(parts)
except Exception as e:
log_debug(f"Reverse geocoding (Nominatim) failed: {e}")
# 2) Open-Meteo reverse (fallback)
try:
base = "https://geocoding-api.open-meteo.com/v1/reverse"
params = {
"latitude": lat,
"longitude": lon,
"language": lang,
"format": "json",
}
resp = SESSION.get(base, params=params, timeout=TIMEOUT)
resp.raise_for_status()
data = resp.json()
results = data.get("results") or []
if results:
p = results[0]
name = p.get("name")
admin1 = p.get("admin1")
country = p.get("country")
parts = [part for part in [name, admin1, country] if part]
if parts:
return ", ".join(parts)
except Exception as e:
log_debug(f"Reverse geocoding (Open-Meteo) failed: {e}")
return None
# =============== Build Output ===============
def safe_get(dct: Dict[str, Any], *keys, default=None):
cur: Any = dct
for k in keys:
if isinstance(cur, dict):
if k not in cur:
return default
cur = cur[k]
elif isinstance(cur, list):
try:
cur = cur[k] # type: ignore[index]
except Exception:
return default
else:
return default
return cur
def build_hourly_precip(forecast: Dict[str, Any]) -> str:
try:
times: List[str] = safe_get(forecast, "hourly", "time", default=[]) or []
probs: List[Optional[float]] = safe_get(
forecast, "hourly", "precipitation_probability", default=[]
) or []
cur_time: Optional[str] = safe_get(forecast, "current", "time")
idx = times.index(cur_time) if cur_time in times else 0
window = probs[idx : idx + 6]
if not window:
return ""
parts = [f"{int(p)}%" if p is not None else "-" for p in window]
return " (next 6h) " + " ".join(parts)
except Exception:
return ""
def build_output(lat: float, lon: float, forecast: Dict[str, Any], aqi: Optional[Dict[str, Any]], place: Optional[str] = None) -> Tuple[Dict[str, Any], str]:
cur = forecast.get("current", {})
cur_units = forecast.get("current_units", {})
daily = forecast.get("daily", {})
daily_units = forecast.get("daily_units", {})
temp_val = cur.get("temperature_2m")
temp_unit = cur_units.get("temperature_2m", "")
temp_str = f"{int(round(temp_val))}{temp_unit}" if isinstance(temp_val, (int, float)) else "N/A"
feels_val = cur.get("apparent_temperature")
feels_unit = cur_units.get("apparent_temperature", "")
feels_str = f"Feels like {int(round(feels_val))}{feels_unit}" if isinstance(feels_val, (int, float)) else ""
is_day = int(cur.get("is_day", 1) or 1)
code = int(cur.get("weather_code", -1) or -1)
icon = wmo_to_icon(code, is_day)
status = wmo_to_status(code)
# min/max today (index 0)
tmin_val = safe_get(daily, "temperature_2m_min", 0)
tmax_val = safe_get(daily, "temperature_2m_max", 0)
dtemp_unit = daily_units.get("temperature_2m_min", temp_unit)
tmin_str = f"{int(round(tmin_val))}{dtemp_unit}" if isinstance(tmin_val, (int, float)) else ""
tmax_str = f"{int(round(tmax_val))}{dtemp_unit}" if isinstance(tmax_val, (int, float)) else ""
min_max = f"{tmin_str}\t\t{tmax_str}" if tmin_str and tmax_str else ""
wind_val = cur.get("wind_speed_10m")
wind_unit = cur_units.get("wind_speed_10m", "")
wind_text = f"{int(round(wind_val))}{wind_unit}" if isinstance(wind_val, (int, float)) else ""
hum_val = cur.get("relative_humidity_2m")
humidity_text = f"{int(hum_val)}%" if isinstance(hum_val, (int, float)) else ""
vis_val = cur.get("visibility")
visibility_text = f"{format_visibility(vis_val)}" if isinstance(vis_val, (int, float)) else ""
aqi_val = safe_get(aqi or {}, "current", "european_aqi")
aqi_text = f"AQI {int(aqi_val)}" if isinstance(aqi_val, (int, float)) else "AQI N/A"
hourly_precip = build_hourly_precip(forecast)
prediction = f"\n\n{hourly_precip}" if hourly_precip else ""
# Build place string (priority: MANUAL_PLACE > ENV_PLACE > reverse geocode > lat,lon)
place_str = (MANUAL_PLACE or ENV_PLACE or place or f"{lat:.3f}, {lon:.3f}")
location_text = f"{LOC_ICON} {place_str}"
# Build tooltip (markup or plain)
if TOOLTIP_MARKUP:
# Escape dynamic text to avoid breaking Pango markup
tooltip_text = str.format(
"\t\t{}\t\t\n{}\n{}\n{}\n{}\n\n{}\n{}\n{}{}",
f'<span size="xx-large">{esc(temp_str)}</span>',
f"<big> {icon}</big>",
f"<b>{esc(status)}</b>",
esc(location_text),
f"<small>{esc(feels_str)}</small>" if feels_str else "",
f"<b>{esc(min_max)}</b>" if min_max else "",
f"{esc(wind_text)}\t{esc(humidity_text)}",
f"{esc(visibility_text)}\t{esc(aqi_text)}",
f"<i> {esc(prediction)}</i>" if prediction else "",
)
else:
lines = [
f"{icon} {temp_str}",
status,
location_text,
]
if feels_str:
lines.append(feels_str)
if min_max:
lines.append(min_max)
lines.append(f"{wind_text} {humidity_text}".strip())
lines.append(f"{visibility_text} {aqi_text}".strip())
if prediction:
lines.append(hourly_precip)
tooltip_text = "\n".join([ln for ln in lines if ln])
out_data = {
"text": f"{icon} {temp_str}",
"alt": status,
"tooltip": tooltip_text,
"class": f"wmo-{code} {'day' if is_day else 'night'}",
}
simple_weather = (
f"{icon} {status}\n"
+ f"{temp_str} ({feels_str})\n"
+ (f"{wind_text} \n" if wind_text else "")
+ (f"{humidity_text} \n" if humidity_text else "")
+ f"{visibility_text} {aqi_text}\n"
)
return out_data, simple_weather
def main() -> None:
lat, lon = get_coords()
# Try cache first
cached = read_api_cache()
if cached and isinstance(cached, dict):
forecast = cached.get("forecast")
aqi = cached.get("aqi")
cached_place = cached.get("place") if isinstance(cached.get("place"), str) else None
place_effective = MANUAL_PLACE or ENV_PLACE or cached_place
try:
out, simple = build_output(lat, lon, forecast, aqi, place_effective)
print(json.dumps(out, ensure_ascii=False))
write_simple_text_cache(simple)
return
except Exception as e:
print(f"Cached data build failed, refetching: {e}", file=sys.stderr)
# Fetch fresh
try:
forecast = fetch_open_meteo(lat, lon)
aqi = fetch_aqi(lat, lon)
# Use manual/env place if provided; otherwise reverse geocode
place_effective = MANUAL_PLACE or ENV_PLACE or fetch_place(lat, lon)
write_api_cache({"forecast": forecast, "aqi": aqi, "place": place_effective})
out, simple = build_output(lat, lon, forecast, aqi, place_effective)
print(json.dumps(out, ensure_ascii=False))
write_simple_text_cache(simple)
except Exception as e:
print(f"Open-Meteo fetch failed: {e}", file=sys.stderr)
# Last resort: try stale cache without TTL
try:
if os.path.exists(API_CACHE_PATH):
with open(API_CACHE_PATH, "r", encoding="utf-8") as f:
stale = json.load(f)
out, simple = build_output(lat, lon, stale.get("forecast", {}), stale.get("aqi"), stale.get("place") if isinstance(stale.get("place"), str) else None)
print(json.dumps(out, ensure_ascii=False))
write_simple_text_cache(simple)
return
except Exception as e2:
print(f"Failed to use stale cache: {e2}", file=sys.stderr)
# Fallback minimal output
fallback = {
"text": f"{WEATHER_ICONS['default']} N/A",
"alt": "Unavailable",
"tooltip": "Weather unavailable",
"class": "unavailable",
}
print(json.dumps(fallback, ensure_ascii=False))
if __name__ == "__main__":
main()