#!/usr/bin/env python
# -*- coding:utf-8 -*-
import os
import json
import time
import logging
import requests
from threading import Timer

white = 255  # 颜色
black = 0
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(message)s')  # 设置日志级别

def get_area_id(city_name):
    """从city.js中检索AREAID，无限重试直到成功"""
    url = "https://j.i8tq.com/weather2020/search/city.js"
    while True:
        try:
            response = requests.get(url)
            response.raise_for_status()  # 检查请求是否成功
            city_data_text = response.text.strip().split('var city_data = ')[-1].rstrip(';')
            if city_data_text:
                city_data = json.loads(city_data_text)
                for province, cities in city_data.items():
                    for city, districts in cities.items():
                        for district, info in districts.items():
                            if info['NAMECN'] == city_name:
                                return info['AREAID']
                logging.error("城市名称 '%s' 在城市数据中未找到", city_name)
            else:
                logging.error("从city.js接收到的数据为空")
        except (requests.RequestException, json.JSONDecodeError) as e:
            logging.error("检索或解析城市数据时发生错误: %s", e)
            time.sleep(180)  # 重试前等待

def get_current_city():
    """获取当前城市名称，无限重试直到成功"""
    url = "http://ip-api.com/json/?lang=zh-CN"
    while True:
        try:
            response = requests.get(url)
            response.raise_for_status()  # 检查请求是否成功
            data = response.json()
            if data['status'] == 'success':
                return data['city']
            else:
                logging.error("获取当前城市失败: %s", data['message'])
        except (requests.RequestException, json.JSONDecodeError) as e:
            logging.error("检索或解析当前城市时出现错误: %s", e)
            time.sleep(180)  # 重试前等待

def schedule_getWeath():
    """调度 getWeath 函数每3分钟执行一次"""
    getWeath()
    Timer(180, schedule_getWeath).start()  # 重新设置定时器

def getWeath(city='101060101'):
    headers = {
        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36',
        'Referer': 'https://www.weather.com.cn/'
    }
    current_city = get_current_city()
    if current_city:
        area_id = get_area_id(current_city)
        if area_id:
            city = area_id
    try:
        response = requests.get(f'https://d1.weather.com.cn/sk_2d/{city}.html', headers=headers)
        response.raise_for_status()  # 检查请求是否成功
        response.encoding = 'utf-8'
        weather_data = response.text[11:]
        with open('/root/2.13-Ink-screen-clock/bin/weather.json', 'w') as fileHandle:
            fileHandle.write(weather_data)
        print("天气文件更新")
    except requests.RequestException as e:
        logging.error("获取天气数据时出现网络错误: %s", e)
    except Exception as e:
        logging.error("发生了意外错误: %s", e)

try:
    schedule_getWeath()  # 开始调度天气获取函数
except KeyboardInterrupt:
    logging.info("检测到键盘中断，正在退出")
except Exception as e:
    logging.error("发生了意外错误: %s", e)
    exit()

# 脚本正常结束后的清理操作
exit()