Files
iAs/resources/skills/query_weather/scripts/weather.sh
T
wunianxiao 5daaba686b refactor: query_weather JSON 过滤空字段
输出前过滤 forecast 和 now 中的空值字段,避免 LLM 看到 null/空字符串
2026-06-02 10:34:55 +08:00

57 lines
2.7 KiB
Bash
Executable File

#!/bin/bash
# 和风天气查询 → 输出 JSON
read -r input
LOCATION=$(echo "$input" | python3 -c "import sys,json; print(json.load(sys.stdin).get('location',''))" 2>/dev/null)
DAYS=$(echo "$input" | python3 -c "import sys,json; print(json.load(sys.stdin).get('days',3))" 2>/dev/null)
[ -z "$LOCATION" ] && echo '{"error":"请提供 location 参数"}' && exit 0
API_HOST="${QWEATHER_API_HOST}"
KEY_FILE="${QWEATHER_JWT_PRIVATE_KEY_FILE:-qweather/ed25519-private.pem}"
KEY_ID="${QWEATHER_JWT_KEY_ID}"
PROJECT_ID="${QWEATHER_JWT_PROJECT_ID}"
TTL="${QWEATHER_JWT_TTL_SECONDS:-3600}"
NOW=$(date +%s)
HEADER=$(echo -n '{"alg":"EdDSA","kid":"'"$KEY_ID"'"}' | base64 -w0 | tr '+/' '-_' | tr -d '=')
PAYLOAD=$(echo -n '{"sub":"'"$PROJECT_ID"'","iat":'$((NOW-30))',"exp":'$((NOW+TTL))'}' | base64 -w0 | tr '+/' '-_' | tr -d '=')
INPUT="${HEADER}.${PAYLOAD}"
TMP=$(mktemp) && echo -n "$INPUT" > "$TMP"
SIG=$(openssl pkeyutl -sign -inkey "$KEY_FILE" -rawin -in "$TMP" 2>/dev/null | base64 -w0 | tr '+/' '-_' | tr -d '=')
rm -f "$TMP"
JWT="${INPUT}.${SIG}"
[ -z "$JWT" ] && JWT="${QWEATHER_JWT}"
[ -z "$JWT" ] && echo '{"error":"JWT 生成失败"}' && exit 0
ENCODED=$(python3 -c "import urllib.parse; print(urllib.parse.quote('$LOCATION'))")
CITY=$(curl -s --compressed "https://geoapi.qweather.com/v2/city/lookup?location=${ENCODED}" -H "Authorization: Bearer $JWT")
CITY_ID=$(echo "$CITY" | python3 -c "import sys,json; d=json.load(sys.stdin); print(d['location'][0]['id'] if d.get('location') else '')" 2>/dev/null)
[ -z "$CITY_ID" ] && echo '{"error":"未找到城市: '"$LOCATION"'"}' && exit 0
EP_DAYS=$([ "$DAYS" -le 3 ] && echo 3 || echo 7)
NOW_RESP=$(curl -s --compressed "https://${API_HOST}/v7/weather/now?location=${CITY_ID}&lang=zh&unit=m" -H "Authorization: Bearer $JWT")
DAILY_RESP=$(curl -s --compressed "https://${API_HOST}/v7/weather/${EP_DAYS}d?location=${CITY_ID}&lang=zh&unit=m" -H "Authorization: Bearer $JWT")
export NOW_RESP DAILY_RESP LOCATION DAYS
python3 -c "
import os, json
now = json.loads(os.environ['NOW_RESP'])
daily = json.loads(os.environ['DAILY_RESP'])
days = daily.get('daily', [])[:int(os.environ['DAYS'])]
result = {
'location': os.environ['LOCATION'],
'now': {
'temp': now['now']['temp'],
'text': now['now']['text'],
},
'forecast': [
{'date': d['fxDate'], 'day': d['textDay'], 'night': d.get('textNight',''), 'tempMin': d['tempMin'], 'tempMax': d['tempMax']}
for d in days
]
}
# 过滤空字段
result['now'] = {k: v for k, v in result['now'].items() if v}
result['forecast'] = [{k: v for k, v in f.items() if v} for f in result['forecast']]
print(json.dumps(result, ensure_ascii=False))
" 2>/dev/null || echo '{"error":"解析天气数据失败"}'