diff --git a/QWeather.py b/QWeather.py new file mode 100644 index 0000000..318a954 --- /dev/null +++ b/QWeather.py @@ -0,0 +1,729 @@ +#!/usr/bin/env python +# -- coding:utf-8 -- +# @Author: markushammered@gmail.com +# @Development Tool: PyCharm +# @Create Time: 2021/10/23 +# @File Name: QWeather.py + +import sys +import json +import time +import pandas +import smtplib +import getpass +import requests +import argparse +import multiprocessing +import logging.handlers + +from ruamel.yaml import YAML +from email.header import Header +from email.mime.text import MIMEText +from colorlog import ColoredFormatter +from email.mime.image import MIMEImage +from email.mime.multipart import MIMEMultipart + + +class SendWeatherMail: + def __init__(self): + with open(config_name, 'r', encoding='utf-8') as config_f: + self.config = YAML().load(config_f) + with open('resource/type_warning.json', 'r', encoding='utf-8') as type_f: + self.type_name = json.loads(type_f.read()) + + self.location = self.config['request-settings']['location'] # 城市ID + self.key = self.config['request-settings']['key'] # API密钥 + self.unit = self.config['request-settings']['unit'] # 度量单位 + self.lang = self.config['request-settings']['lang'] # 语言 + self.mode = self.config['request-settings']['mode'] # 发送模式 --仅作判断标识 + self.receiver = self.config['request-settings']['receiver'] # 接收者邮箱 --> 列表 [无论有几个人都必须是列表] + self.sender = self.config['mail-settings']['sender'] # 发送者邮箱 + self.password = self.config['mail-settings']['password'] # 服务器登录密码 + self.server = self.config['mail-settings']['server'] # 邮箱服务器 + self.port = self.config['mail-settings']['port'] # 邮箱端口号 + self.icon_style = self.config['client-settings']['icon-style'] # 天气图标 + self.enableSSL = self.config['client-settings'] # 是否使用SSL连接到邮箱服务器 + self.city_name = self.config['only-view-settings']['city-name'] # 城市名称仅作邮件内容 + + self.style_list = ['style1', 'style2', 'style3'] + + # 如果配置文件内填错或不填 默认选择第一套图标(style1) + if self.icon_style not in self.style_list: + self.icon_style = 'style1' + + self.dev_url = 'https://devapi.qweather.com/v7/weather/7d' # 开发版本天气信息url + self.free_url = 'https://devapi.qweather.com/v7/weather/3d' # 免费版本天气信息url + self.warning_url = 'https://devapi.qweather.com/v7/warning/now' # 自然灾害预警信息url + + self.headers = {'Accept-Encoding': 'gzip, deflate'} # 请求头 + + self.params = {'location': self.location, + 'key': self.key, + 'unit': self.unit, + 'lang': self.lang} # GET方式请求参数 + + self.message = MIMEMultipart('related') + self.message['From'] = Header('HeWeatherReporter') # 发件人名称 + self.message['To'] = Header('All allowed User') # 收件人显示名称 + self.msg_content = MIMEMultipart('alternative') # 文字目录 + + if self.enableSSL: + self.smtp = smtplib.SMTP_SSL(self.server, self.port) # 登录服务器 使用SSL连接 + else: + self.smtp = smtplib.SMTP(self.server, self.port) # 登录邮箱服务器 不使用SSL连接 + + def dev_mode(self): + """ + 开发版本发送邮件 + [全七天] | date: 日期 (1-7) + [全七天] | day_weather: 白天天气 (1-7) + [全七天] | night_weather: 晚上天气 (1-7) + [全七天] | temperature_max: 最高气温 (1-7) + [全七天] | temperature_min: 最低气温 (1-7) + [全七天] | icon_list: 白天和晚上的天气图标 (1-14) + [第一天] | sunrise: 日出时间 (1) + [第一天] | sunset: 日落时间 (1) + [第一天] | humidity: 空气湿度(相对) (1) + [第一天] | wind_speed: 风速 (1) + [第一天] | wind_scale: 风级 (1) + [第一天] | wind_dir: 风向 (1) + [第一天] | ultraviolet_rays: 紫外线强度指数 (1) + [第一天] | cloud: 云量(相对) (1) + [第一天] | pressure: 气压 (1) + [第一天] | vis: 能见度 (1) + :return: None + """ + r_day = session.get(self.dev_url, headers=self.headers, params=self.params) + + weather_day_text = r_day.json() # 使用json加载数据 + logger.info(f'{language["request_result_weather"]}:{r_day}') + + # 1-7天数据 + day_1 = weather_day_text['daily'][0] + day_2 = weather_day_text['daily'][1] + day_3 = weather_day_text['daily'][2] + day_4 = weather_day_text['daily'][3] + day_5 = weather_day_text['daily'][4] + day_6 = weather_day_text['daily'][5] + day_7 = weather_day_text['daily'][6] + + date = [day_1['fxDate'][5:], day_2['fxDate'][5:], day_3['fxDate'][5:], day_4['fxDate'][5:], day_5['fxDate'][5:], + day_6['fxDate'][5:], + day_7['fxDate'][5:]] + day_weather = [day_1['textDay'], day_2['textDay'], day_3['textDay'], day_4['textDay'], day_5['textDay'], + day_6['textDay'], day_7['textDay']] + night_weather = [day_1['textNight'], day_2['textNight'], day_3['textNight'], day_4['textNight'], + day_5['textNight'], day_6['textNight'], day_7['textNight']] + temperature_max = [day_1['tempMax'], day_2['tempMax'], day_3['tempMax'], day_4['tempMax'], day_5['tempMax'], + day_6['tempMax'], day_7['tempMax']] + temperature_min = [day_1['tempMin'], day_2['tempMin'], day_3['tempMin'], day_4['tempMin'], day_5['tempMin'], + day_6['tempMin'], day_7['tempMin']] + icon_list = [day_1['iconDay'], day_2['iconDay'], day_3['iconDay'], day_4['iconDay'], day_5['iconDay'], + day_6['iconDay'], day_7['iconDay'], day_1['iconNight'], day_2['iconNight'], day_3['iconNight'], + day_4['iconNight'], day_5['iconNight'], day_6['iconNight'], day_7['iconNight']] + + sunset = day_1['sunset'] + sunrise = day_1['sunrise'] + humidity = day_1['humidity'] + wind_speed = day_1['windSpeedDay'] + wind_scale = day_1['windScaleDay'] + wind_dir = day_1['windDirDay'] + ultraviolet_rays = day_1['uvIndex'] + cloud = day_1['cloud'] + pressure = day_1['pressure'] + vis = day_1['vis'] + + # 邮件内容主体 + mail_html = f""" +

+ + + {language['area']}:{self.city_name} +
+ {language['sender']}:{self.sender} + + +

+
+ + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +
| {language['date']}  | {language['weather']}  | {language['lowestTemp']}  | {language['highestTemp']}  
{language['today']}{day_weather[0]}/{night_weather[0]} {temperature_min[0]}℃{temperature_max[0]}℃
{date[1]}{day_weather[1]}/{night_weather[1]} {temperature_min[1]}℃{temperature_max[1]}℃
{date[2]}{day_weather[2]}/{night_weather[2]} {temperature_min[2]}℃{temperature_max[2]}℃
{date[3]}{day_weather[3]}/{night_weather[3]} {temperature_min[3]}℃{temperature_max[3]}℃
{date[4]}{day_weather[4]}/{night_weather[4]} {temperature_min[4]}℃{temperature_max[4]}℃
{date[5]}{day_weather[5]}/{night_weather[5]} {temperature_min[5]}℃{temperature_max[5]}℃
{date[6]}{day_weather[6]}/{night_weather[6]} {temperature_min[6]}℃{temperature_max[6]}℃
+ + + + + + + + + + + + + + + + + + + + + + + + + + + +
 
{language['wind_info']}{language['humidity']}{language['uv_info']}
{wind_speed}m/s  {wind_scale}  {wind_dir} {humidity}% {ultraviolet_rays} 
 
{language['vis']}{language['pressure']}{language['cloud']}
{vis}km {pressure}hPa {cloud}% 
+ + + + + + + + + + + + + + +
 
Sunrise Sunset
 {sunrise} --------------- {sunset} 
+
+
+
+
+ + + QWeather +  ·  + Github Repo + + +
""" + self.message['Subject'] = language['subject_7'] # 邮件标题 + self.message.attach(MIMEText(mail_html, 'html', 'utf-8')) + + # 循环将图片attach到html里 + image_count = 1 + for image_resource in icon_list: + with open(f'./resource/{self.icon_style}/{image_resource}.png', 'rb') as fp: + MyImage = MIMEImage(fp.read()) + MyImage.add_header('Content-ID', f'img{image_count}') + self.message.attach(MyImage) + image_count += 1 + + with open('resource/basic-resources/sunrise.png', 'rb') as sr_f: + sunrise_img = MIMEImage(sr_f.read()) + sunrise_img.add_header('Content-ID', 'sunrise') + self.message.attach(sunrise_img) + with open('resource/basic-resources/sunset.png', 'rb') as ss_f: + sunset_img = MIMEImage(ss_f.read()) + sunset_img.add_header('Content-ID', 'sunset') + self.message.attach(sunset_img) + + try: + self.smtp.login(self.sender, self.password) # 登录 + self.smtp.sendmail(self.sender, self.receiver, self.message.as_string()) # 发送 + except smtplib.SMTPException as e: + logger.critical(f'{language["mail_error"]}:', e) + sys.exit(1) + + def free_mode(self): + """ + 免费版本发送邮件 + :return: + """ + f_request = session.get(self.free_url, headers=self.headers, params=self.params) + f_weather = f_request.json() + + logger.info(f'{language["request_result_weather"]}:{f_request}') + + day_1 = f_weather['daily'][0] + day_2 = f_weather['daily'][1] + day_3 = f_weather['daily'][2] + + date = [day_1['fxDate'][5:], day_2['fxDate'][5:], day_2['fxDate'][5:]] + day_weather = [day_1['textDay'], day_2['textDay'], day_2['textDay']] + night_weather = [day_1['textNight'], day_2['textNight'], day_2['textNight']] + temperature_max = [day_1['tempMax'], day_2['tempMax'], day_3['tempMax']] + temperature_min = [day_1['tempMin'], day_2['tempMin'], day_2['tempMin']] + icon_list = [day_1['iconDay'], day_1['iconDay'], day_3['iconDay'], day_1['iconNight'], day_2['iconNight'], + day_3['iconNight']] + + sunset = day_1['sunset'] + sunrise = day_1['sunrise'] + humidity = day_1['humidity'] + wind_speed = day_1['windSpeedDay'] + wind_scale = day_1['windScaleDay'] + wind_dir = day_1['windDirDay'] + ultraviolet_rays = day_1['uvIndex'] + cloud = day_1['cloud'] + pressure = day_1['pressure'] + vis = day_1['vis'] + + mail_html = f""" +

+ + + {language['area']}:{self.city_name} +
+ {language['sender']}:{self.sender} + + +

+
+ + + + + + + + + + + + + + + + + + + + + + + +
| {language['date']}  | {language['weather']}  | {language['lowestTemp']}  | {language['highestTemp']}  
{language['today']}{day_weather[0]}/{night_weather[0]} {temperature_min[0]}℃{temperature_max[0]}℃
{date[1]}{day_weather[1]}/{night_weather[1]} {temperature_min[1]}℃{temperature_max[1]}℃
{date[2]}{day_weather[2]}/{night_weather[2]} {temperature_min[2]}℃{temperature_max[2]}℃
+ + + + + + + + + + + + + + + + + + + + + + + + + + + +
 
{language['wind_info']}{language['humidity']}{language['uv_info']}
{wind_speed}m/s  {wind_scale}  {wind_dir} {humidity}% {ultraviolet_rays} 
 
{language['vis']}{language['pressure']}{language['cloud']}
{vis}km {pressure}hPa {cloud}% 
+ + + + + + + + + + + + + + +
 
Sunrise Sunset
 {sunrise} --------------- {sunset} 
+
+
+
+
+ + + QWeather +  ·  + Github Repo + + +
""" + self.message['Subject'] = language['subject_3'] + self.message.attach(MIMEText(mail_html, 'html', 'utf-8')) + + image_count = 1 + for image_source_free in icon_list: + with open(f'./resource/{self.icon_style}/{image_source_free}.png', 'rb') as file: + MyImage = MIMEImage(file.read()) + MyImage.add_header('Content-ID', f'img{image_count}') + self.message.attach(MyImage) + image_count += 1 + + with open('resource/basic-resources/sunrise.png', 'rb') as sr_f: + sunrise_img = MIMEImage(sr_f.read()) + sunrise_img.add_header('Content-ID', 'sunrise') + self.message.attach(sunrise_img) + with open('resource/basic-resources/sunset.png', 'rb') as ss_f: + sunset_img = MIMEImage(ss_f.read()) + sunset_img.add_header('Content-ID', 'sunset') + self.message.attach(sunset_img) + + try: + self.smtp.login(self.sender, self.password) + self.smtp.sendmail(self.sender, self.receiver, self.message.as_string()) + except smtplib.SMTPException as e: + logger.critical(f'{language["mail_error"]}:', e) + sys.exit(1) + + def warning_send_mail(self): + """ + 发送自然灾害预警信息 + releaseTime: 更新时间(并不是获取数据的时间 而是API更新数据的时间) + title: 标题 + startTime: 开始时间 + endTime: 结束时间 + status: 状态 + level: 等级 + type: 类型 + text: 详细描述 + 获取自然灾害并判断灾害信息是否为空 + 如果空则跳过 + 如果不为空则单独发送一封邮件 + :return: + """ + r = session.get(self.warning_url, headers=self.headers, params=self.params) + data = r.json() + + logger.info(f'{language["request_result_warning"]}:{data["code"]}') + if data['warning']: + public_time = data['warning'][0]['pubTime'] + title = data['warning'][0]['title'] + start_time = data['warning'][0]['startTime'] + end_time = data['warning'][0]['endTime'] + if not start_time: + start_time = None + elif not end_time: + end_time = None + status = data['warning'][0]['status'] + if status == 'update': + status = {language["new_warning"]} + logger.info(f'{language["new_warning"]}') + elif status == 'active': + status = language["warning_updated"] + logger.info(f'{language["warning_updated"]}') + elif status == 'cancel': + logger.info(f'{language["warning_canceled"]}') + + level = data['warning'][0]['level'] + type_ = data['warning'][0]['type'] + type_ = self.type_name[type_] # 将数字type转换为文字 + text = data['warning'][0]['text'] + self.message['Subject'] = language['subject_war'] + self.message['Subject'] = f'{title}' + mail_html = f""" + + + + + Warning + + +
+

{title}

+

{language['release_time']}:{public_time[:10]} {level}{language['early_warning']}

+

+ {language['warning_status']}:{status} {language['warning_type']}:{type_} {language['warning_duration']}:{start_time[:10]}-{end_time[:10]} +
+ {text} +

+
+
+
+
+
+ + + QWeather +  ·  + Github Repo + + +
+ + + """ + self.msg_content.attach(MIMEText(mail_html, 'html', 'utf-8')) + self.message.attach(self.msg_content) + try: + if status != 'cancel': + self.smtp.login(self.sender, self.password) + self.smtp.sendmail(self.sender, self.receiver, self.message.as_string()) + except smtplib.SMTPException as e: + logger.error(f'{language["mail_error"]}:', e) + sys.exit(1) + + +def loop_check(mode: str, time_list: list): + """ + 循环检测时间如果本地时间等于配置文件内填写的时间则发送一封天气信息的邮件 + :return: + """ + if mode == 'dev': + while True: + local_time = time.strftime("%H:%M", time.localtime()) + time.sleep(1) + if local_time in time_list: + SendWeatherMail().dev_mode() + logger.info(f'{language["mail_succeed"]}') + logger.info(f'{language["wait_seconds"]}') + time.sleep(61) + elif mode == 'free': + while True: + local_time = time.strftime("%H:%M", time.localtime()) + time.sleep(1) + if local_time in time_list: + SendWeatherMail().free_mode() + logger.info(f'{language["mail_succeed"]}') + logger.info(f'{language["wait_seconds"]}') + time.sleep(61) + + +def check_config(): # 检查各项配置是否完成填写 + for mail in config['mail-settings'].values(): + if mail is None: + logger.critical(f'"mail-settings"{language["config_not_filled"]}') + sys.exit(1) + for request in config['request-settings'].values(): + if request is None: + logger.critical(f'"request-settings"{language["config_not_filled"]}') + sys.exit(1) + for other in config['client-settings'].values(): + if other is None: + logger.critical(f'"client-settings"{language["config_not_filled"]}') + sys.exit(1) + + +def read_excel(kw: str): + """ + 读取xlsx城市ID文件并返回搜索到的结果 + :param kw: 用于搜索的关键字 + :return: city_list + """ + index_count = 0 + city_list = [] + logger.info(f'[Search]{language["reading_the_file"]}') + df = pandas.read_excel('./resource/China-City-List.xlsx') + pandas.set_option('max_rows', None) # 读取xlsx文件不折叠 + data_records = df.to_dict(orient='spl0it') + for i in data_records['data']: + if kw in str(i): + city = [index_count, i[0], i[2], i[4], i[6]] + index_count += 1 + city_list.append(city) + return city_list + + +def modify_config(status: bool = False): + """ + 修改配置文件 -> 检测配置文件内的location项是否填写 + 填写 -> 跳过 + 未填写 -> 触发 + (如果不小心在location项内随便输入了什么会导致误判) + :param status: 触发修改模式的条件 + :return: Nothing + """ + with open(config_name, 'r', encoding='utf-8') as _location_check: + _location = YAML().load(_location_check.read())['request-settings']['location'] + + if not _location or not status: # 如果配置中location未填写或status未False则触发条件 + logger.info(f'[Modify]{language["fill_the_config"]}') + logger.info(f'[Modify]{language["input_a_city_name"]}') + time.sleep(0.5) + city_name = input('-->') + logger.info(f'[Modify]{language["reading_the_file"]}') + searched_city = read_excel(city_name) + logger.info(f'[Modify]{language["user_input"]}:[{city_name}]') + for cities in searched_city: + logger.info(f'{cities[0]}-{cities[1]}-{cities[2]}-{cities[3]}-{cities[4]}') + logger.info(f'[Modify]{language["select_a_index"]}') + time.sleep(0.5) + while True: + try: + time.sleep(0.5) + user_input = input('-->') + if user_input == 'quit': + logger.info('User quit') + sys.exit(1) + index = searched_city[int(user_input)] + with open('./config.yml', 'r', encoding='utf-8') as of: + data = YAML().load(of) + data['request-settings']['location'] = index[1] + data['only-view-settings']['city-name'] = f'{index[2]}-{index[3]}-{index[4]}' + data['only-view-settings']['time'] = time.strftime("%a %b %d %Y %H:%M:%S", time.localtime()) + data['only-view-settings']['user'] = getpass.getuser() + with open('./config.yml', 'w', encoding='utf-8') as wf: + YAML().dump(data, wf) + logger.info(f'[Write]{language["write_successfully"]}') + break + except (IndexError, ValueError) as e: + logger.info(e) + logger.error(f'[Write]{language["input_type_error"]}') + continue + finally: + if not status: + logger.info('Program has done.') + sys.exit(0) + + +if __name__ == '__main__': + config_name = 'config.yml' # 配置文件名称 -> 用于开发时快速调试 + + date_format = '%H:%M:%S' + info_format_console = '%(log_color)s[%(asctime)s] |%(levelname)-8s |%(lineno)-3s |%(message)s' + info_format_file = '[%(asctime)s] |%(levelname)-8s |%(lineno)-3s |%(funcName)s |%(pathname)s |%(message)s' + formatter = ColoredFormatter(fmt=info_format_console, + datefmt=date_format, + reset=True, + log_colors={ + 'DEBUG': 'light_purple', + 'INFO': 'light_cyan', + 'WARNING': 'yellow', + 'ERROR': 'red', + 'CRITICAL': 'red,bold_red'}) + formatter_file = logging.Formatter(fmt=info_format_file, + datefmt=date_format) + + logger = logging.getLogger('MainLogger') + logger.setLevel(logging.DEBUG) + ConsoleLogger = logging.StreamHandler() # 输出到终端 + ConsoleLogger.setFormatter(formatter) + log_name = time.strftime('%Y-%m-%d-%H') # 一小时内使用的日志文件都是同一个 + FileLogger = logging.handlers.RotatingFileHandler(filename=f'./logs/{log_name}.log', + maxBytes=102400, + backupCount=5) # 每个日志文件最大102400字节(≈100kb) + FileLogger.setFormatter(formatter_file) + logger.addHandler(ConsoleLogger) + logger.addHandler(FileLogger) + + # 使用本地网络进行请求 + session = requests.Session() + session.trust_env = False + + # 获取语言配置 + with open(config_name, 'r', encoding='utf-8') as lang: + config = YAML().load(lang.read()) + language_sel = config['client-settings']['language'] + if language_sel not in ['zh_cn', 'en_us']: + language_sel = 'zh_cn' + + # 打开语言json文件 + with open(f'./resource/lang/{language_sel}.json', 'r', encoding='utf-8') as lang_f: + language = json.loads(lang_f.read()) + + modify_config(True) + + # 获取参数 + parser = argparse.ArgumentParser() + arg_keywords = ['free', 'dev', 'warning', 'modify'] + parser.add_argument('-t', + '--test', + help='Some operations for test.', + choices=arg_keywords) + arg_test = parser.parse_args().test + if arg_test: + if arg_test == 'dev': + SendWeatherMail().dev_mode() + logger.debug(f'{language["debug_done"]}') + sys.exit(0) + elif arg_test == 'free': + SendWeatherMail().free_mode() + logger.debug(f'{language["debug_done"]}') + sys.exit(0) + elif arg_test == 'warning': + SendWeatherMail().warning_send_mail() + logger.debug(f'{language["debug_done"]}') + sys.exit(0) + elif arg_test == 'modify': + modify_config() + logger.info(f'{language["debug_done"]}') + + # 检查配置文件是否填写完成 + check_config() + + # 输出声明 + logger.info(f'{language["statement_1"]}') + logger.info(f'{language["statement_2"]}') + logger.info(f'{language["statement_3"]}') + logger.info(f'{language["statement_4"]}\n') + + _times = config['client-settings']['send-times'] + _mode = config['request-settings']['mode'] + + # 另开一个进程与主进程同时运行 --> 运行loopCheck --> 循环检查本地时间是否与配置内时间相符 + multiprocessing.Process(target=loop_check, args=(_mode, _times,)).run() + + # 循环检测时间 --> 每10分钟检查一次, 如果有则发送如果无则直接跳过 + loop_timer = 0 + while True: + time.sleep(1) + loop_timer += 1 + if loop_timer == 600: + SendWeatherMail().warning_send_mail() + loop_timer = 0