github/DouyinLiveRecorder
1
0
Fork 0
mirror of https://github.com/ihmily/DouyinLiveRecorder.git synced 2025-12-26 05:48:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: add message batch push

Browse Source
This commit is contained in:
ihmily 2024-10-12 19:03:26 +08:00
parent 05da54cebe
commit f11787d515
3 changed files with 189 additions and 168 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

154
douyinliverecorder/msg_push.py
View File

@ -1,154 +0,0 @@
# -*- coding: utf-8 -*-
"""
Author: Hmily
GitHub: https://github.com/ihmily
Date: 2023-09-03 19:18:36
Update: 2024-10-05 11:45:12
Copyright (c) 2023-2024 by Hmily, All Rights Reserved.
"""
from typing import Dict, Any, Optional, Union
import json
import urllib.request
from .utils import trace_error_decorator
import smtplib
from email.header import Header
from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText
no_proxy_handler = urllib.request.ProxyHandler({})
opener = urllib.request.build_opener(no_proxy_handler)
headers: Dict[str, str] = {'Content-Type': 'application/json'}
@trace_error_decorator
def dingtalk(url: str, content: str, number: Optional[str] = '') -> Union[Dict[str, Any], None]:
json_data = {
'msgtype': 'text',
'text': {
'content': content,
},
"at": {
"atMobiles": [
number # 添加这个手机号，可以被@通知（必须要在群里）
],
},
}
data = json.dumps(json_data).encode('utf-8')
req = urllib.request.Request(url, data=data, headers=headers)
response = opener.open(req, timeout=10)
json_str = response.read().decode('utf-8')
json_data = json.loads(json_str)
if json_data['errcode'] != 0:
print(f'钉钉推送失败, {json_data["errmsg"]}')
return
return json_data
@trace_error_decorator
def xizhi(url: str, content: str, title: str = '直播间状态更新') -> Union[Dict[str, Any], None]:
json_data = {
'title': title,
'content': content
}
data = json.dumps(json_data).encode('utf-8')
req = urllib.request.Request(url, data=data, headers=headers)
response = opener.open(req, timeout=10)
json_str = response.read().decode('utf-8')
json_data = json.loads(json_str)
if json_data['code'] != 200:
print(f'微信推送失败, 失败信息：{json_data["msg"]}')
return
return json_data
@trace_error_decorator
def email_message(mail_host: str, mail_pass: str, from_email: str, to_email: str, title: str, content: str) -> (
Union)[Dict[str, Any], None]:
message = MIMEMultipart()
message['From'] = "{}".format(from_email)
message['Subject'] = Header(title, 'utf-8')
receivers = to_email.replace('，', ',').split(',')
if len(receivers) == 1:
message['To'] = receivers[0]
t_apart = MIMEText(content, 'plain', 'utf-8')
message.attach(t_apart)
try:
smtp_obj = smtplib.SMTP_SSL(mail_host, 465)
smtp_obj.login(from_email, mail_pass)
smtp_obj.sendmail(from_email, receivers, message.as_string())
data = {'code': 200, 'msg': '邮件发送成功'}
return data
except smtplib.SMTPException as e:
raise smtplib.SMTPException(f'邮件发送失败 {e}')
@trace_error_decorator
def tg_bot(chat_id: int, token: str, content: str) -> Dict[str, Any]:
json_data = {
"chat_id": chat_id,
'text': content
}
url = f'https://api.telegram.org/bot{token}/sendMessage'
data = json.dumps(json_data).encode('utf-8')
req = urllib.request.Request(url, data=data, headers=headers)
response = urllib.request.urlopen(req, timeout=15)
json_str = response.read().decode('utf-8')
json_data = json.loads(json_str)
return json_data
@trace_error_decorator
def bark(api: str, title: str = "message", content: str = 'test', level: str = "active",
badge: int = 1, auto_copy: int = 1, sound: str = "", icon: str = "", group: str = "",
is_archive: int = 1, url: str = "") -> Union[Dict[str, Any], None]:
json_data = {
"title": title,
"body": content,
"level": level,
"badge": badge,
"autoCopy": auto_copy,
"sound": sound,
"icon": icon,
"group": group,
"isArchive": is_archive,
"url": url
}
data = json.dumps(json_data).encode('utf-8')
req = urllib.request.Request(api, data=data, headers=headers)
response = opener.open(req, timeout=10)
json_str = response.read().decode("utf-8")
json_data = json.loads(json_str)
if json_data['code'] != 200:
print(f'Bark推送失败, 失败信息：{json_data["message"]}')
return
return json_data
if __name__ == '__main__':
send_title = '直播通知' # 标题
send_content = '张三 开播了！' # 推送内容
# 钉钉推送通知
webhook_api = '' # 替换成自己Webhook链接,参考文档：https://open.dingtalk.com/document/robots/custom-robot-access
phone_number = '' # 被@用户的手机号码
# dingtalk(webhook_api, send_content, phone_number)
# 微信推送通知
# 替换成自己的单点推送接口,获取地址：https://xz.qqoq.net/#/admin/one
# 当然也可以使用其他平台API 如server酱 使用方法一样
xizhi_api = 'https://xizhi.qqoq.net/xxxxxxxxx.send'
# xizhi(xizhi_api, send_content)
# telegram推送通知
tg_token = '' # tg搜索"BotFather"获取的token值
tg_chat_id = 000000 # tg搜索"userinfobot"获取的chat_id值，即可发送推送消息给你自己，如果下面的是群组id则发送到群
# tg_bot(tg_chat_id, tg_token, send_content)
# email_message("", "", "", "", "", "")
bark_url = 'https://xxx.xxx.com/key/'
# bark(bark_url, send_title, send_content)

23
main.py
View File

@ -31,7 +31,7 @@ from douyinliverecorder.utils import (
logger, check_md5, update_config,
get_file_paths, remove_emojis
)
from douyinliverecorder.msg_push import (
from msg_push import (
dingtalk, xizhi, tg_bot, email_message, bark
)
@ -239,7 +239,7 @@ def adjust_max_request():
error_count = 0
def push_message(content: str) -> Union[str, list]:
def push_message(record_name, content: str) -> None:
push_functions = {
'微信': lambda: xizhi(xizhi_api_url, content),
'钉钉': lambda: dingtalk(dingtalk_api_url, content, dingtalk_phone_num),
@ -248,18 +248,15 @@ def push_message(content: str) -> Union[str, list]:
'BARK': lambda: bark(
bark_msg_api, title="直播录制通知", content=content, level=bark_msg_level, sound=bark_msg_ring),
}
push_pts = []
for platform, func in push_functions.items():
if platform in live_status_push.upper():
try:
result = func()
if result:
push_pts.append(platform)
print(f'提示信息：已经将[{record_name}]直播状态消息推送至你的{platform},'
f' 成功{len(result["success"])}, 失败{len(result["error"])}')
except Exception as e:
print(f"推送到{platform}失败: {e}")
return '、'.join(push_pts) if push_pts else ""
print(f"直播消息推送到{platform}失败: {e}")
def run_bash(command):
@ -750,9 +747,8 @@ def start_record(url_data: tuple, count_variable: int = -1):
push_content = (push_content.replace('[直播间名称]', record_name).
replace('[时间]', push_at))
push_pts = push_message(push_content.replace(r'\n', '\n'))
if push_pts:
print(f'提示信息：已经将[{record_name}]直播状态消息推送至你的{push_pts}')
push_message(record_name, push_content.replace(r'\n', '\n'))
start_pushed = False
else:
content = f"\r{record_name} 正在直播中..."
@ -766,9 +762,8 @@ def start_record(url_data: tuple, count_variable: int = -1):
push_content = (push_content.replace('[直播间名称]', record_name).
replace('[时间]', push_at))
push_pts = push_message(push_content.replace(r'\n', '\n'))
if push_pts:
print(f'提示信息：已经将[{record_name}]直播状态消息推送至你的{push_pts}')
push_message(record_name, push_content.replace(r'\n', '\n'))
start_pushed = True
if disable_record:

180
msg_push.py Normal file
View File

@ -0,0 +1,180 @@
# -*- coding: utf-8 -*-
"""
Author: Hmily
GitHub: https://github.com/ihmily
Date: 2023-09-03 19:18:36
Update: 2024-10-05 11:45:12
Copyright (c) 2023-2024 by Hmily, All Rights Reserved.
"""
from typing import Dict, Any, Optional, Union
import json
import urllib.request
import smtplib
from email.header import Header
from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText
no_proxy_handler = urllib.request.ProxyHandler({})
opener = urllib.request.build_opener(no_proxy_handler)
headers: Dict[str, str] = {'Content-Type': 'application/json'}
def dingtalk(url: str, content: str, number: Optional[str] = '') -> Union[Dict[str, Any], None]:
success = []
error = []
for api in url.replace('，', ',').split(','):
json_data = {
'msgtype': 'text',
'text': {
'content': content,
},
"at": {
"atMobiles": [
number # 添加这个手机号，可以被@通知（必须要在群里）
],
},
}
try:
data = json.dumps(json_data).encode('utf-8')
req = urllib.request.Request(api, data=data, headers=headers)
response = opener.open(req, timeout=10)
json_str = response.read().decode('utf-8')
json_data = json.loads(json_str)
if json_data['errcode'] == 0:
success.append(api)
else:
error.append(api)
print(f'钉钉推送失败, 推送地址：{api}, {json_data["errmsg"]}')
except Exception as e:
error.append(api)
print(f'钉钉推送失败, 推送地址：{api}, 错误信息:{e}')
return {"success": success, "error": error}
def xizhi(url: str, content: str, title: str = '直播间状态更新') -> Union[Dict[str, Any], None]:
success = []
error = []
for api in url.replace('，', ',').split(','):
json_data = {
'title': title,
'content': content
}
try:
data = json.dumps(json_data).encode('utf-8')
req = urllib.request.Request(url, data=data, headers=headers)
response = opener.open(req, timeout=10)
json_str = response.read().decode('utf-8')
json_data = json.loads(json_str)
if json_data['code'] == 200:
success.append(api)
else:
error.append(api)
print(f'微信推送失败, 推送地址：{api}, 失败信息：{json_data["msg"]}')
except Exception as e:
error.append(api)
print(f'微信推送失败, 推送地址：{api}, 错误信息:{e}')
return {"success": success, "error": error}
def email_message(mail_host: str, mail_pass: str, from_email: str, to_email: str, title: str, content: str) -> (
Union)[Dict[str, Any], None]:
receivers = to_email.replace('，', ',').split(',')
try:
message = MIMEMultipart()
message['From'] = "{}".format(from_email)
message['Subject'] = Header(title, 'utf-8')
if len(receivers) == 1:
message['To'] = receivers[0]
t_apart = MIMEText(content, 'plain', 'utf-8')
message.attach(t_apart)
smtp_obj = smtplib.SMTP_SSL(mail_host, 465)
smtp_obj.login(from_email, mail_pass)
smtp_obj.sendmail(from_email, receivers, message.as_string())
return {"success": receivers, "error": []}
except smtplib.SMTPException as e:
print(f'邮件推送失败, 推送邮箱：{to_email}, 错误信息:{e}')
return {"success": [], "error": receivers}
def tg_bot(chat_id: int, token: str, content: str) -> Dict[str, Any]:
try:
json_data = {
"chat_id": chat_id,
'text': content
}
url = f'https://api.telegram.org/bot{token}/sendMessage'
data = json.dumps(json_data).encode('utf-8')
req = urllib.request.Request(url, data=data, headers=headers)
response = urllib.request.urlopen(req, timeout=15)
json_str = response.read().decode('utf-8')
_json_data = json.loads(json_str)
return {"success": [1], "error": []}
except Exception as e:
print(f'tg推送失败, 聊天ID：{chat_id}, 错误信息:{e}')
return {"success": [], "error": [1]}
def bark(api: str, title: str = "message", content: str = 'test', level: str = "active",
badge: int = 1, auto_copy: int = 1, sound: str = "", icon: str = "", group: str = "",
is_archive: int = 1, url: str = "") -> Union[Dict[str, Any], None]:
success = []
error = []
for _api in api.replace('，', ',').split(','):
json_data = {
"title": title,
"body": content,
"level": level,
"badge": badge,
"autoCopy": auto_copy,
"sound": sound,
"icon": icon,
"group": group,
"isArchive": is_archive,
"url": url
}
try:
data = json.dumps(json_data).encode('utf-8')
req = urllib.request.Request(_api, data=data, headers=headers)
response = opener.open(req, timeout=10)
json_str = response.read().decode("utf-8")
json_data = json.loads(json_str)
if json_data['code'] == 200:
success.append(_api)
else:
error.append(_api)
print(f'Bark推送失败, 推送地址：{_api}, 失败信息：{json_data["message"]}')
except Exception as e:
error.append(api)
print(f'Bark推送失败, 推送地址：{_api}, 错误信息:{e}')
return {"success": success, "error": error}
if __name__ == '__main__':
send_title = '直播通知' # 标题
send_content = '张三 开播了！' # 推送内容
# 钉钉推送通知
webhook_api = '' # 替换成自己Webhook链接,参考文档：https://open.dingtalk.com/document/robots/custom-robot-access
phone_number = '' # 被@用户的手机号码
# dingtalk(webhook_api, send_content, phone_number)
# 微信推送通知
# 替换成自己的单点推送接口,获取地址：https://xz.qqoq.net/#/admin/one
# 当然也可以使用其他平台API 如server酱 使用方法一样
xizhi_api = 'https://xizhi.qqoq.net/xxxxxxxxx.send'
# xizhi(xizhi_api, send_content)
# telegram推送通知
tg_token = '' # tg搜索"BotFather"获取的token值
tg_chat_id = 000000 # tg搜索"userinfobot"获取的chat_id值，即可发送推送消息给你自己，如果下面的是群组id则发送到群
# tg_bot(tg_chat_id, tg_token, send_content)
# email_message("", "", "", "", "", "")
bark_url = 'https://xxx.xxx.com/key/'
# bark(bark_url, send_title, send_content)