Automation/NoticeElectricity.py at main · cheanus/Automation · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
import re
import json
import smtplib
import requests
from email.mime.text import MIMEText
from config import *


def email_msg(title: str, content: str) -> None:
    # 设置email信息
    # 邮件内容设置
    message = MIMEText(content, "plain", "utf-8")
    # 邮件主题'
    message["Subject"] = title
    # 发送方信息
    message["From"] = sender
    # 接受方信息
    message["To"] = receivers[0]

    # 初始化邮箱服务器
    smtp_obj = smtplib.SMTP_SSL(mail_host, 465)
    # 登录到服务器
    smtp_obj.login(mail_user, mail_pass)
    # 发送
    smtp_obj.sendmail(sender, receivers, message.as_string())
    # 退出
    smtp_obj.quit()


URL = "https://yktapp.nwpu.edu.cn/jfdt/charge/feeitem/getThirdData"
headers = {
    "Host": "yktapp.nwpu.edu.cn",
    "Connection": "keep-alive",
    "Content-Length": "102",
    "Accept": "application/json, text/plain, */*",
    "User-Agent": "Mozilla/5.0 (Linux; Android 12; ANG-AN00 Build/HUAWEIANG-AN00; wv) AppleWebKit/537.36 (KHTML, like Gecko) Version/4.0 Chrome/92.0.4515.105 Mobile Safari/537.36 SuperApp",
    "Origin": "https://yktapp.nwpu.edu.cn",
    "X-Requested-With": "com.lantu.MobileCampus.nwpu",
    "Sec-Fetch-Site": "same-origin",
    "Sec-Fetch-Mode": "cors",
    "Sec-Fetch-Dest": "empty",
    "Accept-Encoding": "gzip, deflate",
    "Accept-Language": "zh-CN,zh;q=0.9,en-US;q=0.8,en;q=0.7",
}

# 查找宿舍对应代码
campus_code = "22" if campus == "长安校区" else "1"
data = {"type": "select", "level": "1", "campus": campus_code, "feeitemid": "182"}
response = requests.post(URL, data=data, headers=headers)
building_code = -1
for building_item in json.loads(response.text)["map"]["data"]:
    if building_item["name"] == building:
        building_code = building_item["value"]
        break
data = {
    "type": "select",
    "level": "2",
    "campus": campus_code,
    "building": building_code,
    "feeitemid": "182",
}
response = requests.post(URL, data=data, headers=headers)
room_code = -1
for room_item in json.loads(response.text)["map"]["data"]:
    if room_item["name"] == room:
        room_code = room_item["value"]
        break
if building_code == -1 or room_code == -1:
    raise ValueError("未找到对应房间")

data = {
    "type": "IEC",
    "level": "3",
    "campus": campus_code,
    "building": building_code,
    "room": room_code,
    "feeitemid": "182",
}
response = requests.post(URL, data=data, headers=headers)
remain = re.search(r'当前剩余电量\\":\\"(.*?)\\', response.text).groups()[0]
# print('当前电费为：'+remain)
if float(remain) < threshold:
    email_msg("宿舍电费提醒", "当前电费为：" + remain + "，请尽快缴费")