340 lines
14 KiB
Python
340 lines
14 KiB
Python
'''
|
||
@File : notify.py
|
||
@Github : https://github.com/y1ndan/genshin-impact-helper
|
||
@Last modified by : y1ndan
|
||
@Last modified time : 2021-01-13 11:01:10
|
||
'''
|
||
import json
|
||
import os
|
||
import time
|
||
import hmac
|
||
import hashlib
|
||
import base64
|
||
|
||
import requests
|
||
from requests.exceptions import HTTPError
|
||
from urllib import parse
|
||
|
||
from settings import log
|
||
|
||
|
||
class Notify(object):
|
||
@staticmethod
|
||
def to_python(json_str: str):
|
||
return json.loads(json_str)
|
||
|
||
@staticmethod
|
||
def to_json(obj):
|
||
return json.dumps(obj, indent=4, ensure_ascii=False)
|
||
|
||
# ============================== Server Chan ==============================
|
||
# 此处填你申请的SCKEY
|
||
# 注: Github Actions用户请到Settings->Secrets里设置,Name=SCKEY,Value=<获取的值>
|
||
SCKEY = ''
|
||
|
||
if os.environ.get('SCKEY', '') != '':
|
||
SCKEY = os.environ['SCKEY']
|
||
|
||
# ============================== Cool Push ================================
|
||
# 此处填你申请的SKEY(详见文档: https://cp.xuthus.cc/)
|
||
# 注: Github Actions用户请到Settings->Secrets里设置,Name=COOL_PUSH_SKEY,Value=<获取的值>
|
||
COOL_PUSH_SKEY = ''
|
||
# 此处填写私聊(send)或群组(group)或者微信(wx)推送方式，默认私聊推送
|
||
# 注: Github Actions用户若要更改,请到Settings->Secrets里设置,Name=COOL_PUSH_MODE,Value=<group或wx>
|
||
COOL_PUSH_MODE = 'send'
|
||
|
||
if os.environ.get('COOL_PUSH_SKEY', '') != '':
|
||
COOL_PUSH_SKEY = os.environ['COOL_PUSH_SKEY']
|
||
if os.environ.get('COOL_PUSH_MODE', '') != '':
|
||
COOL_PUSH_MODE = os.environ['COOL_PUSH_MODE']
|
||
|
||
# ============================== iOS Bark App =============================
|
||
# 此处填你Bark App的信息(IP/设备码,例如: https://api.day.app/XXXXXXXX)
|
||
# 注: Github Actions用户请到Settings->Secrets里设置,Name=BARK_KEY,Value=<获取的值>
|
||
BARK_KEY = ''
|
||
# BARK App推送铃声,铃声列表去App内查看
|
||
# 注: Github Actions用户若要更改,请到Settings->Secrets里设置,Name=BARK_SOUND,Value=<铃声名称>
|
||
BARK_SOUND = 'healthnotification'
|
||
|
||
if os.environ.get('BARK_KEY', '') != '':
|
||
if os.environ['BARK_KEY'].find(
|
||
'https') != -1 or os.environ['BARK_KEY'].find('http') != -1:
|
||
# 兼容BARK自建服务端用户
|
||
BARK_KEY = os.environ['BARK_KEY']
|
||
else:
|
||
BARK_KEY = 'https://api.day.app/' + os.environ['BARK_KEY']
|
||
elif os.environ.get('BARK_SOUND', '') != '':
|
||
BARK_SOUND = os.environ['BARK_SOUND']
|
||
elif BARK_KEY != '' or BARK_KEY.find('https') != -1 or BARK_KEY.find(
|
||
'http') != -1:
|
||
# 兼容BARK本地用户只填写设备码的情况
|
||
BARK_KEY = 'https://api.day.app/' + BARK_KEY
|
||
|
||
# ============================== Telegram Bot =============================
|
||
# 此处填你telegram bot的Token,例如: 1077xxx4424:AAFjv0FcqxxxxxxgEMGfi22B4yh15R5uw
|
||
# 注: Github Actions用户请到Settings->Secrets里设置,Name=TG_BOT_TOKEN,Value=<获取的值>
|
||
TG_BOT_TOKEN = ''
|
||
# 此处填你接收通知消息的telegram用户的id,例如: 129xxx206
|
||
# 注: Github Actions用户请到Settings->Secrets里设置,Name=TG_USER_ID,Value=<获取的值>
|
||
TG_USER_ID = ''
|
||
|
||
if os.environ.get('TG_BOT_TOKEN', '') != '':
|
||
TG_BOT_TOKEN = os.environ['TG_BOT_TOKEN']
|
||
if os.environ.get('TG_USER_ID', '') != '':
|
||
TG_USER_ID = os.environ['TG_USER_ID']
|
||
|
||
# ============================== DingTalk Bot =============================
|
||
# 此处填你钉钉机器人的webhook,例如: 5a544165465465645d0f31dca676e7bd07415asdasd
|
||
# 注: Github Actions用户请到Settings->Secrets里设置,Name=DD_BOT_TOKEN,Value=<获取的值>
|
||
DD_BOT_TOKEN = ''
|
||
# 加签密钥,机器人安全设置页面,加签一栏下面显示的SEC开头的字符串
|
||
# 注: Github Actions用户请到Settings->Secrets里设置,Name=DD_BOT_SECRET,Value=<获取的值>
|
||
DD_BOT_SECRET = ''
|
||
|
||
if os.environ.get('DD_BOT_TOKEN', '') != '':
|
||
DD_BOT_TOKEN = os.environ['DD_BOT_TOKEN']
|
||
if os.environ.get('DD_BOT_SECRET', '') != '':
|
||
DD_BOT_SECRET = os.environ['DD_BOT_SECRET']
|
||
|
||
# ============================== WeChat Work Bot ==========================
|
||
# 此处填你企业微信机器人的webhook(详见文档 https://work.weixin.qq.com/api/doc/90000/90136/91770) 例如: 693a91f6-7xxx-4bc4-97a0-0ec2sifa5aaa
|
||
# 注: Github Actions用户请到Settings->Secrets里设置,Name=WW_BOT_KEY,Value=<获取的值>
|
||
WW_BOT_KEY = ''
|
||
|
||
if os.environ.get('WW_BOT_KEY', '') != '':
|
||
WW_BOT_KEY = os.environ['WW_BOT_KEY']
|
||
|
||
# ============================== iGot聚合推送 =================================
|
||
# 此处填你iGot的信息(推送key,例如: https://push.hellyw.com/XXXXXXXX)
|
||
# 注: Github Actions用户请到Settings->Secrets里设置,Name=IGOT_KEY,Value=<获取的值>
|
||
IGOT_KEY = ''
|
||
|
||
if os.environ.get('IGOT_KEY', '') != '':
|
||
IGOT_KEY = os.environ['IGOT_KEY']
|
||
|
||
# ============================== push+ ====================================
|
||
# 官方文档: https://pushplus.hxtrip.com/
|
||
# PUSH_PLUS_TOKEN: 微信扫码登录后一对一推送或一对多推送下面的token(您的Token)，不配置PUSH_PLUS_USER则默认为一对一推送
|
||
# 注: Github Actions用户请到Settings->Secrets里设置,Name=PUSH_PLUS_TOKEN,Value=<获取的值>
|
||
PUSH_PLUS_TOKEN = ''
|
||
# PUSH_PLUS_USER: 一对多推送的“群组编码”（一对多推送下面->您的群组(如无则新建)->群组编码，如果您是创建群组人。也需点击“查看二维码”扫描绑定，否则不能接受群组消息推送）
|
||
# 注: Github Actions用户请到Settings->Secrets里设置,Name=PUSH_PLUS_USER,Value=<获取的值>
|
||
PUSH_PLUS_USER = ''
|
||
|
||
if os.environ.get('PUSH_PLUS_TOKEN', '') != '':
|
||
PUSH_PLUS_TOKEN = os.environ['PUSH_PLUS_TOKEN']
|
||
if os.environ.get('PUSH_PLUS_USER', '') != '':
|
||
PUSH_PLUS_USER = os.environ['PUSH_PLUS_USER']
|
||
|
||
def serverChan(self, text, status, desp):
|
||
if Notify.SCKEY != '':
|
||
url = 'https://sc.ftqq.com/{}.send'.format(Notify.SCKEY)
|
||
data = {'text': '{} {}'.format(text, status), 'desp': desp}
|
||
try:
|
||
response = self.to_python(requests.post(url, data=data).text)
|
||
except Exception as e:
|
||
log.error(e)
|
||
raise HTTPError
|
||
else:
|
||
if response['errno'] == 0:
|
||
log.info('Server酱推送成功')
|
||
elif response['errno'] == 1024:
|
||
# SCKEY错误或一分钟内发送相同内容
|
||
log.error('Server酱推送失败:\n{}'.format(response['errmsg']))
|
||
else:
|
||
log.error('Server酱推送失败:\n{}'.format(response))
|
||
else:
|
||
log.info('您未配置Server酱推送所需的SCKEY,取消Server酱推送')
|
||
pass
|
||
|
||
def coolPush(self, text, status, desp):
|
||
if Notify.COOL_PUSH_SKEY != '':
|
||
url = 'https://push.xuthus.cc/{}/{}'.format(
|
||
Notify.COOL_PUSH_MODE, Notify.COOL_PUSH_SKEY)
|
||
data = '{} {}\n\n{}'.format(text, status, desp).encode('utf-8')
|
||
try:
|
||
response = self.to_python(requests.post(url, data=data).text)
|
||
except Exception as e:
|
||
log.error(e)
|
||
raise HTTPError
|
||
else:
|
||
if response['code'] == 200:
|
||
log.info('Cool Push推送成功')
|
||
else:
|
||
log.error('Cool Push推送失败:\n{}'.format(response))
|
||
else:
|
||
log.info('您未配置Cool Push推送所需的COOL_PUSH_SKEY,取消Cool Push推送')
|
||
pass
|
||
|
||
def bark(self, text, status, desp):
|
||
if Notify.BARK_KEY != '':
|
||
url = '{}/{} {}/{}?sound={}'.format(Notify.BARK_KEY,
|
||
text, status, parse.quote(desp), Notify.BARK_SOUND)
|
||
try:
|
||
response = self.to_python(requests.get(url).text)
|
||
except Exception as e:
|
||
log.error(e)
|
||
raise HTTPError
|
||
else:
|
||
if response['code'] == 200:
|
||
log.info('Bark推送成功')
|
||
elif response['code'] == 400:
|
||
log.error('Bark推送失败:\n{}'.format(response['message']))
|
||
else:
|
||
log.error('Bark推送失败:\n{}'.format(response))
|
||
else:
|
||
log.info('您未配置Bark推送所需的BARK_KEY,取消Bark推送')
|
||
pass
|
||
|
||
def tgBot(self, text, status, desp):
|
||
if Notify.TG_BOT_TOKEN != '' or Notify.TG_USER_ID != '':
|
||
url = 'https://api.telegram.org/bot{}/sendMessage'.format(
|
||
Notify.TG_BOT_TOKEN)
|
||
data = {
|
||
'chat_id': Notify.TG_USER_ID,
|
||
'text': '{} {}\n\n{}'.format(text, status, desp),
|
||
'disable_web_page_preview': True
|
||
}
|
||
try:
|
||
response = self.to_python(requests.post(url, data=data).text)
|
||
except Exception as e:
|
||
log.error(e)
|
||
raise HTTPError
|
||
else:
|
||
if response['ok']:
|
||
log.info('Telegram推送成功')
|
||
elif response['error_code'] == 400:
|
||
log.error('请主动给bot发送一条消息并检查接收用户ID是否正确')
|
||
elif response['error_code'] == 401:
|
||
log.error('TG_BOT_TOKEN错误')
|
||
else:
|
||
log.error('Telegram推送失败:\n{}'.format(response))
|
||
else:
|
||
log.info('您未配置Telegram推送所需的TG_BOT_TOKEN和TG_USER_ID,取消Telegram推送')
|
||
pass
|
||
|
||
def ddBot(self, text, status, desp):
|
||
if Notify.DD_BOT_TOKEN != '':
|
||
url = 'https://oapi.dingtalk.com/robot/send?access_token={}'.format(
|
||
Notify.DD_BOT_TOKEN)
|
||
data = {
|
||
'msgtype': 'text',
|
||
'text': {
|
||
'content': '{} {}\n\n{}'.format(text, status, desp)
|
||
}
|
||
}
|
||
if Notify.DD_BOT_SECRET != '':
|
||
secret = Notify.DD_BOT_SECRET
|
||
timestamp = int(round(time.time() * 1000))
|
||
secret_enc = bytes(secret).encode('utf-8')
|
||
string_to_sign = '{}\n{}'.format(timestamp, secret)
|
||
string_to_sign_enc = bytes(string_to_sign).encode('utf-8')
|
||
hmac_code = hmac.new(
|
||
secret_enc, string_to_sign_enc,
|
||
digestmod=hashlib.sha256).digest()
|
||
sign = parse.quote_plus(base64.b64encode(hmac_code))
|
||
url = 'https://oapi.dingtalk.com/robot/send?access_token={}×tamp={}&sign={}'.format(
|
||
Notify.DD_BOT_TOKEN, timestamp, sign)
|
||
try:
|
||
response = self.to_python(requests.post(url, data=data).text)
|
||
except Exception as e:
|
||
log.error(e)
|
||
raise HTTPError
|
||
else:
|
||
if response['errcode'] == 0:
|
||
log.info('钉钉推送成功')
|
||
else:
|
||
log.error('钉钉推送失败:\n{}'.format(response))
|
||
else:
|
||
log.info('您未配置钉钉推送所需的DD_BOT_TOKEN或DD_BOT_SECRET,取消钉钉推送')
|
||
pass
|
||
|
||
def wwBot(self, text, status, desp):
|
||
if Notify.WW_BOT_KEY != '':
|
||
url = 'https://qyapi.weixin.qq.com/cgi-bin/webhook/send?key={}'.format(
|
||
Notify.WW_BOT_KEY)
|
||
data = {
|
||
'msgtype': 'text',
|
||
'text': {
|
||
'content': '{} {}\n\n{}'.format(text, status, desp)
|
||
}
|
||
}
|
||
try:
|
||
response = self.to_python(requests.post(url, data=data).text)
|
||
except Exception as e:
|
||
log.error(e)
|
||
raise HTTPError
|
||
else:
|
||
if response['errcode'] == 0:
|
||
log.info('企业微信推送成功')
|
||
else:
|
||
log.error('企业微信推送失败:\n{}'.format(response))
|
||
else:
|
||
log.info('您未配置企业微信推送所需的WW_BOT_KEY,取消企业微信推送')
|
||
pass
|
||
|
||
def iGot(self, text, status, desp):
|
||
if Notify.IGOT_KEY != '':
|
||
url = 'https://push.hellyw.com/{}'.format(Notify.IGOT_KEY)
|
||
data = {'title': '{} {}'.format(text, status), 'content': desp}
|
||
try:
|
||
response = self.to_python(requests.post(url, data=data).text)
|
||
except Exception as e:
|
||
log.error(e)
|
||
raise HTTPError
|
||
else:
|
||
if response['ret'] == 0:
|
||
log.info('iGot推送成功')
|
||
else:
|
||
log.error('iGot推送失败:\n{}'.format(response))
|
||
else:
|
||
log.info('您未配置iGot推送所需的IGOT_KEY,取消iGot推送')
|
||
pass
|
||
|
||
def pushPlus(self, text, status, desp):
|
||
if Notify.PUSH_PLUS_TOKEN != '':
|
||
url = 'https://pushplus.hxtrip.com/send'
|
||
data = {
|
||
'token': Notify.PUSH_PLUS_TOKEN,
|
||
'title': '{} {}'.format(text, status),
|
||
'content': desp,
|
||
'topic': Notify.PUSH_PLUS_USER
|
||
}
|
||
try:
|
||
response = self.to_python(requests.post(url, data=data).text)
|
||
except Exception as e:
|
||
log.error(e)
|
||
raise HTTPError
|
||
else:
|
||
if response['code'] == 200:
|
||
log.info('pushplus推送成功')
|
||
else:
|
||
log.error('pushplus推送失败:\n{}'.format(response))
|
||
else:
|
||
log.info('您未配置pushplus推送所需的PUSH_PLUS_TOKEN,取消pushplus推送')
|
||
pass
|
||
|
||
def send(self, **kwargs):
|
||
app = '原神签到小助手'
|
||
status = kwargs.get('status', '')
|
||
msg = kwargs.get('msg', '')
|
||
if isinstance(msg, list) or isinstance(msg, dict):
|
||
# msg = self.to_json(msg)
|
||
msg = '\n\n'.join(msg)
|
||
log.info(f'签到结果: {status}\n\n{msg}')
|
||
log.info('准备推送通知...')
|
||
|
||
self.serverChan(app, status, msg)
|
||
self.coolPush(app, status, msg)
|
||
self.bark(app, status, msg)
|
||
self.tgBot(app, status, msg)
|
||
self.ddBot(app, status, msg)
|
||
self.wwBot(app, status, msg)
|
||
self.iGot(app, status, msg)
|
||
self.pushPlus(app, status, msg)
|
||
|
||
|
||
if __name__ == '__main__':
|
||
Notify().send(app='原神签到小助手', status='签到状态', msg='内容详情')
|
||
|