80 lines
4.0 KiB
Python
80 lines
4.0 KiB
Python
# notify.py
|
|
import requests
|
|
import json
|
|
from UserDefined import poststate, UseResultPush, PushToken, shixiseng_state, yizhanchi_state
|
|
from system import ResultUrl, successful_deliveries
|
|
from shixiseng import run_shixiseng_script
|
|
from yizhanchi import run_yizhanchi_script
|
|
|
|
def send_notifications():
|
|
findCount=0
|
|
successCount=0
|
|
errorCount=0
|
|
notify_shixiseng_findCount=0
|
|
notify_shixiseng_successCount=0
|
|
notify_shixiseng_errorCount=0
|
|
notify_yizhanchi_findCount=0
|
|
notify_yizhanchi_successCount=0
|
|
notify_yizhanchi_errorCount=0
|
|
notify = "------------Notify---------------\n"
|
|
if poststate in [1, 2]:
|
|
if(shixiseng_state==1):
|
|
shixiseng_findCount, shixiseng_successCount, shixiseng_errorCount = run_shixiseng_script()
|
|
findCount=findCount + shixiseng_findCount
|
|
notify_shixiseng_findCount=shixiseng_findCount
|
|
successCount=successCount + shixiseng_successCount
|
|
notify_shixiseng_successCount=shixiseng_successCount
|
|
errorCount=errorCount + shixiseng_errorCount
|
|
notify_shixiseng_errorCount=shixiseng_errorCount
|
|
if(yizhanchi_state==1):
|
|
yizhanchi_findCount, yizhanchi_successCount, yizhanchi_errorCount = run_yizhanchi_script()
|
|
findCount=findCount + yizhanchi_findCount
|
|
notify_yizhanchi_findCount=yizhanchi_findCount
|
|
successCount=successCount + yizhanchi_successCount
|
|
notify_yizhanchi_successCount=yizhanchi_successCount
|
|
errorCount=errorCount + yizhanchi_errorCount
|
|
notify_yizhanchi_errorCount=yizhanchi_errorCount
|
|
notify += (f"今日总共找到{findCount}家公司\n"
|
|
f"投递成功{successCount}份，投递失败{errorCount}份\n"
|
|
f"-------------实习僧数据------------\n"
|
|
f"找到{notify_shixiseng_findCount}家公司\n"
|
|
f"投递成功{notify_shixiseng_successCount}份，投递失败{notify_shixiseng_errorCount}份\n")
|
|
if(notify_shixiseng_findCount!=0 and notify_shixiseng_successCount==0 and notify_shixiseng_errorCount==0):
|
|
notify += ( f"您未开启实习僧投递，本次投递跳过\n")
|
|
notify += ( f"-------------翼展翅数据------------\n"
|
|
f"找到{notify_yizhanchi_findCount}家公司\n"
|
|
f"投递成功{notify_yizhanchi_successCount}份，投递失败{notify_yizhanchi_errorCount}份\n")
|
|
if(notify_yizhanchi_findCount!=0 and notify_yizhanchi_successCount==0 and notify_yizhanchi_errorCount==0):
|
|
notify += ( f"您未开启翼展翅投递，本次投递跳过\n")
|
|
try:
|
|
notify += "---------成功投递的岗位信息-------\n"
|
|
for delivery in successful_deliveries:
|
|
notify += (f"UUID: {delivery['UUID']}\n"
|
|
f"公司: {delivery['公司']}\n"
|
|
f"城市: {delivery['城市']}\n"
|
|
f"岗位: {delivery['岗位']}\n"
|
|
f"薪资: {delivery['薪资']}\n"
|
|
f"福利: {delivery['福利']}\n"
|
|
# f"投递来自: {delivery['投递来自']}\n"
|
|
)
|
|
notify += "-------------------------------\n"
|
|
|
|
except Exception as e:
|
|
notify += "无\n"
|
|
|
|
print(notify)
|
|
else:
|
|
notify += ("-------------------------------\n"
|
|
"未投递，您已选择不投递任何公司\n"
|
|
"-------------------------------\n")
|
|
|
|
if UseResultPush == 1:
|
|
print("您已经开启推送，即将为您推送投递通知")
|
|
data = {"token": PushToken, "title": "Super Employees投递通知", "content": notify}
|
|
body = json.dumps(data).encode(encoding='utf-8')
|
|
Resultheaders = {'Content-Type': 'application/json'}
|
|
requests.post(ResultUrl, data=body, headers=Resultheaders)
|
|
print("推送完成，任务结束")
|
|
else:
|
|
print("未开启推送，任务结束")
|