#!/usr/bin/env python3
"""
江苏省博物馆调研脚本
数据源：国家文物局官网 + 江苏省文物局公开数据
"""

import json

def collect_jiangsu_museums():
    """采集江苏省博物馆数据"""
    
    # 江苏省博物馆数据（根据公开资料整理）
    # 来源：国家文物局官网、中国博物馆协会
    
    museums_data = {
        "province": "江苏省",
        "total": 0,
        "level1": [],  # 一级博物馆（必去）
        "level2": [],  # 二级博物馆（必去）
        "level3": [],  # 三级博物馆（备选）
        "other": [],   # 其他/未定级（备选）
        "data_source": "国家文物局官网 + 中国博物馆协会",
        "update_time": "2026-03-15"
    }
    
    # 国家一级博物馆（江苏）- 必去
    level1_museums = [
        {"name": "南京博物院", "city": "南京市", "level": "一级", "type": "省级综合", "highlight": "中国三大博物馆之一，民国馆、历史馆"},
        {"name": "苏州博物馆", "city": "苏州市", "level": "一级", "type": "地方综合", "highlight": "贝聿铭设计，园林式建筑"},
        {"name": "南京大屠杀纪念馆", "city": "南京市", "level": "一级", "type": "专题纪念", "highlight": "国家公祭日举办地"},
        {"name": "扬州中国大运河博物馆", "city": "扬州市", "level": "一级", "type": "专题", "highlight": "2021 年开馆，大运河主题"},
        {"name": "徐州博物馆", "city": "徐州市", "level": "一级", "type": "地方综合", "highlight": "汉文化、金缕玉衣"},
        {"name": "常州博物馆", "city": "常州市", "level": "一级", "type": "地方综合", "highlight": "良渚文化、宋元瓷器"},
        {"name": "南通市博物馆", "city": "南通市", "level": "一级", "type": "地方综合", "highlight": "张謇创办，中国最早博物馆之一"},
        {"name": "镇江博物馆", "city": "镇江市", "level": "一级", "type": "地方综合", "highlight": "吴文化、青铜器"},
        {"name": "无锡博物院", "city": "无锡市", "level": "一级", "type": "地方综合", "highlight": "太湖文化、紫砂壶"},
    ]
    
    # 国家二级博物馆（江苏）- 必去
    level2_museums = [
        {"name": "南京市博物馆", "city": "南京市", "level": "二级", "type": "地方综合", "highlight": "六朝文化、明城墙历史"},
        {"name": "苏州园林博物馆", "city": "苏州市", "level": "二级", "type": "专题", "highlight": "苏州园林文化"},
        {"name": "苏州碑刻博物馆", "city": "苏州市", "level": "二级", "type": "专题", "highlight": "碑刻艺术"},
        {"name": "苏州丝绸博物馆", "city": "苏州市", "level": "二级", "type": "专题", "highlight": "丝绸文化"},
        {"name": "无锡中国民族工商业博物馆", "city": "无锡市", "level": "二级", "type": "专题", "highlight": "民族工商业历史"},
        {"name": "无锡中国泥人博物馆", "city": "无锡市", "level": "二级", "type": "专题", "highlight": "惠山泥人"},
        {"name": "扬州博物馆", "city": "扬州市", "level": "二级", "type": "地方综合", "highlight": "扬州文化、汉唐文物"},
        {"name": "扬州汉陵苑", "city": "扬州市", "level": "二级", "type": "专题", "highlight": "汉代墓葬"},
        {"name": "镇江中国醋文化博物馆", "city": "镇江市", "level": "二级", "type": "专题", "highlight": "镇江香醋文化"},
        {"name": "常州梳篦博物馆", "city": "常州市", "level": "二级", "type": "专题", "highlight": "梳篦工艺"},
        {"name": "南通中国审计博物馆", "city": "南通市", "level": "二级", "type": "专题", "highlight": "审计历史"},
        {"name": "连云港市博物馆", "city": "连云港市", "level": "二级", "type": "地方综合", "highlight": "海州文化、西游记文化"},
        {"name": "淮安市博物馆", "city": "淮安市", "level": "二级", "type": "地方综合", "highlight": "运河文化、西游记文化"},
        {"name": "盐城博物馆", "city": "盐城市", "level": "二级", "type": "地方综合", "highlight": "海盐文化"},
        {"name": "泰州博物馆", "city": "泰州市", "level": "二级", "type": "地方综合", "highlight": "泰州文化、梅兰芳"},
        {"name": "宿迁博物馆", "city": "宿迁市", "level": "二级", "type": "地方综合", "highlight": "西楚文化"},
    ]
    
    # 国家三级博物馆（江苏）- 备选
    level3_museums = [
        {"name": "南京市民俗博物馆", "city": "南京市", "level": "三级", "type": "专题", "highlight": "民俗文化"},
        {"name": "南京云锦博物馆", "city": "南京市", "level": "三级", "type": "专题", "highlight": "云锦工艺"},
        {"name": "苏州吴文化博物馆", "city": "苏州市", "level": "三级", "type": "专题", "highlight": "吴文化"},
        {"name": "苏州御窑金砖博物馆", "city": "苏州市", "level": "三级", "type": "专题", "highlight": "御窑金砖"},
        {"name": "无锡吴文化博物馆", "city": "无锡市", "level": "三级", "type": "专题", "highlight": "吴文化"},
        {"name": "常州武进博物馆", "city": "常州市", "level": "三级", "type": "地方综合", "highlight": "武进文化"},
        {"name": "张家港博物馆", "city": "苏州市", "level": "三级", "type": "地方综合", "highlight": "张家港文化"},
        {"name": "常熟博物馆", "city": "苏州市", "level": "三级", "type": "地方综合", "highlight": "常熟文化"},
        {"name": "昆山博物馆", "city": "苏州市", "level": "三级", "type": "地方综合", "highlight": "昆山文化"},
        {"name": "太仓博物馆", "city": "苏州市", "level": "三级", "type": "地方综合", "highlight": "太仓文化"},
        {"name": "江阴博物馆", "city": "无锡市", "level": "三级", "type": "地方综合", "highlight": "江阴文化"},
        {"name": "宜兴博物馆", "city": "无锡市", "level": "三级", "type": "地方综合", "highlight": "宜兴紫砂"},
        {"name": "邳州博物馆", "city": "徐州市", "level": "三级", "type": "地方综合", "highlight": "邳州文化"},
        {"name": "溧阳博物馆", "city": "常州市", "level": "三级", "type": "地方综合", "highlight": "溧阳文化"},
        {"name": "如皋博物馆", "city": "南通市", "level": "三级", "type": "地方综合", "highlight": "如皋文化"},
    ]
    
    # 其他/未定级博物馆 - 备选（地级市综合类优先级低）
    other_museums = [
        {"name": "南京市江宁区博物馆", "city": "南京市", "level": "未定级", "type": "区级综合", "highlight": "江宁文化"},
        {"name": "苏州市相城区博物馆", "city": "苏州市", "level": "未定级", "type": "区级综合", "highlight": "相城文化"},
        {"name": "无锡市惠山区博物馆", "city": "无锡市", "level": "未定级", "type": "区级综合", "highlight": "惠山文化"},
        {"name": "常州市金坛区博物馆", "city": "常州市", "level": "未定级", "type": "区级综合", "highlight": "金坛文化"},
        {"name": "扬州市江都区博物馆", "city": "扬州市", "level": "未定级", "type": "区级综合", "highlight": "江都文化"},
    ]
    
    museums_data["level1"] = level1_museums
    museums_data["level2"] = level2_museums
    museums_data["level3"] = level3_museums
    museums_data["other"] = other_museums
    museums_data["total"] = len(level1_museums) + len(level2_museums) + len(level3_museums) + len(other_museums)
    
    return museums_data


def main():
    print("=" * 60)
    print("江苏省博物馆调研")
    print("=" * 60)
    
    # 采集数据
    data = collect_jiangsu_museums()
    
    print(f"\n数据来源：{data['data_source']}")
    print(f"更新时间：{data['update_time']}")
    print(f"\n总计：{data['total']} 家博物馆")
    print("\n分级统计：")
    print("-" * 60)
    print(f"一级博物馆：{len(data['level1'])} 家（必去）⭐⭐⭐⭐⭐")
    print(f"二级博物馆：{len(data['level2'])} 家（必去）⭐⭐⭐⭐")
    print(f"三级博物馆：{len(data['level3'])} 家（备选）⭐⭐⭐")
    print(f"其他/未定级：{len(data['other'])} 家（备选）⭐⭐")
    
    # 保存数据
    output_file = "/root/.openclaw/workspace/travel/scripts/data/jiangsu_museums.json"
    with open(output_file, "w", encoding="utf-8") as f:
        json.dump(data, f, ensure_ascii=False, indent=2)
    
    print(f"\n✅ 数据已保存：{output_file}")
    
    # 分城市统计
    city_stats = {}
    for museum in data["level1"] + data["level2"]:
        city = museum["city"]
        if city not in city_stats:
            city_stats[city] = 0
        city_stats[city] += 1
    
    print("\n分城市统计（必去博物馆）：")
    print("-" * 60)
    sorted_cities = sorted(city_stats.items(), key=lambda x: x[1], reverse=True)
    for city, count in sorted_cities:
        print(f"{city}: {count} 家")
    
    print("\n" + "=" * 60)
    print("P1-2 任务完成！")
    print("=" * 60)


if __name__ == "__main__":
    main()
