138 lines
4.4 KiB
Python
138 lines
4.4 KiB
Python
#!/usr/bin/env python
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
中国天气网天气查询命令行工具
|
||
数据来源：www.weather.com.cn
|
||
"""
|
||
|
||
import argparse
|
||
import json
|
||
import sys
|
||
from weather import query_weather, query_current_weather, search_city, find_city_code
|
||
|
||
|
||
def cmd_query(args):
|
||
"""查询天气预报"""
|
||
city_code = find_city_code(args.city)
|
||
if not city_code:
|
||
print(f"错误：未找到城市 \"{args.city}\"，请使用 search 命令搜索城市")
|
||
return 1
|
||
|
||
if args.current:
|
||
# 查询当前天气
|
||
result = query_current_weather(args.city)
|
||
if result:
|
||
print(result)
|
||
return 0
|
||
else:
|
||
print(f"错误：无法获取 {args.city} 的当前天气")
|
||
return 1
|
||
else:
|
||
# 查询天气预报
|
||
days = args.days or 7
|
||
result = query_weather(args.city, days)
|
||
if result:
|
||
print(result)
|
||
return 0
|
||
else:
|
||
print(f"错误：无法获取 {args.city} 的天气预报")
|
||
return 1
|
||
|
||
|
||
def cmd_search(args):
|
||
"""搜索城市"""
|
||
cities = search_city(args.keyword)
|
||
if not cities:
|
||
print(f"未找到包含 \"{args.keyword}\" 的城市")
|
||
return 1
|
||
|
||
print(f"找到 {len(cities)} 个匹配的城市:")
|
||
print("-" * 40)
|
||
for city in cities[:args.limit]:
|
||
print(f" {city['name']}: {city['code']}")
|
||
return 0
|
||
|
||
|
||
def cmd_json(args):
|
||
"""输出JSON格式的天气数据"""
|
||
city_code = find_city_code(args.city)
|
||
if not city_code:
|
||
print(f"错误：未找到城市 \"{args.city}\"")
|
||
return 1
|
||
|
||
from weather import WeatherQuery
|
||
query = WeatherQuery()
|
||
weather_info = query.query(args.city, args.days or 7)
|
||
|
||
if weather_info:
|
||
data = {
|
||
"city": weather_info[0].city,
|
||
"update_time": weather_info[0].update_time,
|
||
"forecast": [
|
||
{
|
||
"date": info.date,
|
||
"weather": info.weather,
|
||
"temperature": info.temperature,
|
||
"wind": info.wind,
|
||
}
|
||
for info in weather_info
|
||
]
|
||
}
|
||
print(json.dumps(data, ensure_ascii=False, indent=2))
|
||
return 0
|
||
else:
|
||
print(f"错误：无法获取 {args.city} 的天气数据")
|
||
return 1
|
||
|
||
|
||
def main():
|
||
parser = argparse.ArgumentParser(
|
||
description="中国天气网天气查询工具",
|
||
formatter_class=argparse.RawDescriptionHelpFormatter,
|
||
epilog="""
|
||
示例:
|
||
%(prog)s query 杭州 # 查询杭州7天天气预报
|
||
%(prog)s query 北京 --days 3 # 查询北京3天天气预报
|
||
%(prog)s query 深圳 --current # 查询深圳当前天气
|
||
%(prog)s search 州 # 搜索带"州"字的城市
|
||
%(prog)s json 杭州 --days 3 # 输出JSON格式的天气数据
|
||
"""
|
||
)
|
||
|
||
subparsers = parser.add_subparsers(dest="command", help="子命令")
|
||
|
||
# query 命令
|
||
query_parser = subparsers.add_parser("query", help="查询天气预报")
|
||
query_parser.add_argument("city", help="城市名称，如：杭州、北京、上海")
|
||
query_parser.add_argument("--days", type=int, choices=[1, 3, 7, 15, 40],
|
||
help="查询天数 (1, 3, 7, 15, 40)，默认7天")
|
||
query_parser.add_argument("--current", "-c", action="store_true",
|
||
help="查询当前天气而非预报")
|
||
query_parser.set_defaults(func=cmd_query)
|
||
|
||
# search 命令
|
||
search_parser = subparsers.add_parser("search", help="搜索城市")
|
||
search_parser.add_argument("keyword", help="搜索关键词")
|
||
search_parser.add_argument("--limit", type=int, default=20,
|
||
help="返回结果数量限制，默认20")
|
||
search_parser.set_defaults(func=cmd_search)
|
||
|
||
# json 命令
|
||
json_parser = subparsers.add_parser("json", help="输出JSON格式的天气数据")
|
||
json_parser.add_argument("city", help="城市名称")
|
||
json_parser.add_argument("--days", type=int, choices=[1, 3, 7, 15, 40],
|
||
help="查询天数，默认7天")
|
||
json_parser.set_defaults(func=cmd_json)
|
||
|
||
args = parser.parse_args()
|
||
|
||
if not args.command:
|
||
parser.print_help()
|
||
return 1
|
||
|
||
return args.func(args)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
sys.exit(main())
|