@@ -1,743 +0,0 @@
#!/usr/bin/env python3
# -- coding:utf-8 --
# @Author: markushammered@gmail.com
# @Development Tool: PyCharm
# @Create Time: 2021/10/23
# @File Name: QWeather.py
import os
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 )
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 . enableSSL = self . config [ ' client-settings ' ] # 是否使用SSL连接到邮箱服务器
self . city_name = self . config [ ' only-view-settings ' ] [ ' city-name ' ] # 城市名称仅作邮件内容
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 ( ' QWeather ' ) # 发件人名称
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 """
<p style= " text-align: center " >
<i>
<b>
{ language [ ' area ' ] } : { self . city_name }
<br>
{ language [ ' sender ' ] } : { self . sender }
</b>
</i>
</p>
<br />
<table style= " border: 0; text-align: center; margin:0 auto " >
<tr>
<th>| { language [ ' date ' ] } </th>
<th>| { language [ ' weather ' ] } </th>
<th>| { language [ ' lowestTemp ' ] } </th>
<th>| { language [ ' highestTemp ' ] } </th>
</tr>
<tr>
<!--日期 天气 最低 最高/Date Weather LowestTemp HighestTemp-->
<td> { language [ ' today ' ] } </td>
<td> { day_weather [ 0 ] } <img src= " cid:img1 " width= " 20 " alt= " " >/ { night_weather [ 0 ] } <img src= " cid:img2 "
width= " 20 " alt= " " ></td> <td> { temperature_min [ 0 ] } ℃</td>
<td> { temperature_max [ 0 ] } ℃</td>
</tr>
<tr>
<td> { date [ 1 ] } </td>
<td> { day_weather [ 1 ] } <img src= " cid:img3 " width= " 20 " alt= " " >/ { night_weather [ 1 ] } <img src= " cid:img4 "
width= " 20 " alt= " " ></td> <td> { temperature_min [ 1 ] } ℃</td>
<td> { temperature_max [ 1 ] } ℃</td>
</tr>
<tr>
<td> { date [ 2 ] } </td>
<td> { day_weather [ 2 ] } <img src= " cid:img5 " width= " 20 " alt= " " >/ { night_weather [ 2 ] } <img src= " cid:img6 "
width= " 20 " alt= " " ></td> <td> { temperature_min [ 2 ] } ℃</td>
<td> { temperature_max [ 2 ] } ℃</td>
</tr>
<tr>
<td> { date [ 3 ] } </td>
<td> { day_weather [ 3 ] } <img src= " cid:img7 " width= " 20 " alt= " " >/ { night_weather [ 3 ] } <img src= " cid:img8 "
width= " 20 " alt= " " ></td> <td> { temperature_min [ 3 ] } ℃</td>
<td> { temperature_max [ 3 ] } ℃</td>
</tr>
<tr>
<td> { date [ 4 ] } </td>
<td> { day_weather [ 4 ] } <img src= " cid:img9 " width= " 20 " alt= " " >/ { night_weather [ 4 ] } <img src= " cid:img10 "
width= " 20 " alt= " " ></td> <td> { temperature_min [ 4 ] } ℃</td>
<td> { temperature_max [ 4 ] } ℃</td>
</tr>
<tr>
<td> { date [ 5 ] } </td>
<td> { day_weather [ 5 ] } <img src= " cid:img11 " width= " 20 " alt= " " >/ { night_weather [ 5 ] } <img src= " cid:img12 "
width= " 20 " alt= " " ></td> <td> { temperature_min [ 5 ] } ℃</td>
<td> { temperature_max [ 5 ] } ℃</td>
</tr>
<tr>
<td> { date [ 6 ] } </td>
<td> { day_weather [ 6 ] } <img src= " cid:img13 " width= " 20 " alt= " " >/ { night_weather [ 6 ] } <img src= " cid:img14 "
width= " 20 " alt= " " ></td> <td> { temperature_min [ 6 ] } ℃</td>
<td> { temperature_max [ 6 ] } ℃</td>
</tr>
</table>
<table style= " border: 0;text-align: center; margin:0 auto " >
<tr>
<th> </th>
</tr>
<tr>
<th> { language [ ' wind_info ' ] } </th>
<th> { language [ ' humidity ' ] } </th>
<th> { language [ ' uv_info ' ] } </th>
</tr>
<tr>
<td> { wind_speed } m/s { wind_scale } { wind_dir } </td>
<td> { humidity } % </td>
<td> { ultraviolet_rays } </td>
</tr>
<tr>
<th> </th>
</tr>
<tr>
<th> { language [ ' vis ' ] } </th>
<th> { language [ ' pressure ' ] } </th>
<th> { language [ ' cloud ' ] } </th>
</tr>
<tr>
<td> { vis } km </td>
<td> { pressure } hPa </td>
<td> { cloud } % </td>
</tr>
</table>
<table style= " border: 0;text-align: center; margin:0 auto " >
<tr>
<th> </th>
</tr>
<tr>
<td><img src= " cid:sunrise " alt= " Sunrise " width= " 30 " ></td>
<td> </td>
<td><img src= " cid:sunset " alt= " Sunset " width= " 30 " ></td>
</tr>
<tr>
<td> { sunrise } </td>
<td>---------------</td>
<td> { sunset } </td>
</tr>
</table>
<div style= " text-align: center; " id= " About " >
<br />
<br />
<br />
<i>
<b>
<a href= " https://dev.qweather.com/ " style= " color: black " target= " _blank " >QWeather</a>
<a style= " color: black " > · </a>
<a href= " https://github.com/MarkusJoe/QWeather " style= " color: black " target= " _blank " >Github Repo</a>
</b>
</i>
</div> """
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/icons/ { 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 """
<p style= " text-align: center " >
<i>
<b>
{ language [ ' area ' ] } : { self . city_name }
<br>
{ language [ ' sender ' ] } : { self . sender }
</b>
</i>
</p>
<br />
<table style= " border: 0; text-align: center; margin:0 auto " >
<tr>
<th>| { language [ ' date ' ] } </th>
<th>| { language [ ' weather ' ] } </th>
<th>| { language [ ' lowestTemp ' ] } </th>
<th>| { language [ ' highestTemp ' ] } </th>
</tr>
<tr>
<!--日期 天气 最低 最高/Date Weather LowestTemp HighestTemp-->
<td> { language [ ' today ' ] } </td>
<td> { day_weather [ 0 ] } <img src= " cid:img1 " width= " 20 " alt= " " >/ { night_weather [ 0 ] } <img src= " cid:img2 "
width= " 20 " alt= " " ></td> <td> { temperature_min [ 0 ] } ℃</td>
<td> { temperature_max [ 0 ] } ℃</td>
</tr>
<tr>
<td> { date [ 1 ] } </td>
<td> { day_weather [ 1 ] } <img src= " cid:img3 " width= " 20 " alt= " " >/ { night_weather [ 1 ] } <img src= " cid:img4 "
width= " 20 " alt= " " ></td> <td> { temperature_min [ 1 ] } ℃</td>
<td> { temperature_max [ 1 ] } ℃</td>
</tr>
<tr>
<td> { date [ 2 ] } </td>
<td> { day_weather [ 2 ] } <img src= " cid:img5 " width= " 20 " alt= " " >/ { night_weather [ 2 ] } <img src= " cid:img6 "
width= " 20 " alt= " " ></td> <td> { temperature_min [ 2 ] } ℃</td>
<td> { temperature_max [ 2 ] } ℃</td>
</tr>
</table>
<table style= " border: 0;text-align: center; margin:0 auto " >
<tr>
<th> </th>
</tr>
<tr>
<th> { language [ ' wind_info ' ] } </th>
<th> { language [ ' humidity ' ] } </th>
<th> { language [ ' uv_info ' ] } </th>
</tr>
<tr>
<td> { wind_speed } m/s { wind_scale } { wind_dir } </td>
<td> { humidity } % </td>
<td> { ultraviolet_rays } </td>
</tr>
<tr>
<th> </th>
</tr>
<tr>
<th> { language [ ' vis ' ] } </th>
<th> { language [ ' pressure ' ] } </th>
<th> { language [ ' cloud ' ] } </th>
</tr>
<tr>
<td> { vis } km </td>
<td> { pressure } hPa </td>
<td> { cloud } % </td>
</tr>
</table>
<table style= " border: 0;text-align: center; margin:0 auto " >
<tr>
<th> </th>
</tr>
<tr>
<td><img src= " cid:sunrise " alt= " Sunrise " width= " 30 " ></td>
<td> </td>
<td><img src= " cid:sunset " alt= " Sunset " width= " 30 " ></td>
</tr>
<tr>
<td> { sunrise } </td>
<td>---------------</td>
<td> { sunset } </td>
</tr>
</table>
<div style= " text-align: center; " id= " About " >
<br />
<br />
<br />
<i>
<b>
<a href= " https://dev.qweather.com/ " style= " color: black " target= " _blank " >QWeather</a>
<a style= " color: black " > · </a>
<a href= " https://github.com/MarkusJoe/QWeather " style= " color: black " target= " _blank " >Github Repo</a>
</b>
</i>
</div> """
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/icons/ { 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 ) . json ( )
logger . info ( f ' { language [ " request_result_warning " ] } : { r [ " code " ] } ' )
if r [ ' warning ' ] :
public_time = r [ ' warning ' ] [ 0 ] [ ' pubTime ' ]
title = r [ ' warning ' ] [ 0 ] [ ' title ' ]
start_time = r [ ' warning ' ] [ 0 ] [ ' startTime ' ]
end_time = r [ ' warning ' ] [ 0 ] [ ' endTime ' ]
if not start_time :
start_time = None
elif not end_time :
end_time = None
warning_status = r [ ' warning ' ] [ 0 ] [ ' status ' ]
if warning_status == ' update ' :
warning_status = { language [ " new_warning " ] }
logger . info ( f ' { language [ " new_warning " ] } ' )
elif warning_status == ' active ' :
warning_status = language [ " warning_updated " ]
logger . info ( f ' { language [ " warning_updated " ] } ' )
elif warning_status == ' cancel ' :
logger . info ( f ' { language [ " warning_canceled " ] } ' )
level = r [ ' warning ' ] [ 0 ] [ ' level ' ]
type_ = r [ ' warning ' ] [ 0 ] [ ' typeName ' ]
text = r [ ' warning ' ] [ 0 ] [ ' text ' ]
self . message [ ' Subject ' ] = language [ ' subject_war ' ]
self . message [ ' Subject ' ] = f ' { title } '
mail_html = f """
<!DOCTYPE html>
<html lang= " zh " >
<head>
<meta charset= " UTF-8 " >
<title>Warning</title>
</head>
<body>
<div style= " text-align: center;color: black; " >
<h2> { title } </h2>
<h3> { language [ ' release_time ' ] } : { public_time [ : 10 ] } { level } { language [ ' early_warning ' ] } </h3>
<p>
{ language [ ' warning_status ' ] } : { warning_status } { language [ ' warning_type ' ] } : { type_ } { language [ ' warning_duration ' ] } : { start_time [ : 10 ] } ~ { end_time [ : 10 ] }
<br />
{ text }
</p>
</div>
<div style= " text-align: center; " id= " About " >
<br />
<br />
<br />
<i>
<b>
<a href= " https://dev.qweather.com/ " style= " color: black " target= " _blank " >QWeather</a>
<a style= " color: black " > · </a>
<a href= " https://github.com/MarkusJoe/QWeather " style= " color: black " target= " _blank " >Github Repo</a>
</b>
</i>
</div>
</body>
</html>
"""
self . msg_content . attach ( MIMEText ( mail_html , ' html ' , ' utf-8 ' ) )
self . message . attach ( self . msg_content )
try :
if warning_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:
"""
match mode :
case ' 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 )
case ' 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 ( ) :
"""
检查各项配置是否填写完成
:return:
"""
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 ( f ' ./resource/ { LOCATION_ID_FILE_NAME } ' )
pandas . set_option ( ' max_rows ' , None ) # 读取xlsx文件不折叠
data_records = df . to_dict ( orient = ' split ' )
for i in data_records [ ' data ' ] :
if kw in str ( i ) :
logger . info ( f ' { index_count } | { i [ 2 ] } - { i [ 4 ] } - { i [ 6 ] } ' )
city = [ index_count , i [ 0 ] , i [ 1 ] , i [ 2 ] , i [ 3 ] , i [ 4 ] , i [ 5 ] , i [ 6 ] , i [ 7 ] , i [ 8 ] , i [ 9 ] ]
index_count + = 1
city_list . append ( city )
return city_list
def modify_config ( mode : bool = False ) :
"""
修改配置文件 -> 检测配置文件内的location项是否填写
填写 -> 跳过
未填写 -> 触发
(如果不小心在location项内随便输入了什么会导致误判)
:param mode: 触发修改模式的条件
:return: Nothing
"""
if not _LOCATION or not mode : # 如果配置中location未填写或status未False则触发条件
logger . info ( f ' [Modify] { language [ " fill_the_config " ] } ' )
logger . info ( f ' [Modify] { language [ " input_a_city_name " ] } ' )
while True :
time . sleep ( 0.3 )
city_name = input ( ' --> ' )
match city_name :
case ' q ' :
logger . info ( f ' [Modify]User quit. ' )
sys . exit ( 0 )
case ' ' :
logger . critical ( f ' [Modify] { language [ " null_value " ] } ' )
continue
case _ :
break
searched_city = read_excel ( city_name )
logger . info ( f ' [Modify] { language [ " user_input " ] } :[ { city_name } ] ' )
logger . info ( f ' [Modify] { language [ " select_a_index " ] } ' )
if not searched_city :
logger . error ( f ' [Modify] { language [ " no_result " ] } ' )
sys . exit ( 1 )
time . sleep ( 0.3 )
while True :
try :
time . sleep ( 0.3 )
user_input = input ( ' --> ' )
if user_input == ' q ' :
logger . info ( ' [Quit]User quit ' )
sys . exit ( 1 )
index = searched_city [ int ( user_input ) ]
with open ( CONFIG_NAME , ' r ' , encoding = ' utf-8 ' ) as of :
data = YAML ( ) . load ( of )
data [ ' request-settings ' ] [ ' location ' ] = index [ 1 ]
data [ ' only-view-settings ' ] [ ' city-name ' ] = f ' { index [ 3 ] } - { index [ 7 ] } - { index [ 7 ] } '
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_NAME , ' w ' , encoding = ' utf-8 ' ) as wf :
YAML ( ) . dump ( data , wf )
logger . info ( f ' [Write] { language [ " write_successfully " ] } : { CONFIG_NAME } ' )
break
except ( IndexError , ValueError ) as e :
logger . info ( e )
logger . error ( f ' [Write] { language [ " input_type_error " ] } ' )
continue
finally :
logger . info ( ' [Done]Program has done. ' )
sys . exit ( 0 )
if __name__ == ' __main__ ' :
CONFIG_NAME = ' config.yml ' # 配置文件名称
LOCATION_ID_FILE_NAME = ' china_city_list.xlsx ' # 城市id文件必须为xlsx文件
# logs为空文件夹git上传时会自动忽略此文件夹，
if not os . path . isdir ( ' ./logs ' ) :
os . mkdir ( ' ./logs ' )
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 ( ) )
_TIMES = config [ ' client-settings ' ] [ ' send-times ' ]
_MODE = config [ ' request-settings ' ] [ ' mode ' ]
_LOCATION = config [ ' request-settings ' ] [ ' location ' ]
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 )
startup_arg = parser . parse_args ( ) . test
# Python3.10 更新的match-case 语法
match startup_arg :
case ' free ' :
SendWeatherMail ( ) . free_mode ( )
logger . debug ( f ' { language [ " debug_done " ] } ' )
sys . exit ( 0 )
case ' dev ' :
SendWeatherMail ( ) . dev_mode ( )
logger . debug ( f ' { language [ " debug_done " ] } ' )
sys . exit ( 0 )
case ' warning ' :
SendWeatherMail ( ) . warning_send_mail ( )
logger . debug ( f ' { language [ " debug_done " ] } ' )
sys . exit ( 0 )
case ' modify ' :
modify_config ( )
logger . info ( f ' { language [ " debug_done " ] } ' )
case _ :
pass
# 检查配置文件是否填写完成
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 " ] } ' )
logger . info ( f ' { language [ " current_profile " ] } : { CONFIG_NAME } ' )
# 另开一个进程与主进程同时运行 --> 运行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
