jd/smzdm_script
1
0
Fork 0
mirror of https://github.com/hex-ci/smzdm_script.git synced 2026-02-03 02:24:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

Introduce loguru

Browse Source
This commit is contained in:
LuckyHunter
2022-12-18 13:32:18 +08:00
parent 312e9a38eb
commit 5966c7f1bd
5 changed files with 73 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

61
notify/notify.py
View File

@@ -1,13 +1,12 @@
import json
from pprint import pprint
from typing import Dict
from urllib.parse import urljoin
import requests
from loguru import logger
class NotifyBot(object):
def __init__(self, content, title="什么值得买签到", **kwargs: Dict) -> None:
self.content = content
self.title = title
@@ -17,68 +16,72 @@ class NotifyBot(object):
self.server_chain()
self.tg_bot()
def push_plus(self, template='html'):
def push_plus(self, template="html"):
try:
if self.kwargs.get("PUSH_PLUS_TOKEN", None):
PUSH_PLUS_TOKEN = self.kwargs.get("PUSH_PLUS_TOKEN")
else:
pprint("⚠️ PUSH_PLUS_TOKEN not set, skip PushPlus nofitication")
logger.info("⚠️ PUSH_PLUS_TOKEN not set, skip PushPlus nofitication")
return
url = 'https://www.pushplus.plus/send'
url = "https://www.pushplus.plus/send"
body = {
'token': PUSH_PLUS_TOKEN,
'title': self.title,
'content': self.content,
'template': template
"token": PUSH_PLUS_TOKEN,
"title": self.title,
"content": self.content,
"template": template,
}
data = json.dumps(body).encode(encoding='utf-8')
headers = {'Content-Type': 'application/json'}
data = json.dumps(body).encode(encoding="utf-8")
headers = {"Content-Type": "application/json"}
resp = requests.post(url, data=data, headers=headers)
if resp.status_code == 200:
pprint("✅ Push Plus notified")
logger.info("✅ Push Plus notified")
return resp.json()
except Exception as e:
pprint(e)
logger.error(e)
def server_chain(self):
try:
if self.kwargs.get("SC_KEY", None):
SC_KEY = self.kwargs.get("SC_KEY")
else:
pprint("⚠️ SC_KEY not set, skip ServerChain notification")
logger.info("⚠️ SC_KEY not set, skip ServerChain notification")
return
url = f'http://sc.ftqq.com/{SC_KEY}.send'
url = f"http://sc.ftqq.com/{SC_KEY}.send"
data = {'text': self.title, 'desp': self.content}
data = {"text": self.title, "desp": self.content}
resp = requests.post(url, data=data)
if resp.status_code == 200:
pprint("✅ Server Chain notified")
logger.info("✅ Server Chain notified")
return resp.json()
except Exception as e:
pprint(e)
logger.error(e)
def tg_bot(self):
try:
if self.kwargs.get("TG_BOT_TOKEN", None) and self.kwargs.get("TG_USER_ID", None):
if self.kwargs.get("TG_BOT_TOKEN", None) and self.kwargs.get(
"TG_USER_ID", None
):
TG_BOT_TOKEN = self.kwargs.get("TG_BOT_TOKEN")
TG_USER_ID = self.kwargs.get("TG_USER_ID")
else:
pprint(
"⚠️ TG_BOT_TOKEN & TG_USER_ID not set, skip TelegramBot notification")
logger.info(
"⚠️ TG_BOT_TOKEN & TG_USER_ID not set, skip TelegramBot notification"
)
return
TG_BOT_API = self.kwargs.get(
"TG_BOT_API", "https://api.telegram.org")
TG_BOT_API = self.kwargs.get("TG_BOT_API", "https://api.telegram.org")
url = urljoin(TG_BOT_API, f"/bot{TG_BOT_TOKEN}/sendMessage")
headers = {'Content-Type': 'application/x-www-form-urlencoded'}
payload = {'chat_id': str(TG_USER_ID),
'text': f'{self.title}\n\n{self.content}',
'disable_web_page_preview': 'true'}
headers = {"Content-Type": "application/x-www-form-urlencoded"}
payload = {
"chat_id": str(TG_USER_ID),
"text": f"{self.title}\n\n{self.content}",
"disable_web_page_preview": "true",
}
resp = requests.post(url=url, headers=headers, params=payload)
if resp.status_code == 200:
pprint("✅ Telegram Bot notified")
logger.info("✅ Telegram Bot notified")
return resp.json()
except Exception as e:
pprint(e)
logger.error(e)