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 custom push message configuration

Browse Source
This commit is contained in:
ihmily 2024-07-01 22:21:38 +08:00
parent 332e7f1927
commit a29d57fb7e
3 changed files with 23 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
config/config.ini
View File

@ -28,6 +28,8 @@ ts录制完成后自动增加生成m4a格式 = 否
钉钉通知@对象(填手机号) =
tgapi令牌 =
tg聊天id(个人或者群组id) =
自定义开播推送内容 =
自定义关播推送内容 =
只推送通知不录制（是/否） = 否
直播推送检测频率（秒） = 1800
开播推送开启（是/否）= 是

21
main.py
View File

@ -4,7 +4,7 @@
Author: Hmily
GitHub: https://github.com/ihmily
Date: 2023-07-17 23:52:05
Update: 2024-06-26 12:41:30
Update: 2024-07-01 22:19:30
Copyright (c) 2023-2024 by Hmily, All Rights Reserved.
Function: Record live stream video.
"""
@ -999,8 +999,12 @@ def start_record(url_data: tuple, count_variable: int = -1):
if start_pushed:
if over_show_push:
content = f"{record_name} 直播已结束！时间：{push_at}"
push_pts = push_message(content)
push_content = f"直播间状态更新：[直播间名称] 直播已结束！时间：[时间]"
if over_push_message_text:
push_content = over_push_message_text
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}')
start_pushed = False
@ -1008,11 +1012,14 @@ def start_record(url_data: tuple, count_variable: int = -1):
content = f"\r{record_name} 正在直播中..."
print(content)
# 推送通知
if live_status_push and not start_pushed:
if begin_show_push:
push_pts = push_message(
f"{content.split('...')[0]}，时间：{push_at}")
push_content = f"直播间状态更新：[直播间名称] 正在直播中，时间：[时间]"
if begin_push_message_text:
push_content = begin_push_message_text
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}')
start_pushed = True
@ -1640,6 +1647,8 @@ while True:
dingtalk_phone_num = read_config_value(config, '推送配置', '钉钉通知@对象(填手机号)', "")
tg_token = read_config_value(config, '推送配置', 'tgapi令牌', "")
tg_chat_id = read_config_value(config, '推送配置', 'tg聊天id(个人或者群组id)', "")
begin_push_message_text = read_config_value(config, '推送配置', '自定义开播推送内容', "")
over_push_message_text = read_config_value(config, '推送配置', '自定义关播推送内容', "")
disable_record = options.get(read_config_value(config, '推送配置', '只推送通知不录制（是/否）', "否"), False)
push_check_seconds = int(read_config_value(config, '推送配置', '直播推送检测频率（秒）', 1800))
begin_show_push = options.get(read_config_value(config, '推送配置', '开播推送开启（是/否）', "是"), True)

10
msg_push.py
View File

@ -4,7 +4,7 @@
Author: Hmily
GitHub: https://github.com/ihmily
Date: 2023-09-03 19:18:36
Update: 2024-01-16 02:31:36
Update: 2024-07-01 22:16:36
Copyright (c) 2023 by Hmily, All Rights Reserved.
"""
@ -23,7 +23,7 @@ def dingtalk(url: str, content: str, phone_number: Optional[str] = '') -> Dict[s
json_data = {
'msgtype': 'text',
'text': {
'content': '直播间状态更新：\n' + content,
'content': content,
},
"at": {
"atMobiles": [
@ -77,10 +77,12 @@ if __name__ == '__main__':
# dingtalk(webhook_api, content, phone_number)
# 微信推送通知
xizhi_api = 'https://xizhi.qqoq.net/XZa5a4a224987c88ab828acbcc0aa4c853.send' # 替换成自己的单点推送接口,获取地址：https://xz.qqoq.net/#/admin/one
# 替换成自己的单点推送接口,获取地址：https://xz.qqoq.net/#/admin/one
# 当然也可以使用其他平台API 如server酱 使用方法一样
xizhi_api = 'https://xizhi.qqoq.net/XZa5a4a224987c88ab828acbcc0aa4c853.send'
# xizhi(xizhi_api, content)
# telegram推送通知
token = '' # tg搜索"BotFather"获取的token值
chat_id = 000000 # tg搜索"userinfobot"获取的chat_id值，即可发送推送消息给你自己，如果下面的是群组id则发送到群
# tg_bot(chat_id, token, content)
# tg_bot(chat_id, token, content)