import datetime import sqlite3 import json from loguru import logger from conf import base_directory from file import config_center path = f'{base_directory}/config/data/xiaomi_weather.db' api_config = json.load(open(f'{base_directory}/config/data/weather_api.json', encoding='utf-8')) def update_path(): global path path = (f"{base_directory}/config/" f"data/{api_config['weather_api_parameters'][config_center.read_conf('Weather', 'api')]['database']}") def search_by_name(search_term): update_path() conn = sqlite3.connect(path) cursor = conn.cursor() cursor.execute('SELECT * FROM citys WHERE name LIKE ?', ('%' + search_term + '%',)) # 模糊查询 cities_results = cursor.fetchall() conn.close() result_list = [] for city in cities_results: result_list.append(city[2]) # 返回两个表的搜索结果 return result_list def search_code_by_name(search_term): if search_term == ('', ''): return 101010100 update_path() conn = sqlite3.connect(path) cursor = conn.cursor() logger.info(f"Searching for city: {search_term}") search_term = (search_term[0].replace('市',''), search_term[1].replace('区','')) cursor.execute('SELECT * FROM citys WHERE name = ?', (f"{search_term[0]}.{search_term[1]}",)) exact_results = cursor.fetchall() if not exact_results: search_term = search_term[0] cursor.execute('SELECT * FROM citys WHERE name LIKE ?', ('%' + f"{search_term}" + '%',)) cities_results = cursor.fetchall() else: cities_results = exact_results conn.close() if cities_results: # 多结果优先完全匹配,否则返回第一个 for city in cities_results: if city[2] == search_term or city[2] == search_term + '市' or city[2] + '市' == search_term: logger.debug(f"找到城市: {city[2]}，代码: {city[3]}") return city[3] result = cities_results[0][3] logger.debug(f"模糊找到城市: {cities_results[0][2]}，代码: {result}") else: result = "101010100" # 默认城市代码 logger.warning(f'未找到城市: {search_term}，使用默认城市代码') return result def search_by_num(search_term): update_path() conn = sqlite3.connect(path) cursor = conn.cursor() cursor.execute('SELECT * FROM citys WHERE city_num LIKE ?', ('%' + search_term + '%',)) # 模糊查询 cities_results = cursor.fetchall() conn.close() if cities_results: result = cities_results[0][2] else: result = '北京' # 默认城市 # 返回两个表的搜索结果 return result def get_weather_by_code(code): # 用代码获取天气描述 weather_status = json.load( open(f"{base_directory}/config/data/{config_center.read_conf('Weather', 'api')}_status.json", encoding="utf-8")) for weather in weather_status['weatherinfo']: if str(weather['code']) == code: return weather['wea'] return '未知' def get_weather_icon_by_code(code): # 用代码获取天气图标 weather_status = json.load( open(f"{base_directory}/config/data/{config_center.read_conf('Weather', 'api')}_status.json", encoding="utf-8") ) weather_code = None current_time = datetime.datetime.now() # 遍历获取天气代码 for weather in weather_status['weatherinfo']: if str(weather['code']) == code: original_code = weather.get('original_code') if original_code is not None: weather_code = str(weather['original_code']) else: weather_code = str(weather['code']) break if not weather_code: logger.error(f'未找到天气代码 {code}') return f'{base_directory}/img/weather/99.svg' # 根据天气和时间获取天气图标 if weather_code in ('0', '1', '3', '13'): # 晴、多云、阵雨、阵雪 if current_time.hour < 6 or current_time.hour >= 18: # 如果是夜间 return f'{base_directory}/img/weather/{weather_code}d.svg' return f'{base_directory}/img/weather/{weather_code}.svg' def get_weather_stylesheet(code): # 天气背景样式 current_time = datetime.datetime.now() weather_status = json.load( open(f"{base_directory}/config/data/{config_center.read_conf('Weather', 'api')}_status.json", encoding="utf-8")) weather_code = '99' for weather in weather_status['weatherinfo']: if str(weather['code']) == code: original_code = weather.get('original_code') if original_code is not None: weather_code = str(weather['original_code']) else: weather_code = str(weather['code']) break if weather_code in ('0', '1', '3', '99', '900'): # 晴、多云、阵雨、未知 if 6 <= current_time.hour < 18: # 如果是日间 # return 'spread:pad, x1:0, y1:0, x2:1, y2:1, stop:0 rgba(40, 60, 110, 255), stop:1 rgba(75, 175, 245, 255)' return 'img/weather/bkg/day.png' else: # 如果是夜间 return 'img/weather/bkg/night.png' # return 'spread:pad, x1:0, y1:0, x2:1, y2:1, stop:0 rgba(20, 60, 90, 255), stop:1 rgba(10, 20, 29, 255)' return 'img/weather/bkg/rain.png' def get_weather_url(): if config_center.read_conf('Weather', 'api') in api_config['weather_api_list']: return api_config['weather_api'][config_center.read_conf('Weather', 'api')] else: return api_config['weather_api']['xiaomi_weather'] def get_weather_alert_url(): if not api_config['weather_api_parameters'][config_center.read_conf('Weather', 'api')]['alerts']: return 'NotSupported' if config_center.read_conf('Weather', 'api') in api_config['weather_api_list']: return api_config['weather_api_parameters'][config_center.read_conf('Weather', 'api')]['alerts']['url'] else: return api_config['weather_api_parameters']['xiaomi_weather']['alerts']['url'] def get_weather_code_by_description(value): weather_status = json.load( open(f"{base_directory}/config/data/{config_center.read_conf('Weather', 'api')}_status.json", encoding="utf-8")) for weather in weather_status['weatherinfo']: if str(weather['wea']) == value: return str(weather['code']) return '99' def get_alert_image(alert_type): alerts_list = api_config['weather_api_parameters'][config_center.read_conf('Weather', 'api')]['alerts']['types'] return f'{base_directory}/img/weather/alerts/{alerts_list[alert_type]}' def is_supported_alert(): if not api_config['weather_api_parameters'][config_center.read_conf('Weather', 'api')]['alerts']: return False return True def get_weather_data(key='temp', weather_data=None): # 获取天气数据 if weather_data is None: logger.error('weather_data is None!') return None ''' 根据key值获取weather_data中的对应值 key值可以为：temp、icon、alert_title ''' # 各个天气api的可访问值 api_parameters = api_config['weather_api_parameters'][config_center.read_conf('Weather', 'api')] if key == 'alert': parameter = api_parameters['alerts']['type'].split('.') elif key == 'alert_title': if 'alerts' not in api_parameters or 'title' not in api_parameters['alerts']: return None parameter = api_parameters['alerts']['title'].split('.') else: parameter = api_parameters[key].split('.') # 遍历获取值 value = weather_data if config_center.read_conf('Weather', 'api') == 'amap_weather': value = weather_data['lives'][0][api_parameters[key]] elif config_center.read_conf('Weather', 'api') == 'qq_weather': value = str(weather_data['result']['realtime'][0]['infos'][api_parameters[key]]) else: for parameter in parameter: if not value: logger.warning(f'天气信息值{key}为空') return None if parameter == '0': value = value[0] continue if parameter in value: value = value[parameter] else: logger.error(f'获取天气参数失败，{parameter}不存在于{config_center.read_conf("Weather", "api")}中') return '错误' if key == 'temp': value += '°' elif key == 'icon': # 修复此代码影响其他天气源的问题 if api_parameters['return_desc']: # 如果此api返回的是天气描述而不是代码 value = get_weather_code_by_description(value) return value if __name__ == '__main__': # 测试代码 try: num_results = search_by_num('101310101') # [2]城市名称 print(num_results) cities_results_ = search_by_name('上海') # [3]城市代码 print(cities_results_) cities_results_ = search_code_by_name('上海','') # [3]城市代码 print(cities_results_) get_weather_by_code(3) except Exception as e: print(e)