forked from ChenXuzheng/clock-in-dockerfile
382 lines
16 KiB
Python
382 lines
16 KiB
Python
import requests
|
||
import re
|
||
import json
|
||
import datetime
|
||
import time
|
||
import os
|
||
import random
|
||
import schedule
|
||
import sys
|
||
import ddddocr
|
||
|
||
# 签到程序模块
|
||
|
||
|
||
class LoginError(Exception):
|
||
"""Login Exception"""
|
||
pass
|
||
|
||
|
||
def get_day(delta=0):
|
||
"""
|
||
获得指定格式的日期
|
||
"""
|
||
today = datetime.date.today()
|
||
oneday = datetime.timedelta(days=delta)
|
||
yesterday = today - oneday
|
||
return yesterday.strftime("%Y%m%d")
|
||
|
||
|
||
def take_out_json(content):
|
||
"""
|
||
从字符串jsonp中提取json数据
|
||
"""
|
||
s = re.search("^jsonp_\d+_\((.*?)\);?$", content)
|
||
return json.loads(s.group(1) if s else "{}")
|
||
|
||
|
||
def get_date():
|
||
"""Get current date"""
|
||
today = datetime.date.today()
|
||
return "%4d%02d%02d" % (today.year, today.month, today.day)
|
||
|
||
|
||
def get_photo(res):
|
||
try:
|
||
with open('./verify.png', 'wb') as f:
|
||
f.write(res)
|
||
return True
|
||
|
||
except Exception as e:
|
||
print(e)
|
||
return False
|
||
|
||
|
||
def parse():
|
||
ocr = ddddocr.DdddOcr(show_ad=False, old=True)
|
||
with open('./verify.png', 'rb') as f:
|
||
image = f.read()
|
||
res = ocr.classification(image)
|
||
print('验证码识别结果：', res)
|
||
return res
|
||
|
||
|
||
def get_and_parse(res):
|
||
if get_photo(res):
|
||
return parse()
|
||
else:
|
||
return False
|
||
|
||
|
||
class ZJULogin(object):
|
||
"""
|
||
Attributes:
|
||
username: (str) 浙大统一认证平台用户名（一般为学号）
|
||
password: (str) 浙大统一认证平台密码
|
||
sess: (requests.Session) 统一的session管理
|
||
"""
|
||
headers = {
|
||
'user-agent': 'Mozilla/5.0 (Linux; U; Android 11; zh-CN; M2012K11AC Build/RKQ1.200826.002) AppleWebKit/537.36 (KHTML, like Gecko) Version/4.0 Chrome/69.0.3497.100 UWS/3.22.0.36 Mobile Safari/537.36 AliApp(DingTalk/6.0.7.1) com.alibaba.android.rimet.zju/14785964 Channel/1543545060864 language/zh-CN 2ndType/exclusive UT4Aplus/0.2.25 colorScheme/light',
|
||
}
|
||
|
||
def __init__(self, username, password, delay_run=False):
|
||
self.username = username
|
||
self.password = password
|
||
self.delay_run = delay_run
|
||
self.sess = requests.Session()
|
||
self.imgaddress = 'https://healthreport.zju.edu.cn/ncov/wap/default/code'
|
||
self.BASE_URL = "https://healthreport.zju.edu.cn/ncov/wap/default/index"
|
||
self.LOGIN_URL = "https://zjuam.zju.edu.cn/cas/login?service=http%3A%2F%2Fservice.zju.edu.cn%2F"
|
||
|
||
self.TG_TOKEN = os.getenv("TG_TOKEN") # TG机器人的TOKEN
|
||
self.CHAT_ID = os.getenv("CHAT_ID") # 推送消息的CHAT_ID
|
||
self.DD_BOT_TOKEN = os.getenv("DD_BOT_TOKEN")
|
||
self.DD_BOT_SECRET = os.getenv("DD_BOT_SECRET") # 哈希算法验证(可选)
|
||
self.reminders = os.getenv("REMINDERS")
|
||
|
||
self.lng = "120.123138"
|
||
self.lat = "30.267931"
|
||
|
||
def login(self):
|
||
"""Login to ZJU platform"""
|
||
res = self.sess.get(self.LOGIN_URL)
|
||
execution = re.search(
|
||
'name="execution" value="(.*?)"', res.text).group(1)
|
||
res = self.sess.get(
|
||
url='https://zjuam.zju.edu.cn/cas/v2/getPubKey').json()
|
||
n, e = res['modulus'], res['exponent']
|
||
encrypt_password = self._rsa_encrypt(self.password, e, n)
|
||
|
||
data = {
|
||
'username': self.username,
|
||
'password': encrypt_password,
|
||
'execution': execution,
|
||
'_eventId': 'submit',
|
||
"authcode": ""
|
||
}
|
||
res = self.sess.post(url=self.LOGIN_URL, data=data)
|
||
# check if login successfully
|
||
if '用户名或密码错误' in res.content.decode():
|
||
raise LoginError('登录失败，请核实账号密码重新登录')
|
||
print("统一认证平台登录成功~")
|
||
return self.sess
|
||
|
||
def _rsa_encrypt(self, password_str, e_str, M_str):
|
||
password_bytes = bytes(password_str, 'ascii')
|
||
password_int = int.from_bytes(password_bytes, 'big')
|
||
e_int = int(e_str, 16)
|
||
M_int = int(M_str, 16)
|
||
result_int = pow(password_int, e_int, M_int)
|
||
return hex(result_int)[2:].rjust(128, '0')
|
||
|
||
|
||
class HealthCheckInHelper(ZJULogin):
|
||
headers = {
|
||
'user-agent': 'Mozilla/5.0 (Linux; U; Android 11; zh-CN; M2012K11AC Build/RKQ1.200826.002) AppleWebKit/537.36 (KHTML, like Gecko) Version/4.0 Chrome/69.0.3497.100 UWS/3.22.0.36 Mobile Safari/537.36 AliApp(DingTalk/6.0.7.1) com.alibaba.android.rimet.zju/14785964 Channel/1543545060864 language/zh-CN 2ndType/exclusive UT4Aplus/0.2.25 colorScheme/light',
|
||
}
|
||
|
||
REDIRECT_URL = "https://zjuam.zju.edu.cn/cas/login?service=https%3A%2F%2Fhealthreport.zju.edu.cn%2Fa_zju%2Fapi%2Fsso%2Findex%3Fredirect%3Dhttps%253A%252F%252Fhealthreport.zju.edu.cn%252Fncov%252Fwap%252Fdefault%252Findex%26from%3Dwap"
|
||
|
||
def get_ip_location(self):
|
||
headers = {
|
||
'authority': 'webapi.amap.com',
|
||
'pragma': 'no-cache',
|
||
'cache-control': 'no-cache',
|
||
'sec-ch-ua': '"Chromium";v="92", " Not A;Brand";v="99", "Google Chrome";v="92"',
|
||
'sec-ch-ua-mobile': '?0',
|
||
'user-agent': 'Mozilla/5.0 (Linux; U; Android 11; zh-CN; M2012K11AC Build/RKQ1.200826.002) AppleWebKit/537.36 (KHTML, like Gecko) Version/4.0 Chrome/69.0.3497.100 UWS/3.22.0.36 Mobile Safari/537.36 AliApp(DingTalk/6.0.7.1) com.alibaba.android.rimet.zju/14785964 Channel/1543545060864 language/zh-CN 2ndType/exclusive UT4Aplus/0.2.25 colorScheme/light',
|
||
'accept': '*/*',
|
||
'sec-fetch-site': 'cross-site',
|
||
'sec-fetch-mode': 'no-cors',
|
||
'sec-fetch-dest': 'script',
|
||
'referer': 'https://healthreport.zju.edu.cn/',
|
||
'accept-language': 'zh-CN,zh;q=0.9,en;q=0.8',
|
||
'cookie': 'isg=BIaGbUMSG7BxFM4x941hm4D913wI58qhRFwZi3CvdKmEcyaN2nUJsfYKT6-_W8K5',
|
||
}
|
||
|
||
params = (
|
||
('key', '729923f88542d91590470f613adb27b5'),
|
||
('callback', 'jsonp_859544_'),
|
||
('platform', 'JS'),
|
||
('logversion', '2.0'),
|
||
('appname', 'https://healthreport.zju.edu.cn/ncov/wap/default/index'),
|
||
('csid', '17F714D6-756D-49E4-96F2-B31F04B14A5A'),
|
||
('sdkversion', '1.4.16'),
|
||
)
|
||
response = self.sess.get(
|
||
'https://webapi.amap.com/maps/ipLocation?key=729923f88542d91590470f613adb27b5&callback=jsonp_859544_&platform=JS&logversion=2.0&appname=https%3A%2F%2Fhealthreport.zju.edu.cn%2Fncov%2Fwap%2Fdefault%2Findex&csid=17F714D6-756D-49E4-96F2-B31F04B14A5A&sdkversion=1.4.16',
|
||
headers=headers, params=params)
|
||
return take_out_json(response.text)
|
||
|
||
def get_geo_info(self, location: dict):
|
||
params = (
|
||
('key', '729923f88542d91590470f613adb27b5'),
|
||
('s', 'rsv3'),
|
||
('language', 'zh_cn'),
|
||
('location', '{lng},{lat}'.format(
|
||
lng=location.get("lng"), lat=location.get("lat"))),
|
||
('extensions', 'base'),
|
||
('callback', 'jsonp_607701_'),
|
||
('platform', 'JS'),
|
||
('logversion', '2.0'),
|
||
('appname', 'https://healthreport.zju.edu.cn/ncov/wap/default/index'),
|
||
('csid', '63157A4E-D820-44E1-B032-A77418183A4C'),
|
||
('sdkversion', '1.4.16'),
|
||
)
|
||
|
||
response = self.sess.get(
|
||
'https://restapi.amap.com/v3/geocode/regeo', headers=self.headers, params=params, )
|
||
return take_out_json(response.text)
|
||
|
||
def take_in(self, geo_info: dict):
|
||
formatted_address = geo_info.get("regeocode").get("formatted_address")
|
||
address_component = geo_info.get("regeocode").get("addressComponent")
|
||
if not formatted_address or not address_component:
|
||
return
|
||
|
||
# 获得id和uid参数
|
||
time.sleep(3)
|
||
res = self.sess.get(self.BASE_URL, headers=self.headers)
|
||
if len(res.content) == 0:
|
||
print('网页获取失败，请检查网络并重试')
|
||
html = res.content.decode()
|
||
try:
|
||
done = re.findall(
|
||
'温馨提示： 不外出、不聚集、不吃野味， 戴口罩、勤洗手、咳嗽有礼，开窗通风，发热就诊', html)[0]
|
||
print(done)
|
||
try:
|
||
res = self.sess.get(self.imgaddress, headers=self.headers)
|
||
code = get_and_parse(res.content)
|
||
if not code:
|
||
return
|
||
except:
|
||
print('验证码识别失败')
|
||
except:
|
||
print('打卡网页获取失败')
|
||
finally:
|
||
new_info_tmp = json.loads(re.findall(r'def = ({[^\n]+})', html)[0])
|
||
new_id = new_info_tmp['id']
|
||
new_uid = new_info_tmp['uid']
|
||
# 拼凑geo信息
|
||
lng, lat = address_component.get(
|
||
"streetNumber").get("location").split(",")
|
||
geo_api_info_dict = {"type": "complete", "info": "SUCCESS", "status": 1,
|
||
"position": {"Q": lat, "R": lng, "lng": lng, "lat": lat},
|
||
"message": "Get geolocation success.Convert Success.Get address success.", "location_type": "ip",
|
||
"accuracy": "null", "isConverted": "true", "addressComponent": address_component,
|
||
"formattedAddress": formatted_address, "roads": [], "crosses": [], "pois": []}
|
||
#print('打卡地点：', formatted_address)
|
||
# 拿到校验值
|
||
verify_data = re.findall(
|
||
r'"([a-z0-9]*?)": "([0-9]*?)","([a-z0-9]*?)":"([a-z0-9]*?)"', html)[0]
|
||
verify_code = {
|
||
verify_data[0]: verify_data[1],
|
||
verify_data[2]: verify_data[3],
|
||
}
|
||
data = {
|
||
'sfymqjczrj': '0',
|
||
'zjdfgj': '',
|
||
'sfyrjjh': '0',
|
||
'cfgj': '',
|
||
'tjgj': '',
|
||
'nrjrq': '0',
|
||
'rjka': '',
|
||
'jnmddsheng': '',
|
||
'jnmddshi': '',
|
||
'jnmddqu': '',
|
||
'jnmddxiangxi': '',
|
||
'rjjtfs': '',
|
||
'rjjtfs1': '',
|
||
'rjjtgjbc': '',
|
||
'jnjtfs': '',
|
||
'jnjtfs1': '',
|
||
'jnjtgjbc': '',
|
||
# 是否确认信息属实
|
||
'sfqrxxss': '1',
|
||
'sfqtyyqjwdg': '0',
|
||
'sffrqjwdg': '0',
|
||
'sfhsjc': '',
|
||
'zgfx14rfh': '0',
|
||
'zgfx14rfhdd': '',
|
||
'sfyxjzxgym': '1',
|
||
# 是否不宜接种人群
|
||
'sfbyjzrq': '5',
|
||
'jzxgymqk': '6', # 这里是第三针相关参数，1是已接种第一针，4是已接种第二针（已满6个月），5是已接种第二针（未满6个月），6是已接种第三针，3是未接种，记得自己改
|
||
'tw': '0',
|
||
'sfcxtz': '0',
|
||
'sfjcbh': '0',
|
||
'sfcxzysx': '0',
|
||
'jcjg': '',
|
||
'qksm': '',
|
||
'sfyyjc': '0',
|
||
'jcjgqr': '0',
|
||
'remark': '',
|
||
'address': formatted_address,
|
||
# {"type":"complete","position":{"Q":30.30975640191,"R":120.085647515191,"lng":120.085648,"lat":30.309756},"location_type":"html5","message":"Get geolocation success.Convert Success.Get address success.","accuracy":40,"isConverted":true,"status":1,"addressComponent":{"citycode":"0571","adcode":"330106","businessAreas":[],"neighborhoodType":"","neighborhood":"","building":"","buildingType":"","street":"龙宇街","streetNumber":"17-18号","country":"中国","province":"浙江省","city":"杭州市","district":"西湖区","towncode":"330106109000","township":"三墩镇"},"formattedAddress":"浙江省杭州市西湖区三墩镇翠柏浙江大学(紫金港校区)","roads":[],"crosses":[],"pois":[],"info":"SUCCESS"}
|
||
'geo_api_info': geo_api_info_dict,
|
||
# 浙江省 杭州市 西湖区
|
||
# '\u6D59\u6C5F\u7701 \u676D\u5DDE\u5E02 \u897F\u6E56\u533A'
|
||
'area': "{} {} {}".format(address_component.get("province"), address_component.get("city"),
|
||
address_component.get("district")),
|
||
# 浙江省
|
||
# '\u6D59\u6C5F\u7701'
|
||
'province': address_component.get("province"),
|
||
# 杭州市
|
||
# '\u676D\u5DDE\u5E02'
|
||
'city': address_component.get("city"),
|
||
# 是否在校：在校将'sfzx'改为1
|
||
'sfzx': '1',
|
||
'sfjcwhry': '0',
|
||
'sfjchbry': '0',
|
||
'sfcyglq': '0',
|
||
'gllx': '',
|
||
'glksrq': '',
|
||
'jcbhlx': '',
|
||
'jcbhrq': '',
|
||
'bztcyy': '4',
|
||
'sftjhb': '0',
|
||
'sftjwh': '0',
|
||
'fjsj': '0',
|
||
'sfjcqz': '',
|
||
'jcqzrq': '',
|
||
'jrsfqzys': '',
|
||
'jrsfqzfy': '',
|
||
'sfyqjzgc': '',
|
||
# 是否申领杭州健康码
|
||
'sfsqhzjkk': '1',
|
||
# 杭州健康吗颜色，1:绿色 2:红色 3:黄色
|
||
'sqhzjkkys': '1',
|
||
'gwszgzcs': '',
|
||
'szgj': '',
|
||
'fxyy': '',
|
||
'jcjg': '',
|
||
# uid每个用户不一致
|
||
'uid': new_uid,
|
||
# id每个用户不一致
|
||
'id': new_id,
|
||
# 日期
|
||
'date': get_date(),
|
||
'created': round(time.time()),
|
||
'szsqsfybl': '0',
|
||
'sfygtjzzfj': '0',
|
||
'gtjzzfjsj': '',
|
||
'gwszdd': '',
|
||
'szgjcs': '',
|
||
'ismoved': '0', # 位置变化为1，不变为0
|
||
'zgfx14rfhsj': '',
|
||
'jrdqjcqk': '',
|
||
'jcwhryfs': '',
|
||
'jchbryfs': '',
|
||
'xjzd': '',
|
||
'sfsfbh': '0',
|
||
'jhfjrq': '',
|
||
'jhfjjtgj': '',
|
||
'jhfjhbcc': '',
|
||
'jhfjsftjwh': '0',
|
||
'jhfjsftjhb': '0',
|
||
'szsqsfybl': '0',
|
||
'gwszgz': '',
|
||
'campus': '紫金港校区', # 紫金港校区 玉泉校区 西溪校区 华家池校区 之江校区 海宁校区 舟山校区 宁波校区 工程师学院 杭州国际科创中心 其他
|
||
# 👇-----2022.5.7日修改-----👇
|
||
'verifyCode': code,
|
||
# 👆-----2022.5.7日修改-----👆
|
||
}
|
||
data.update(verify_code)
|
||
response = self.sess.post('https://healthreport.zju.edu.cn/ncov/wap/default/save', data=data,
|
||
headers=self.headers)
|
||
return response.json()
|
||
|
||
def run(self):
|
||
print("正在为{}健康打卡".format(self.username))
|
||
if self.delay_run:
|
||
# 确保定时脚本执行时间不太一致
|
||
time.sleep(random.randint(10, 100))
|
||
try:
|
||
self.login()
|
||
# 拿取eai-sess的cookies信息
|
||
self.sess.get(self.REDIRECT_URL)
|
||
# location = get_ip_location()
|
||
# print(location)
|
||
location = {'info': 'LOCATE_SUCCESS',
|
||
'status': 1, 'lng': self.lng, 'lat': self.lat}
|
||
geo_info = self.get_geo_info(location)
|
||
# print(geo_info)
|
||
res = self.take_in(geo_info)
|
||
print(res)
|
||
except requests.exceptions.ConnectionError:
|
||
# reraise as KubeException, but log stacktrace.
|
||
print("打卡失败,请检查github服务器网络状态")
|
||
|
||
|
||
if __name__ == '__main__':
|
||
# 因为是github action版本，所以不加上循环多人打卡功能
|
||
account = os.getenv("account")
|
||
password = os.getenv("password")
|
||
s = HealthCheckInHelper(account, password, delay_run=True)
|
||
schedule.every().day.at("06:04").do(s.run)
|
||
while True:
|
||
try:
|
||
schedule.run_pending()
|
||
time.sleep(30)
|
||
except Exception:
|
||
print("ERROR") |