mirror of
https://github.com/ihmily/DouyinLiveRecorder.git
synced 2025-12-26 05:48:32 +08:00
Compare commits
85 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
add187f8d8 | ||
|
|
0333cb4a01 | ||
|
|
73857755a7 | ||
|
|
fec734ae74 | ||
|
|
853d03ea14 | ||
|
|
2fb7f7afd7 | ||
|
|
200e5b5b58 | ||
|
|
abb204e6e9 | ||
|
|
271a53621d | ||
|
|
d77760f3c9 | ||
|
|
9c913e23cc | ||
|
|
af37bf28f0 | ||
|
|
d4796409c7 | ||
|
|
93a12ab41d | ||
|
|
525b720627 | ||
|
|
e9f2a55ceb | ||
|
|
3965487746 | ||
|
|
e80f1e653a | ||
|
|
63787f1743 | ||
|
|
a328c6a1c5 | ||
|
|
be2c3a393f | ||
|
|
c7e3cf47ce | ||
|
|
199186fb09 | ||
|
|
5778ebc4b3 | ||
|
|
d9f985303a | ||
|
|
110d5bded4 | ||
|
|
e478d72e62 | ||
|
|
bcfc268c1c | ||
|
|
8e4e9b098f | ||
|
|
9f499f3fa6 | ||
|
|
ae8200e01c | ||
|
|
952eeb9b7c | ||
|
|
ba8979e4ee | ||
|
|
c157c08e5a | ||
|
|
effcfcc76f | ||
|
|
d787838ed2 | ||
|
|
86a822f0db | ||
|
|
51f79d70f8 | ||
|
|
d37d9f25ad | ||
|
|
c40a10235d | ||
|
|
d860023808 | ||
|
|
57739410f8 | ||
|
|
9bc629c1b0 | ||
|
|
2dd9c42767 | ||
|
|
fd06bc89da | ||
|
|
019b30203e | ||
|
|
0bd2a3a360 | ||
|
|
4fa3fbb773 | ||
|
|
ba046660fc | ||
|
|
f73ce7b68e | ||
|
|
9e494f4377 | ||
|
|
151a6a45c9 | ||
|
|
807f4b758e | ||
|
|
bf7381bf6c | ||
|
|
d055602e81 | ||
|
|
ef97e01dba | ||
|
|
e189640d3a | ||
|
|
7bb778a875 | ||
|
|
99ea143c78 | ||
|
|
52c0287150 | ||
|
|
246632de78 | ||
|
|
71fbf93ffe | ||
|
|
a61bdc852e | ||
|
|
a5d567092d | ||
|
|
272e2dd28e | ||
|
|
c9b2310fa4 | ||
|
|
46842c3b48 | ||
|
|
736943cbca | ||
|
|
5d3c295c7a | ||
|
|
8c30787353 | ||
|
|
73d9ee1334 | ||
|
|
2fa830307f | ||
|
|
bbb0c5ebaa | ||
|
|
fa92a4196f | ||
|
|
29a16ba00b | ||
|
|
d5de8bd77b | ||
|
|
0751107ae1 | ||
|
|
c637e5617c | ||
|
|
a00305cafb | ||
|
|
c9f26b116d | ||
|
|
4181e9745a | ||
|
|
1a8f2fe9bd | ||
|
|
0c97d0612f | ||
|
|
80f88f6420 | ||
|
|
9835374174 |
68
.github/workflows/build-image.yml
vendored
68
.github/workflows/build-image.yml
vendored
@ -4,45 +4,51 @@ on:
|
||||
push:
|
||||
tags:
|
||||
- '*'
|
||||
workflow_dispatch:
|
||||
inputs:
|
||||
tag_name:
|
||||
description: 'Tag name for the Docker image'
|
||||
required: false
|
||||
default: 'latest'
|
||||
|
||||
jobs:
|
||||
build_and_push:
|
||||
runs-on: ubuntu-latest
|
||||
|
||||
steps:
|
||||
- name: Checkout code
|
||||
uses: actions/checkout@v2
|
||||
- name: Checkout code
|
||||
uses: actions/checkout@v2
|
||||
|
||||
- name: Set up QEMU
|
||||
uses: docker/setup-qemu-action@v3
|
||||
- name: Set up QEMU
|
||||
uses: docker/setup-qemu-action@v3
|
||||
|
||||
- name: Set up Docker Buildx
|
||||
uses: docker/setup-buildx-action@v3
|
||||
- name: Set up Docker Buildx
|
||||
uses: docker/setup-buildx-action@v3
|
||||
|
||||
- name: Cache Docker layers
|
||||
uses: actions/cache@v2
|
||||
with:
|
||||
path: /tmp/.buildx-cache
|
||||
key: ${{ runner.os }}-buildx-${{ github.sha }}
|
||||
restore-keys: |
|
||||
${{ runner.os }}-buildx-
|
||||
- name: Cache Docker layers
|
||||
uses: actions/cache@v3
|
||||
with:
|
||||
path: /tmp/.buildx-cache
|
||||
key: ${{ runner.os }}-buildx-${{ github.sha }}
|
||||
restore-keys: |
|
||||
${{ runner.os }}-buildx-
|
||||
|
||||
- name: Log in to Docker Hub
|
||||
uses: docker/login-action@v2
|
||||
with:
|
||||
username: ${{ secrets.DOCKERHUB_USERNAME }}
|
||||
password: ${{ secrets.DOCKERHUB_PASSWORD }}
|
||||
registry: docker.io
|
||||
- name: Log in to Docker Hub
|
||||
uses: docker/login-action@v2
|
||||
with:
|
||||
username: ${{ secrets.DOCKERHUB_USERNAME }}
|
||||
password: ${{ secrets.DOCKERHUB_PASSWORD }}
|
||||
registry: docker.io
|
||||
|
||||
- name: Build and push Docker image
|
||||
uses: docker/build-push-action@v5
|
||||
with:
|
||||
context: .
|
||||
file: ./Dockerfile
|
||||
push: true
|
||||
tags: |
|
||||
ihmily/douyin-live-recorder:${{ github.ref_name }}
|
||||
ihmily/douyin-live-recorder:latest
|
||||
platforms: linux/amd64,linux/arm64
|
||||
cache-from: type=local,src=/tmp/.buildx-cache
|
||||
cache-to: type=local,dest=/tmp/.buildx-cache
|
||||
- name: Build and push Docker image
|
||||
uses: docker/build-push-action@v5
|
||||
with:
|
||||
context: .
|
||||
file: ./Dockerfile
|
||||
push: true
|
||||
tags: |
|
||||
ihmily/douyin-live-recorder:${{ github.event.inputs.tag_name || github.ref_name }}
|
||||
ihmily/douyin-live-recorder:latest
|
||||
platforms: linux/amd64,linux/arm64
|
||||
cache-from: type=local,src=/tmp/.buildx-cache
|
||||
cache-to: type=local,dest=/tmp/.buildx-cache
|
||||
|
||||
8
.gitignore
vendored
8
.gitignore
vendored
@ -81,10 +81,16 @@ target/
|
||||
profile_default/
|
||||
ipython_config.py
|
||||
|
||||
# DouyinLiveRecord
|
||||
backup_config/
|
||||
logs/
|
||||
node/
|
||||
node-v*.zip
|
||||
|
||||
# pyenv
|
||||
# For a library or package, you might want to ignore these files since the code is
|
||||
# intended to run in multiple environments; otherwise, check them in:
|
||||
# .python-version
|
||||
.python-version
|
||||
|
||||
# pipenv
|
||||
# According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
|
||||
|
||||
2
LICENSE
2
LICENSE
@ -1,6 +1,6 @@
|
||||
MIT License
|
||||
|
||||
Copyright (c) 2023 Hmily
|
||||
Copyright (c) 2025 Hmily
|
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
of this software and associated documentation files (the "Software"), to deal
|
||||
|
||||
192
README.md
192
README.md
@ -31,7 +31,7 @@
|
||||
- [x] 猫耳FM
|
||||
- [x] Look直播
|
||||
- [x] WinkTV
|
||||
- [x] FlexTV
|
||||
- [x] TTingLive(原Flextv)
|
||||
- [x] PopkonTV
|
||||
- [x] TwitCasting
|
||||
- [x] 百度直播
|
||||
@ -58,6 +58,13 @@
|
||||
- [x] 花猫直播
|
||||
- [x] Shopee
|
||||
- [x] Youtube
|
||||
- [x] 淘宝
|
||||
- [x] 京东
|
||||
- [x] Faceit
|
||||
- [x] 咪咕
|
||||
- [x] 连接直播
|
||||
- [x] 来秀直播
|
||||
- [x] Picarto
|
||||
- [ ] 更多平台正在更新中
|
||||
|
||||
</div>
|
||||
@ -77,6 +84,7 @@
|
||||
├── utils.py -> (contains utility functions)
|
||||
├── logger.py -> (logger handdle)
|
||||
├── room.py -> (get room info)
|
||||
├── ab_sign.py-> (generate dy token)
|
||||
├── /javascript -> (some decrypt code)
|
||||
├── main.py -> (main file)
|
||||
├── ffmpeg_install.py -> (ffmpeg install script)
|
||||
@ -116,54 +124,53 @@
|
||||
直播间链接示例：
|
||||
|
||||
```
|
||||
抖音：
|
||||
抖音:
|
||||
https://live.douyin.com/745964462470
|
||||
https://v.douyin.com/iQFeBnt/
|
||||
https://live.douyin.com/yall1102 （链接+抖音号）
|
||||
https://v.douyin.com/CeiU5cbX （主播主页地址）
|
||||
|
||||
TikTok：
|
||||
TikTok:
|
||||
https://www.tiktok.com/@pearlgaga88/live
|
||||
|
||||
快手：
|
||||
快手:
|
||||
https://live.kuaishou.com/u/yall1102
|
||||
|
||||
虎牙：
|
||||
虎牙:
|
||||
https://www.huya.com/52333
|
||||
|
||||
斗鱼：
|
||||
斗鱼:
|
||||
https://www.douyu.com/3637778?dyshid=
|
||||
https://www.douyu.com/topic/wzDBLS6?rid=4921614&dyshid=
|
||||
|
||||
YY:
|
||||
https://www.yy.com/22490906/22490906
|
||||
|
||||
B站：
|
||||
B站:
|
||||
https://live.bilibili.com/320
|
||||
|
||||
小红书（推荐使用主页地址)：
|
||||
https://www.xiaohongshu.com/user/profile/6330049c000000002303c7ed?appuid=5f3f478a00000000010005b3
|
||||
小红书（直播间分享地址):
|
||||
http://xhslink.com/xpJpfM
|
||||
|
||||
bigo直播：
|
||||
bigo直播:
|
||||
https://www.bigo.tv/cn/716418802
|
||||
|
||||
buled直播：
|
||||
buled直播:
|
||||
https://app.blued.cn/live?id=Mp6G2R
|
||||
|
||||
SOOP：
|
||||
SOOP:
|
||||
https://play.sooplive.co.kr/sw7love
|
||||
|
||||
网易cc：
|
||||
网易cc:
|
||||
https://cc.163.com/583946984
|
||||
|
||||
千度热播：
|
||||
千度热播:
|
||||
https://qiandurebo.com/web/video.php?roomnumber=33333
|
||||
|
||||
PandaTV：
|
||||
PandaTV:
|
||||
https://www.pandalive.co.kr/live/play/bara0109
|
||||
|
||||
猫耳FM：
|
||||
猫耳FM:
|
||||
https://fm.missevan.com/live/868895007
|
||||
|
||||
Look直播:
|
||||
@ -172,7 +179,7 @@ https://look.163.com/live?id=65108820&position=3
|
||||
WinkTV:
|
||||
https://www.winktv.co.kr/live/play/anjer1004
|
||||
|
||||
FlexTV:
|
||||
FlexTV(TTinglive)::
|
||||
https://www.flextv.co.kr/channels/593127/live
|
||||
|
||||
PopkonTV:
|
||||
@ -209,10 +216,10 @@ https://www.showroom-live.com/room/profile?room_id=480206 （主播主页地址
|
||||
Acfun:
|
||||
https://live.acfun.cn/live/179922
|
||||
|
||||
映客直播：
|
||||
映客直播:
|
||||
https://www.inke.cn/liveroom/index.html?uid=22954469&id=1720860391070904
|
||||
|
||||
音播直播：
|
||||
音播直播:
|
||||
https://live.ybw1666.com/800002949
|
||||
|
||||
知乎直播:
|
||||
@ -234,7 +241,7 @@ https://17.live/en/live/6302408
|
||||
https://www.lang.live/en-US/room/3349463
|
||||
|
||||
畅聊直播:
|
||||
https://www.tlclw.com/801044397
|
||||
https://live.tlclw.com/106188
|
||||
|
||||
飘飘直播:
|
||||
https://m.pp.weimipopo.com/live/preview.html?uid=91648673&anchorUid=91625862&app=plpl
|
||||
@ -253,12 +260,34 @@ https://sg.shp.ee/GmpXeuf?uid=1006401066&session=802458
|
||||
|
||||
Youtube:
|
||||
https://www.youtube.com/watch?v=cS6zS5hi1w0
|
||||
|
||||
淘宝(需cookie):
|
||||
https://tbzb.taobao.com/live?liveId=532359023188
|
||||
https://m.tb.cn/h.TWp0HTd
|
||||
|
||||
京东:
|
||||
https://3.cn/28MLBy-E
|
||||
|
||||
Faceit:
|
||||
https://www.faceit.com/zh/players/Compl1/stream
|
||||
|
||||
连接直播:
|
||||
https://show.lailianjie.com/10000258
|
||||
|
||||
咪咕直播:
|
||||
https://www.miguvideo.com/p/live/120000541321
|
||||
|
||||
来秀直播:
|
||||
https://www.imkktv.com/h5/share/video.html?uid=1845195&roomId=1710496
|
||||
|
||||
Picarto:
|
||||
https://www.picarto.tv/cuteavalanche
|
||||
```
|
||||
|
||||
 
|
||||
|
||||
## 🎃源码运行
|
||||
使用源码运行，前提要有**Python>=3.10**环境，如果没有请先自行安装Python，再执行下面步骤。
|
||||
使用源码运行，可参考下面的步骤。
|
||||
|
||||
1.首先拉取或手动下载本仓库项目代码
|
||||
|
||||
@ -270,9 +299,94 @@ git clone https://github.com/ihmily/DouyinLiveRecorder.git
|
||||
|
||||
```bash
|
||||
cd DouyinLiveRecorder
|
||||
pip3 install -r requirements.txt
|
||||
```
|
||||
|
||||
> [!TIP]
|
||||
> - 不论你是否已安装 **Python>=3.10** 环境, 都推荐使用 [**uv**](https://github.com/astral-sh/uv) 运行, 因为它可以自动管理虚拟环境和方便地管理 **Python** 版本, **不过这完全是可选的**<br />
|
||||
> 使用以下命令安装
|
||||
> ```bash
|
||||
> # 在 macOS 和 Linux 上安装 uv
|
||||
> curl -LsSf https://astral.sh/uv/install.sh | sh
|
||||
> ```
|
||||
> ```powershell
|
||||
> # 在 Windows 上安装 uv
|
||||
> powershell -ExecutionPolicy ByPass -c "irm https://astral.sh/uv/install.ps1 | iex"
|
||||
> ```
|
||||
> - 如果安装依赖速度太慢, 你可以考虑使用国内 pip 镜像源:<br />
|
||||
> 在 `pip` 命令使用 `-i` 参数指定, 如 `pip3 install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple`<br />
|
||||
> 或者在 `uv` 命令 `--index` 选项指定, 如 `uv sync --index https://pypi.tuna.tsinghua.edu.cn/simple`
|
||||
|
||||
<details>
|
||||
|
||||
<summary>如果已安装 <b>Python>=3.10</b> 环境</summary>
|
||||
|
||||
- :white_check_mark: 在虚拟环境中安装 (推荐)
|
||||
|
||||
1. 创建虚拟环境
|
||||
|
||||
- 使用系统已安装的 Python, 不使用 uv
|
||||
|
||||
```bash
|
||||
python -m venv .venv
|
||||
```
|
||||
|
||||
- 使用 uv, 默认使用系统 Python, 你可以添加 `--python` 选项指定 Python 版本而不使用系统 Python [uv官方文档](https://docs.astral.sh/uv/concepts/python-versions/)
|
||||
|
||||
```bash
|
||||
uv venv
|
||||
```
|
||||
|
||||
2. 在终端激活虚拟环境 (在未安装 uv 或你想要手动激活虚拟环境时执行, 若已安装 uv, 可以跳过这一步, uv 会自动激活并使用虚拟环境)
|
||||
|
||||
**Bash** 中
|
||||
```bash
|
||||
source .venv/Scripts/activate
|
||||
```
|
||||
|
||||
**Powershell** 中
|
||||
```powershell
|
||||
.venv\Scripts\activate.ps1
|
||||
```
|
||||
|
||||
**Windows CMD** 中
|
||||
```bat
|
||||
.venv\Scripts\activate.bat
|
||||
```
|
||||
|
||||
3. 安装依赖
|
||||
|
||||
```bash
|
||||
# 使用 pip (若安装太慢或失败, 可使用 `-i` 指定镜像源)
|
||||
pip3 install -U pip && pip3 install -r requirements.txt
|
||||
# 或者使用 uv (可使用 `--index` 指定镜像源)
|
||||
uv sync
|
||||
# 或者
|
||||
uv pip sync requirements.txt
|
||||
```
|
||||
|
||||
- :x: 在系统 Python 环境中安装 (不推荐)
|
||||
|
||||
```bash
|
||||
pip3 install -U pip && pip3 install -r requirements.txt
|
||||
```
|
||||
|
||||
</details>
|
||||
|
||||
<details>
|
||||
|
||||
<summary>如果未安装 <b>Python>=3.10</b> 环境</summary>
|
||||
|
||||
你可以使用 [**uv**](https://github.com/astral-sh/uv) 安装依赖
|
||||
|
||||
```bash
|
||||
# uv 将使用 3.10 及以上的最新 python 发行版自动创建并使用虚拟环境, 可使用 --python 选项指定 python 版本, 参见 https://docs.astral.sh/uv/reference/cli/#uv-sync--python 和 https://docs.astral.sh/uv/reference/cli/#uv-pip-sync--python
|
||||
uv sync
|
||||
# 或
|
||||
uv pip sync requirements.txt
|
||||
```
|
||||
|
||||
</details>
|
||||
|
||||
3.安装[FFmpeg](https://ffmpeg.org/download.html#build-linux)，如果是Windows系统，这一步可跳过。对于Linux系统，执行以下命令安装
|
||||
|
||||
CentOS执行
|
||||
@ -305,6 +419,12 @@ brew install ffmpeg
|
||||
|
||||
```python
|
||||
python main.py
|
||||
|
||||
```
|
||||
或
|
||||
|
||||
```bash
|
||||
uv run main.py
|
||||
```
|
||||
|
||||
其中Linux系统请使用`python3 main.py` 运行。
|
||||
@ -361,6 +481,13 @@ docker-compose stop
|
||||
|
||||
 
|
||||
|
||||
## 🤖相关项目
|
||||
|
||||
- StreamCap: https://github.com/ihmily/StreamCap
|
||||
- streamget: https://github.com/ihmily/streamget
|
||||
|
||||
 
|
||||
|
||||
## ❤️贡献者
|
||||
|
||||
   [](https://github.com/ihmily)
|
||||
@ -380,10 +507,31 @@ docker-compose stop
|
||||
|
||||
   [](https://github.com/HoratioShaw)
|
||||
[](https://github.com/nov30th)
|
||||
[](https://github.com/727155455)
|
||||
[](https://github.com/nixingshiguang)
|
||||
[](https://github.com/1411430556)
|
||||
[](https://github.com/Ovear)
|
||||
 
|
||||
|
||||
## ⏳提交日志
|
||||
|
||||
- 20251024
|
||||
- 修复抖音风控无法获取数据问题
|
||||
|
||||
- 新增soop.com录制支持
|
||||
|
||||
- 修复bigo录制
|
||||
|
||||
- 20250127
|
||||
- 新增淘宝、京东、faceit直播录制
|
||||
- 修复小红书直播流录制以及转码问题
|
||||
- 修复畅聊、VV星球、flexTV直播录制
|
||||
- 修复批量微信直播推送
|
||||
- 新增email发送ssl和port配置
|
||||
- 新增强制转h264配置
|
||||
- 更新ffmpeg版本
|
||||
- 重构包为异步函数！
|
||||
|
||||
- 20241130
|
||||
- 新增shopee、youtube直播录制
|
||||
- 新增支持自定义m3u8、flv地址录制
|
||||
|
||||
@ -21,6 +21,7 @@ language(zh_cn/en) = zh_cn
|
||||
录制空间剩余阈值(gb) = 1.0
|
||||
视频分段时间(秒) = 1800
|
||||
录制完成后自动转为mp4格式 = 是
|
||||
mp4格式重新编码为h264 = 否
|
||||
追加格式后删除原文件 = 是
|
||||
生成时间字幕文件 = 否
|
||||
是否录制完成后执行自定义脚本 = 否
|
||||
@ -29,29 +30,32 @@ language(zh_cn/en) = zh_cn
|
||||
额外使用代理录制的平台(逗号分隔) =
|
||||
|
||||
[推送配置]
|
||||
# 可选微信|钉钉|tg|邮箱|bark|ntfy 可填多个
|
||||
直播状态推送渠道 =
|
||||
钉钉推送接口链接 =
|
||||
微信推送接口链接 =
|
||||
bark推送接口链接 =
|
||||
# 可选微信|钉钉|tg|邮箱|bark|ntfy|pushplus 可填多个
|
||||
直播状态推送渠道 =
|
||||
钉钉推送接口链接 =
|
||||
微信推送接口链接 =
|
||||
bark推送接口链接 =
|
||||
bark推送中断级别 = active
|
||||
bark推送铃声 =
|
||||
钉钉通知@对象(填手机号) =
|
||||
bark推送铃声 =
|
||||
钉钉通知@对象(填手机号) =
|
||||
钉钉通知@全体(是/否) = 否
|
||||
tgapi令牌 =
|
||||
tg聊天id(个人或者群组id) =
|
||||
smtp邮件服务器 =
|
||||
邮箱登录账号 =
|
||||
发件人密码(授权码) =
|
||||
发件人邮箱 =
|
||||
发件人显示昵称 =
|
||||
收件人邮箱 =
|
||||
tgapi令牌 =
|
||||
tg聊天id(个人或者群组id) =
|
||||
smtp邮件服务器 =
|
||||
是否使用SMTP服务SSL加密(是/否) =
|
||||
SMTP邮件服务器端口 =
|
||||
邮箱登录账号 =
|
||||
发件人密码(授权码) =
|
||||
发件人邮箱 =
|
||||
发件人显示昵称 =
|
||||
收件人邮箱 =
|
||||
ntfy推送地址 = https://ntfy.sh/xxxx
|
||||
ntfy推送标签 = tada
|
||||
ntfy推送邮箱 =
|
||||
自定义推送标题 =
|
||||
自定义开播推送内容 =
|
||||
自定义关播推送内容 =
|
||||
ntfy推送邮箱 =
|
||||
pushplus推送token =
|
||||
自定义推送标题 =
|
||||
自定义开播推送内容 =
|
||||
自定义关播推送内容 =
|
||||
只推送通知不录制(是/否) = 否
|
||||
直播推送检测频率(秒) = 1800
|
||||
开播推送开启(是/否) = 是
|
||||
@ -60,60 +64,68 @@ ntfy推送邮箱 =
|
||||
[Cookie]
|
||||
# 录制抖音必填
|
||||
抖音cookie = ttwid=1%7CB1qls3GdnZhUov9o2NxOMxxYS2ff6OSvEWbv0ytbES4%7C1680522049%7C280d802d6d478e3e78d0c807f7c487e7ffec0ae4e5fdd6a0fe74c3c6af149511; my_rd=1; passport_csrf_token=3ab34460fa656183fccfb904b16ff742; passport_csrf_token_default=3ab34460fa656183fccfb904b16ff742; d_ticket=9f562383ac0547d0b561904513229d76c9c21; n_mh=hvnJEQ4Q5eiH74-84kTFUyv4VK8xtSrpRZG1AhCeFNI; store-region=cn-fj; store-region-src=uid; LOGIN_STATUS=1; __security_server_data_status=1; FORCE_LOGIN=%7B%22videoConsumedRemainSeconds%22%3A180%7D; pwa2=%223%7C0%7C3%7C0%22; download_guide=%223%2F20230729%2F0%22; volume_info=%7B%22isUserMute%22%3Afalse%2C%22isMute%22%3Afalse%2C%22volume%22%3A0.6%7D; strategyABtestKey=%221690824679.923%22; stream_recommend_feed_params=%22%7B%5C%22cookie_enabled%5C%22%3Atrue%2C%5C%22screen_width%5C%22%3A1536%2C%5C%22screen_height%5C%22%3A864%2C%5C%22browser_online%5C%22%3Atrue%2C%5C%22cpu_core_num%5C%22%3A8%2C%5C%22device_memory%5C%22%3A8%2C%5C%22downlink%5C%22%3A10%2C%5C%22effective_type%5C%22%3A%5C%224g%5C%22%2C%5C%22round_trip_time%5C%22%3A150%7D%22; VIDEO_FILTER_MEMO_SELECT=%7B%22expireTime%22%3A1691443863751%2C%22type%22%3Anull%7D; home_can_add_dy_2_desktop=%221%22; __live_version__=%221.1.1.2169%22; device_web_cpu_core=8; device_web_memory_size=8; xgplayer_user_id=346045893336; csrf_session_id=2e00356b5cd8544d17a0e66484946f28; odin_tt=724eb4dd23bc6ffaed9a1571ac4c757ef597768a70c75fef695b95845b7ffcd8b1524278c2ac31c2587996d058e03414595f0a4e856c53bd0d5e5f56dc6d82e24004dc77773e6b83ced6f80f1bb70627; __ac_nonce=064caded4009deafd8b89; __ac_signature=_02B4Z6wo00f01HLUuwwAAIDBh6tRkVLvBQBy9L-AAHiHf7; ttcid=2e9619ebbb8449eaa3d5a42d8ce88ec835; webcast_leading_last_show_time=1691016922379; webcast_leading_total_show_times=1; webcast_local_quality=sd; live_can_add_dy_2_desktop=%221%22; msToken=1JDHnVPw_9yTvzIrwb7cQj8dCMNOoesXbA_IooV8cezcOdpe4pzusZE7NB7tZn9TBXPr0ylxmv-KMs5rqbNUBHP4P7VBFUu0ZAht_BEylqrLpzgt3y5ne_38hXDOX8o=; msToken=jV_yeN1IQKUd9PlNtpL7k5vthGKcHo0dEh_QPUQhr8G3cuYv-Jbb4NnIxGDmhVOkZOCSihNpA2kvYtHiTW25XNNX_yrsv5FN8O6zm3qmCIXcEe0LywLn7oBO2gITEeg=; tt_scid=mYfqpfbDjqXrIGJuQ7q-DlQJfUSG51qG.KUdzztuGP83OjuVLXnQHjsz-BRHRJu4e986
|
||||
快手cookie =
|
||||
tiktok_cookie =
|
||||
虎牙cookie =
|
||||
斗鱼cookie =
|
||||
yy_cookie =
|
||||
b站cookie =
|
||||
小红书cookie =
|
||||
bigo_cookie =
|
||||
blued_cookie =
|
||||
sooplive_cookie =
|
||||
netease_cookie =
|
||||
千度热播_cookie =
|
||||
pandatv_cookie =
|
||||
猫耳fm_cookie =
|
||||
winktv_cookie =
|
||||
flextv_cookie =
|
||||
look_cookie =
|
||||
twitcasting_cookie =
|
||||
baidu_cookie =
|
||||
weibo_cookie =
|
||||
kugou_cookie =
|
||||
twitch_cookie =
|
||||
liveme_cookie =
|
||||
huajiao_cookie =
|
||||
liuxing_cookie =
|
||||
showroom_cookie =
|
||||
acfun_cookie =
|
||||
changliao_cookie =
|
||||
快手cookie =
|
||||
tiktok_cookie =
|
||||
虎牙cookie =
|
||||
斗鱼cookie =
|
||||
yy_cookie =
|
||||
b站cookie =
|
||||
小红书cookie =
|
||||
bigo_cookie =
|
||||
blued_cookie =
|
||||
sooplive_cookie =
|
||||
netease_cookie =
|
||||
千度热播_cookie =
|
||||
pandatv_cookie =
|
||||
猫耳fm_cookie =
|
||||
winktv_cookie =
|
||||
flextv_cookie =
|
||||
look_cookie =
|
||||
twitcasting_cookie =
|
||||
baidu_cookie =
|
||||
weibo_cookie =
|
||||
kugou_cookie =
|
||||
twitch_cookie =
|
||||
liveme_cookie =
|
||||
huajiao_cookie =
|
||||
liuxing_cookie =
|
||||
showroom_cookie =
|
||||
acfun_cookie =
|
||||
changliao_cookie =
|
||||
yinbo_cookie =
|
||||
yingke_cookie =
|
||||
zhihu_cookie =
|
||||
chzzk_cookie =
|
||||
haixiu_cookie =
|
||||
vvxqiu_cookie =
|
||||
17live_cookie =
|
||||
langlive_cookie =
|
||||
pplive_cookie =
|
||||
6room_cookie =
|
||||
lehaitv_cookie =
|
||||
huamao_cookie =
|
||||
shopee_cookie =
|
||||
youtube_cookie =
|
||||
yingke_cookie =
|
||||
zhihu_cookie =
|
||||
chzzk_cookie =
|
||||
haixiu_cookie =
|
||||
vvxqiu_cookie =
|
||||
17live_cookie =
|
||||
langlive_cookie =
|
||||
pplive_cookie =
|
||||
6room_cookie =
|
||||
lehaitv_cookie =
|
||||
huamao_cookie =
|
||||
shopee_cookie =
|
||||
youtube_cookie =
|
||||
taobao_cookie =
|
||||
jd_cookie =
|
||||
faceit_cookie =
|
||||
migu_cookie =
|
||||
lianjie_cookie =
|
||||
laixiu_cookie =
|
||||
picarto_cookie =
|
||||
|
||||
|
||||
[Authorization]
|
||||
popkontv_token =
|
||||
popkontv_token =
|
||||
|
||||
[账号密码]
|
||||
sooplive账号 =
|
||||
sooplive密码 =
|
||||
flextv账号 =
|
||||
flextv密码 =
|
||||
popkontv账号 =
|
||||
sooplive账号 =
|
||||
sooplive密码 =
|
||||
flextv账号 =
|
||||
flextv密码 =
|
||||
popkontv账号 =
|
||||
partner_code = P-00001
|
||||
popkontv密码 =
|
||||
popkontv密码 =
|
||||
twitcasting账号类型 = normal
|
||||
twitcasting账号 =
|
||||
twitcasting密码 =
|
||||
twitcasting账号 =
|
||||
twitcasting密码 =
|
||||
45
demo.py
45
demo.py
@ -1,9 +1,10 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
from douyinliverecorder.logger import logger
|
||||
from douyinliverecorder import spider
|
||||
import asyncio
|
||||
from src.logger import logger
|
||||
from src import spider
|
||||
|
||||
# 以下示例直播间链接不保证时效性，请自行查看链接是否能正常访问
|
||||
# Please note that the following example live room links may not be up-to-date;
|
||||
# Please note that the following example live room links may not be up-to-date
|
||||
LIVE_STREAM_CONFIG = {
|
||||
"douyin": {
|
||||
"url": "https://live.douyin.com/745964462470",
|
||||
@ -70,7 +71,7 @@ LIVE_STREAM_CONFIG = {
|
||||
"func": spider.get_winktv_stream_data,
|
||||
},
|
||||
"flextv": {
|
||||
"url": "https://www.flextv.co.kr/channels/593127/live",
|
||||
"url": "https://www.ttinglive.com/channels/685479/live",
|
||||
"func": spider.get_flextv_stream_data,
|
||||
},
|
||||
"looklive": {
|
||||
@ -142,8 +143,8 @@ LIVE_STREAM_CONFIG = {
|
||||
"func": spider.get_haixiu_stream_url,
|
||||
},
|
||||
"vvxqiu": {
|
||||
"url": "https://h5webcdn-pro.vvxqiu.com//activity/videoShare/videoShare.html?h5Server=https://h5p.vvxqiu.com"
|
||||
"&roomId=LP115924473&platformId=vvstar",
|
||||
"url": "https://h5webcdnp.vvxqiu.com//activity/videoShare/videoShare.html?h5Server=https://h5p.vvxqiu.com&"
|
||||
"roomId=LP115664695&platformId=vvstar",
|
||||
"func": spider.get_vvxqiu_stream_url,
|
||||
},
|
||||
"17live": {
|
||||
@ -177,15 +178,43 @@ LIVE_STREAM_CONFIG = {
|
||||
"youtube": {
|
||||
"url": "https://www.youtube.com/watch?v=cS6zS5hi1w0",
|
||||
"func": spider.get_youtube_stream_url,
|
||||
},
|
||||
"taobao": {
|
||||
"url": "https://m.tb.cn/h.TWp0HTd",
|
||||
"func": spider.get_taobao_stream_url,
|
||||
},
|
||||
"jd": {
|
||||
"url": "https://3.cn/28MLBy-E",
|
||||
"func": spider.get_jd_stream_url,
|
||||
},
|
||||
"faceit": {
|
||||
"url": "https://www.faceit.com/zh/players/Compl1/stream",
|
||||
"func": spider.get_faceit_stream_data,
|
||||
},
|
||||
"lianjie": {
|
||||
"url": "https://show.lailianjie.com/10000258",
|
||||
"func": spider.get_lianjie_stream_url,
|
||||
},
|
||||
"migu": {
|
||||
"url": "https://www.miguvideo.com/p/live/120000541321",
|
||||
"func": spider.get_migu_stream_url,
|
||||
},
|
||||
"laixiu": {
|
||||
"url": "https://www.imkktv.com/h5/share/video.html?uid=1845195&roomId=1710496",
|
||||
"func": spider.get_laixiu_stream_url,
|
||||
},
|
||||
"picarto": {
|
||||
"url": "https://www.picarto.tv/cuteavalanche",
|
||||
"func": spider.get_picarto_stream_url,
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
def test_live_stream(platform_name: str, proxy_addr=None) -> None:
|
||||
def test_live_stream(platform_name: str, proxy_addr=None, cookies=None) -> None:
|
||||
if platform_name in LIVE_STREAM_CONFIG:
|
||||
config = LIVE_STREAM_CONFIG[platform_name]
|
||||
try:
|
||||
stream_data = config['func'](config['url'], proxy_addr=proxy_addr)
|
||||
stream_data = asyncio.run(config['func'](config['url'], proxy_addr=proxy_addr, cookies=cookies))
|
||||
logger.debug(f"Stream data for {platform_name}: {stream_data}")
|
||||
except Exception as e:
|
||||
logger.error(f"Error fetching stream data for {platform_name}: {e}")
|
||||
|
||||
@ -1,126 +0,0 @@
|
||||
# -*- encoding: utf-8 -*-
|
||||
|
||||
"""
|
||||
Author: Hmily
|
||||
GitHub:https://github.com/ihmily
|
||||
Date: 2023-07-17 23:52:05
|
||||
Update: 2024-10-08 23:35:00
|
||||
Copyright (c) 2023 by Hmily, All Rights Reserved.
|
||||
"""
|
||||
import json
|
||||
import re
|
||||
import urllib.parse
|
||||
import execjs
|
||||
import requests
|
||||
import urllib.request
|
||||
from . import JS_SCRIPT_PATH
|
||||
|
||||
no_proxy_handler = urllib.request.ProxyHandler({})
|
||||
opener = urllib.request.build_opener(no_proxy_handler)
|
||||
|
||||
HEADERS = {
|
||||
'User-Agent': 'Mozilla/5.0 (Linux; Android 11; SAMSUNG SM-G973U) AppleWebKit/537.36 (KHTML, like Gecko) '
|
||||
'SamsungBrowser/14.2 Chrome/87.0.4280.141 Mobile Safari/537.36',
|
||||
'Accept-Language': 'zh-CN,zh;q=0.8,zh-TW;q=0.7,zh-HK;q=0.5,en-US;q=0.3,en;q=0.2',
|
||||
'Cookie': 's_v_web_id=verify_lk07kv74_QZYCUApD_xhiB_405x_Ax51_GYO9bUIyZQVf'
|
||||
}
|
||||
|
||||
HEADERS_PC = {
|
||||
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/116.0.5845.97 '
|
||||
'Safari/537.36 Core/1.116.438.400 QQBrowser/13.0.6070.400',
|
||||
'Cookie': 'sessionid=7494ae59ae06784454373ce25761e864; __ac_nonce=0670497840077ee4c9eb2; '
|
||||
'__ac_signature=_02B4Z6wo00f012DZczQAAIDCJJBb3EjnINdg-XeAAL8-db; '
|
||||
's_v_web_id=verify_m1ztgtjj_vuHnMLZD_iwZ9_4YO4_BdN1_7wLP3pyqXsf2; ',
|
||||
}
|
||||
|
||||
|
||||
# X-bogus算法
|
||||
def get_xbogus(url: str, headers: dict | None = None) -> str:
|
||||
if not headers or 'user-agent' not in (k.lower() for k in headers):
|
||||
headers = HEADERS
|
||||
query = urllib.parse.urlparse(url).query
|
||||
xbogus = execjs.compile(open(f'{JS_SCRIPT_PATH}/x-bogus.js').read()).call('sign', query, headers.get("User-Agent", "user-agent"))
|
||||
return xbogus
|
||||
|
||||
|
||||
# 获取房间ID和用户secID
|
||||
def get_sec_user_id(url: str, proxy_addr: str | None = None, headers: dict | None = None) -> tuple | None:
|
||||
if not headers or all(k.lower() not in ['user-agent', 'cookie'] for k in headers):
|
||||
headers = HEADERS
|
||||
|
||||
if proxy_addr:
|
||||
proxies = {
|
||||
'http': proxy_addr,
|
||||
'https': proxy_addr
|
||||
}
|
||||
response = requests.get(url, headers=headers, proxies=proxies, timeout=15)
|
||||
else:
|
||||
response = opener.open(url, timeout=15)
|
||||
redirect_url = response.url
|
||||
if 'reflow/' in redirect_url:
|
||||
sec_user_id = re.search(r'sec_user_id=([\w_\-]+)&', redirect_url).group(1)
|
||||
room_id = redirect_url.split('?')[0].rsplit('/', maxsplit=1)[1]
|
||||
return room_id, sec_user_id
|
||||
|
||||
|
||||
# 获取抖音号
|
||||
def get_unique_id(url: str, proxy_addr: str | None = None, headers: dict | None = None) -> str:
|
||||
if not headers or all(k.lower() not in ['user-agent', 'cookie'] for k in headers):
|
||||
headers = HEADERS_PC
|
||||
|
||||
if proxy_addr:
|
||||
proxies = {
|
||||
'http': proxy_addr,
|
||||
'https': proxy_addr
|
||||
}
|
||||
response = requests.get(url, headers=headers, proxies=proxies, timeout=15)
|
||||
else:
|
||||
response = opener.open(url, timeout=15)
|
||||
redirect_url = response.url
|
||||
sec_user_id = redirect_url.split('?')[0].rsplit('/', maxsplit=1)[1]
|
||||
resp = requests.get(f'https://www.douyin.com/user/{sec_user_id}', headers=headers)
|
||||
unique_id = re.findall(r'undefined\\"},\\"uniqueId\\":\\"(.*?)\\",\\"customVerify', resp.text)[-1]
|
||||
return unique_id
|
||||
|
||||
|
||||
# 获取直播间webID
|
||||
def get_live_room_id(room_id: str, sec_user_id: str, proxy_addr: str | None = None,
|
||||
params: dict | None = None, headers: dict | None = None) -> str:
|
||||
if not headers or all(k.lower() not in ['user-agent', 'cookie'] for k in headers):
|
||||
headers = HEADERS
|
||||
|
||||
if not params:
|
||||
params = {
|
||||
"verifyFp": "verify_lk07kv74_QZYCUApD_xhiB_405x_Ax51_GYO9bUIyZQVf",
|
||||
"type_id": "0",
|
||||
"live_id": "1",
|
||||
"room_id": room_id,
|
||||
"sec_user_id": sec_user_id,
|
||||
"app_id": "1128",
|
||||
"msToken": "wrqzbEaTlsxt52-vxyZo_mIoL0RjNi1ZdDe7gzEGMUTVh_HvmbLLkQrA_1HKVOa2C6gkxb6IiY6TY2z8enAkPEwGq--gM"
|
||||
"-me3Yudck2ailla5Q4osnYIHxd9dI4WtQ==",
|
||||
}
|
||||
api = f'https://webcast.amemv.com/webcast/room/reflow/info/?{urllib.parse.urlencode(params)}'
|
||||
xbogus = get_xbogus(api)
|
||||
api = api + "&X-Bogus=" + xbogus
|
||||
|
||||
if proxy_addr:
|
||||
proxies = {
|
||||
'http': proxy_addr,
|
||||
'https': proxy_addr
|
||||
}
|
||||
response = requests.get(api, headers=headers, proxies=proxies, timeout=15)
|
||||
json_str = response.text
|
||||
else:
|
||||
req = urllib.request.Request(api, headers=headers)
|
||||
response = opener.open(req, timeout=15)
|
||||
json_str = response.read().decode('utf-8')
|
||||
json_data = json.loads(json_str)
|
||||
return json_data['data']['room']['owner']['web_rid']
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
room_url = "https://v.douyin.com/iQLgKSj/"
|
||||
_room_id, sec_uid = get_sec_user_id(room_url)
|
||||
web_rid = get_live_room_id(_room_id, sec_uid)
|
||||
print("return web_rid:", web_rid)
|
||||
@ -15,11 +15,12 @@ import zipfile
|
||||
from pathlib import Path
|
||||
import requests
|
||||
from tqdm import tqdm
|
||||
from douyinliverecorder.logger import logger
|
||||
from src.logger import logger
|
||||
|
||||
current_platform = platform.system()
|
||||
execute_dir = os.path.split(os.path.realpath(sys.argv[0]))[0]
|
||||
current_env_path = os.environ.get('PATH')
|
||||
ffmpeg_path = os.path.join(execute_dir, 'ffmpeg')
|
||||
|
||||
|
||||
def unzip_file(zip_path: str | Path, extract_to: str | Path, delete: bool = True) -> None:
|
||||
@ -62,11 +63,11 @@ def get_lanzou_download_link(url: str, password: str | None = None) -> str | Non
|
||||
def install_ffmpeg_windows():
|
||||
try:
|
||||
logger.warning("ffmpeg is not installed.")
|
||||
logger.debug("Installing the stable version of ffmpeg for Windows...")
|
||||
ffmpeg_url = get_lanzou_download_link('https://wweb.lanzouv.com/in54b2gmj24b', 'e3ut')
|
||||
logger.debug("Installing the latest version of ffmpeg for Windows...")
|
||||
ffmpeg_url = get_lanzou_download_link('https://wweb.lanzouv.com/iHAc22ly3r3g', 'eots')
|
||||
if ffmpeg_url:
|
||||
full_file_name = 'ffmpeg-7.1.zip'
|
||||
version = 'v7.1'
|
||||
full_file_name = 'ffmpeg_latest_build_20250124.zip'
|
||||
version = 'v20250124'
|
||||
zip_file_path = Path(execute_dir) / full_file_name
|
||||
if Path(zip_file_path).exists():
|
||||
logger.debug("ffmpeg installation file already exists, start install...")
|
||||
@ -83,8 +84,7 @@ def install_ffmpeg_windows():
|
||||
f.write(data)
|
||||
|
||||
unzip_file(zip_file_path, execute_dir)
|
||||
extract_dir_path = str(zip_file_path).rsplit('.', maxsplit=1)[0]
|
||||
os.environ['PATH'] = execute_dir + '/' + extract_dir_path + os.pathsep + current_env_path
|
||||
os.environ['PATH'] = ffmpeg_path + os.pathsep + current_env_path
|
||||
result = subprocess.run(["ffmpeg", "-version"], capture_output=True)
|
||||
if result.returncode == 0:
|
||||
logger.debug('ffmpeg installation was successful')
|
||||
@ -122,7 +122,7 @@ def install_ffmpeg_linux():
|
||||
logger.debug("Trying to install the stable version of ffmpeg")
|
||||
result = subprocess.run(['yum', '-y', 'update'], capture_output=True)
|
||||
if result.returncode != 0:
|
||||
logger.error(f"Failed to update package lists using yum.")
|
||||
logger.error("Failed to update package lists using yum.")
|
||||
return False
|
||||
|
||||
result = subprocess.run(['yum', 'install', '-y', 'ffmpeg'], capture_output=True)
|
||||
|
||||
2
i18n.py
2
i18n.py
@ -20,7 +20,7 @@ else:
|
||||
locale_path = Path(execute_dir) / 'i18n'
|
||||
_tr = init_gettext(locale_path, 'zh_CN')
|
||||
original_print = builtins.print
|
||||
package_name = 'douyinliverecorder'
|
||||
package_name = 'src'
|
||||
|
||||
|
||||
def translated_print(*args, **kwargs):
|
||||
|
||||
55
msg_push.py
55
msg_push.py
@ -4,7 +4,7 @@
|
||||
Author: Hmily
|
||||
GitHub: https://github.com/ihmily
|
||||
Date: 2023-09-03 19:18:36
|
||||
Update: 2024-10-23 23:37:12
|
||||
Update: 2025-01-23 17:16:12
|
||||
Copyright (c) 2023-2024 by Hmily, All Rights Reserved.
|
||||
"""
|
||||
from typing import Dict, Any
|
||||
@ -67,7 +67,7 @@ def xizhi(url: str, title: str, content: str) -> Dict[str, Any]:
|
||||
}
|
||||
try:
|
||||
data = json.dumps(json_data).encode('utf-8')
|
||||
req = urllib.request.Request(url, data=data, headers=headers)
|
||||
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)
|
||||
@ -83,7 +83,7 @@ def xizhi(url: str, title: str, content: str) -> Dict[str, Any]:
|
||||
|
||||
|
||||
def send_email(email_host: str, login_email: str, email_pass: str, sender_email: str, sender_name: str,
|
||||
to_email: str, title: str, content: str) -> Dict[str, Any]:
|
||||
to_email: str, title: str, content: str, smtp_port: str = None, open_ssl: bool = True) -> Dict[str, Any]:
|
||||
receivers = to_email.replace('，', ',').split(',') if to_email.strip() else []
|
||||
|
||||
try:
|
||||
@ -97,7 +97,12 @@ def send_email(email_host: str, login_email: str, email_pass: str, sender_email:
|
||||
t_apart = MIMEText(content, 'plain', 'utf-8')
|
||||
message.attach(t_apart)
|
||||
|
||||
smtp_obj = smtplib.SMTP_SSL(email_host, 465)
|
||||
if open_ssl:
|
||||
smtp_port = int(smtp_port) or 465
|
||||
smtp_obj = smtplib.SMTP_SSL(email_host, smtp_port)
|
||||
else:
|
||||
smtp_port = int(smtp_port) or 25
|
||||
smtp_obj = smtplib.SMTP(email_host, smtp_port)
|
||||
smtp_obj.login(login_email, email_pass)
|
||||
smtp_obj.sendmail(sender_email, receivers, message.as_string())
|
||||
return {"success": receivers, "error": []}
|
||||
@ -208,6 +213,42 @@ def ntfy(api: str, title: str = "message", content: str = 'test', tags: str = 't
|
||||
return {"success": success, "error": error}
|
||||
|
||||
|
||||
def pushplus(token: str, title: str, content: str) -> Dict[str, Any]:
|
||||
"""
|
||||
PushPlus推送通知
|
||||
API文档: https://www.pushplus.plus/doc/
|
||||
"""
|
||||
success = []
|
||||
error = []
|
||||
token_list = token.replace('，', ',').split(',') if token.strip() else []
|
||||
|
||||
for _token in token_list:
|
||||
json_data = {
|
||||
'token': _token,
|
||||
'title': title,
|
||||
'content': content
|
||||
}
|
||||
|
||||
try:
|
||||
url = 'https://www.pushplus.plus/send'
|
||||
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.get('code') == 200:
|
||||
success.append(_token)
|
||||
else:
|
||||
error.append(_token)
|
||||
print(f'PushPlus推送失败, Token：{_token}, 失败信息：{json_data.get("msg", "未知错误")}')
|
||||
except Exception as e:
|
||||
error.append(_token)
|
||||
print(f'PushPlus推送失败, Token：{_token}, 错误信息:{e}')
|
||||
|
||||
return {"success": success, "error": error}
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
send_title = '直播通知' # 标题
|
||||
send_content = '张三 开播了！' # 推送内容
|
||||
@ -237,7 +278,7 @@ if __name__ == '__main__':
|
||||
# sender_name="",
|
||||
# to_email="",
|
||||
# title="",
|
||||
# content=""
|
||||
# content="",
|
||||
# )
|
||||
|
||||
bark_url = 'https://xxx.xxx.com/key/'
|
||||
@ -248,3 +289,7 @@ if __name__ == '__main__':
|
||||
title="直播推送",
|
||||
content="xxx已开播",
|
||||
)
|
||||
|
||||
# PushPlus推送通知
|
||||
pushplus_token = '' # 替换成自己的PushPlus Token，获取地址：https://www.pushplus.plus/
|
||||
# pushplus(pushplus_token, send_title, send_content)
|
||||
|
||||
@ -1,4 +0,0 @@
|
||||
[virtualenvs]
|
||||
in-project = true
|
||||
create = true
|
||||
prefer-active-python = true
|
||||
@ -1,26 +1,23 @@
|
||||
[project]
|
||||
requires-python = ">=3.10"
|
||||
|
||||
[build-system]
|
||||
requires = ["poetry-core>=1.0.0"]
|
||||
build-backend = "poetry.core.masonry.api"
|
||||
|
||||
[tool.poetry]
|
||||
name = "douyinliverecorder"
|
||||
version = "3.0.9"
|
||||
description = "An easy tool for recording live streams"
|
||||
authors = ["Hmily"]
|
||||
license = "MIT"
|
||||
name = "DouyinLiveRecorder"
|
||||
version = "4.0.7"
|
||||
description = "可循环值守和多人录制的直播录制软件, 支持抖音、TikTok、Youtube、快手、虎牙、斗鱼、B站、小红书、pandatv、sooplive、flextv、popkontv、twitcasting、winktv、百度、微博、酷狗、17Live、Twitch、Acfun、CHZZK、shopee等40+平台直播录制"
|
||||
readme = "README.md"
|
||||
homepage = "https://github.com/ihmily/DouyinLiveRecorder"
|
||||
repository = "https://github.com/ihmily/DouyinLiveRecorder"
|
||||
keywords = ["douyin", "live", "recorder"]
|
||||
authors = [{name = "Hmily"}]
|
||||
license = { text = "MIT" }
|
||||
requires-python = ">=3.10"
|
||||
dependencies = [
|
||||
"requests>=2.31.0",
|
||||
"loguru>=0.7.3",
|
||||
"pycryptodome>=3.20.0",
|
||||
"distro>=1.9.0",
|
||||
"tqdm>=4.67.1",
|
||||
"httpx[http2]>=0.28.1",
|
||||
"PyExecJS>=1.5.1"
|
||||
]
|
||||
|
||||
[tool.poetry.dependencies]
|
||||
python = "^3.10"
|
||||
requests = "^2.25.1"
|
||||
PyExecJS = "^1.5.1"
|
||||
loguru = "^0.5.3"
|
||||
pycryptodome = "^3.10.1"
|
||||
distro = "^1.9.0"
|
||||
tqdm = "^4.66.5"
|
||||
[project.urls]
|
||||
"Homepage" = "https://github.com/ihmily/DouyinLiveRecorder"
|
||||
"Documentation" = "https://github.com/ihmily/DouyinLiveRecorder"
|
||||
"Repository" = "https://github.com/ihmily/DouyinLiveRecorder"
|
||||
"Issues" = "https://github.com/ihmily/DouyinLiveRecorder/issues"
|
||||
|
||||
@ -1,6 +1,7 @@
|
||||
requests
|
||||
PyExecJS
|
||||
loguru==0.7.2
|
||||
pycryptodome==3.20.0
|
||||
distro==1.9.0
|
||||
tqdm==4.66.5
|
||||
requests>=2.31.0
|
||||
loguru>=0.7.3
|
||||
pycryptodome>=3.20.0
|
||||
distro>=1.9.0
|
||||
tqdm>=4.67.1
|
||||
httpx[http2]>=0.28.1
|
||||
PyExecJS>=1.5.1
|
||||
31
setup.py
31
setup.py
@ -1,31 +0,0 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
from setuptools import setup, find_packages
|
||||
|
||||
setup(
|
||||
name='douyinliverecorder',
|
||||
version='3.0.9',
|
||||
author='Hmily',
|
||||
description='An easy tool for recording live streams',
|
||||
long_description=open('README.md', encoding='utf-8').read(),
|
||||
long_description_content_type='text/markdown',
|
||||
url='https://github.com/ihmily/DouyinLiveRecorder',
|
||||
packages=find_packages(),
|
||||
install_requires=[
|
||||
'requests',
|
||||
'PyExecJS',
|
||||
'loguru',
|
||||
'pycryptodome',
|
||||
'distro',
|
||||
'tqdm'
|
||||
],
|
||||
classifiers=[
|
||||
'Development Status :: 3 - Alpha',
|
||||
'Intended Audience :: Developers',
|
||||
'Programming Language :: Python :: 3',
|
||||
'Programming Language :: Python :: 3 :: Only',
|
||||
'Programming Language :: Python :: 3.10',
|
||||
'Programming Language :: Python :: 3.11',
|
||||
'Programming Language :: Python :: 3.12',
|
||||
'Programming Language :: Python :: 3.13',
|
||||
]
|
||||
)
|
||||
454
src/ab_sign.py
Normal file
454
src/ab_sign.py
Normal file
@ -0,0 +1,454 @@
|
||||
# -*- encoding: utf-8 -*-
|
||||
import math
|
||||
import time
|
||||
|
||||
|
||||
def rc4_encrypt(plaintext: str, key: str) -> str:
|
||||
# 初始化状态数组
|
||||
s = list(range(256))
|
||||
|
||||
# 使用密钥对状态数组进行置换
|
||||
j = 0
|
||||
for i in range(256):
|
||||
j = (j + s[i] + ord(key[i % len(key)])) % 256
|
||||
s[i], s[j] = s[j], s[i]
|
||||
|
||||
# 生成密钥流并加密
|
||||
i = j = 0
|
||||
result = []
|
||||
for char in plaintext:
|
||||
i = (i + 1) % 256
|
||||
j = (j + s[i]) % 256
|
||||
s[i], s[j] = s[j], s[i]
|
||||
t = (s[i] + s[j]) % 256
|
||||
result.append(chr(s[t] ^ ord(char)))
|
||||
|
||||
return ''.join(result)
|
||||
|
||||
|
||||
def left_rotate(x: int, n: int) -> int:
|
||||
n %= 32
|
||||
return ((x << n) | (x >> (32 - n))) & 0xFFFFFFFF
|
||||
|
||||
|
||||
def get_t_j(j: int) -> int:
|
||||
if 0 <= j < 16:
|
||||
return 2043430169 # 0x79CC4519
|
||||
elif 16 <= j < 64:
|
||||
return 2055708042 # 0x7A879D8A
|
||||
else:
|
||||
raise ValueError("invalid j for constant Tj")
|
||||
|
||||
|
||||
def ff_j(j: int, x: int, y: int, z: int) -> int:
|
||||
if 0 <= j < 16:
|
||||
return (x ^ y ^ z) & 0xFFFFFFFF
|
||||
elif 16 <= j < 64:
|
||||
return ((x & y) | (x & z) | (y & z)) & 0xFFFFFFFF
|
||||
else:
|
||||
raise ValueError("invalid j for bool function FF")
|
||||
|
||||
|
||||
def gg_j(j: int, x: int, y: int, z: int) -> int:
|
||||
if 0 <= j < 16:
|
||||
return (x ^ y ^ z) & 0xFFFFFFFF
|
||||
elif 16 <= j < 64:
|
||||
return ((x & y) | (~x & z)) & 0xFFFFFFFF
|
||||
else:
|
||||
raise ValueError("invalid j for bool function GG")
|
||||
|
||||
|
||||
class SM3:
|
||||
def __init__(self):
|
||||
self.reg = []
|
||||
self.chunk = []
|
||||
self.size = 0
|
||||
self.reset()
|
||||
|
||||
def reset(self):
|
||||
# 初始化寄存器值 - 修正为与JS版本相同的值
|
||||
self.reg = [
|
||||
1937774191, 1226093241, 388252375, 3666478592,
|
||||
2842636476, 372324522, 3817729613, 2969243214
|
||||
]
|
||||
self.chunk = []
|
||||
self.size = 0
|
||||
|
||||
def write(self, data):
|
||||
# 将输入转换为字节数组
|
||||
if isinstance(data, str):
|
||||
# 直接转换为UTF-8字节列表
|
||||
a = list(data.encode('utf-8'))
|
||||
else:
|
||||
a = data
|
||||
|
||||
self.size += len(a)
|
||||
f = 64 - len(self.chunk)
|
||||
|
||||
if len(a) < f:
|
||||
# 如果数据长度小于剩余空间，直接添加
|
||||
self.chunk.extend(a)
|
||||
else:
|
||||
# 否则分块处理
|
||||
self.chunk.extend(a[:f])
|
||||
|
||||
while len(self.chunk) >= 64:
|
||||
self._compress(self.chunk)
|
||||
if f < len(a):
|
||||
self.chunk = a[f:min(f + 64, len(a))]
|
||||
else:
|
||||
self.chunk = []
|
||||
f += 64
|
||||
|
||||
def _fill(self):
|
||||
# 计算比特长度
|
||||
bit_length = 8 * self.size
|
||||
|
||||
# 添加填充位
|
||||
padding_pos = len(self.chunk)
|
||||
self.chunk.append(0x80)
|
||||
padding_pos = (padding_pos + 1) % 64
|
||||
|
||||
# 如果剩余空间不足8字节，则填充到下一个块
|
||||
if 64 - padding_pos < 8:
|
||||
padding_pos -= 64
|
||||
|
||||
# 填充0直到剩余8字节用于存储长度
|
||||
while padding_pos < 56:
|
||||
self.chunk.append(0)
|
||||
padding_pos += 1
|
||||
|
||||
# 添加消息长度（高32位）
|
||||
high_bits = bit_length // 4294967296
|
||||
for i in range(4):
|
||||
self.chunk.append((high_bits >> (8 * (3 - i))) & 0xFF)
|
||||
|
||||
# 添加消息长度（低32位）
|
||||
for i in range(4):
|
||||
self.chunk.append((bit_length >> (8 * (3 - i))) & 0xFF)
|
||||
|
||||
def _compress(self, data):
|
||||
if len(data) < 64:
|
||||
raise ValueError("compress error: not enough data")
|
||||
else:
|
||||
# 消息扩展
|
||||
w = [0] * 132
|
||||
|
||||
# 将字节数组转换为字
|
||||
for t in range(16):
|
||||
w[t] = (data[4 * t] << 24) | (data[4 * t + 1] << 16) | (data[4 * t + 2] << 8) | data[4 * t + 3]
|
||||
w[t] &= 0xFFFFFFFF
|
||||
|
||||
# 消息扩展
|
||||
for j in range(16, 68):
|
||||
a = w[j - 16] ^ w[j - 9] ^ left_rotate(w[j - 3], 15)
|
||||
a = a ^ left_rotate(a, 15) ^ left_rotate(a, 23)
|
||||
w[j] = (a ^ left_rotate(w[j - 13], 7) ^ w[j - 6]) & 0xFFFFFFFF
|
||||
|
||||
# 计算w'
|
||||
for j in range(64):
|
||||
w[j + 68] = (w[j] ^ w[j + 4]) & 0xFFFFFFFF
|
||||
|
||||
# 压缩
|
||||
a, b, c, d, e, f, g, h = self.reg
|
||||
|
||||
for j in range(64):
|
||||
ss1 = left_rotate((left_rotate(a, 12) + e + left_rotate(get_t_j(j), j)) & 0xFFFFFFFF, 7)
|
||||
ss2 = ss1 ^ left_rotate(a, 12)
|
||||
tt1 = (ff_j(j, a, b, c) + d + ss2 + w[j + 68]) & 0xFFFFFFFF
|
||||
tt2 = (gg_j(j, e, f, g) + h + ss1 + w[j]) & 0xFFFFFFFF
|
||||
|
||||
d = c
|
||||
c = left_rotate(b, 9)
|
||||
b = a
|
||||
a = tt1
|
||||
h = g
|
||||
g = left_rotate(f, 19)
|
||||
f = e
|
||||
e = (tt2 ^ left_rotate(tt2, 9) ^ left_rotate(tt2, 17)) & 0xFFFFFFFF
|
||||
|
||||
# 更新寄存器
|
||||
self.reg[0] ^= a
|
||||
self.reg[1] ^= b
|
||||
self.reg[2] ^= c
|
||||
self.reg[3] ^= d
|
||||
self.reg[4] ^= e
|
||||
self.reg[5] ^= f
|
||||
self.reg[6] ^= g
|
||||
self.reg[7] ^= h
|
||||
|
||||
def sum(self, data=None, output_format=None):
|
||||
"""
|
||||
计算哈希值
|
||||
"""
|
||||
# 如果提供了输入，则重置并写入
|
||||
if data is not None:
|
||||
self.reset()
|
||||
self.write(data)
|
||||
|
||||
self._fill()
|
||||
|
||||
# 分块压缩
|
||||
for f in range(0, len(self.chunk), 64):
|
||||
self._compress(self.chunk[f:f + 64])
|
||||
|
||||
if output_format == 'hex':
|
||||
# 十六进制输出
|
||||
result = ''.join(f'{val:08x}' for val in self.reg)
|
||||
else:
|
||||
# 字节数组输出
|
||||
result = []
|
||||
for f in range(8):
|
||||
c = self.reg[f]
|
||||
result.append((c >> 24) & 0xFF)
|
||||
result.append((c >> 16) & 0xFF)
|
||||
result.append((c >> 8) & 0xFF)
|
||||
result.append(c & 0xFF)
|
||||
|
||||
self.reset()
|
||||
return result
|
||||
|
||||
|
||||
def result_encrypt(long_str: str, num: str | None = None) -> str:
|
||||
# 魔改base64编码表
|
||||
encoding_tables = {
|
||||
"s0": "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/=",
|
||||
"s1": "Dkdpgh4ZKsQB80/Mfvw36XI1R25+WUAlEi7NLboqYTOPuzmFjJnryx9HVGcaStCe=",
|
||||
"s2": "Dkdpgh4ZKsQB80/Mfvw36XI1R25-WUAlEi7NLboqYTOPuzmFjJnryx9HVGcaStCe=",
|
||||
"s3": "ckdp1h4ZKsUB80/Mfvw36XIgR25+WQAlEi7NLboqYTOPuzmFjJnryx9HVGDaStCe",
|
||||
"s4": "Dkdpgh2ZmsQB80/MfvV36XI1R45-WUAlEixNLwoqYTOPuzKFjJnry79HbGcaStCe"
|
||||
}
|
||||
|
||||
# 位移常量
|
||||
masks = [16515072, 258048, 4032, 63] # 对应 0, 1, 2 的掩码，添加63作为第四个掩码
|
||||
shifts = [18, 12, 6, 0] # 对应的位移量
|
||||
|
||||
encoding_table = encoding_tables[num]
|
||||
|
||||
result = ""
|
||||
round_num = 0
|
||||
long_int = get_long_int(round_num, long_str)
|
||||
|
||||
total_chars = math.ceil(len(long_str) / 3 * 4)
|
||||
|
||||
for i in range(total_chars):
|
||||
# 每4个字符处理一组3字节
|
||||
if i // 4 != round_num:
|
||||
round_num += 1
|
||||
long_int = get_long_int(round_num, long_str)
|
||||
|
||||
# 计算当前位置的索引
|
||||
index = i % 4
|
||||
|
||||
# 使用掩码和位移提取6位值
|
||||
char_index = (long_int & masks[index]) >> shifts[index]
|
||||
|
||||
result += encoding_table[char_index]
|
||||
|
||||
return result
|
||||
|
||||
|
||||
def get_long_int(round_num: int, long_str: str) -> int:
|
||||
round_num = round_num * 3
|
||||
|
||||
# 获取字符串中的字符，如果超出范围则使用0
|
||||
char1 = ord(long_str[round_num]) if round_num < len(long_str) else 0
|
||||
char2 = ord(long_str[round_num + 1]) if round_num + 1 < len(long_str) else 0
|
||||
char3 = ord(long_str[round_num + 2]) if round_num + 2 < len(long_str) else 0
|
||||
|
||||
return (char1 << 16) | (char2 << 8) | char3
|
||||
|
||||
|
||||
def gener_random(random_num: int, option: list[int]) -> list[int]:
|
||||
byte1 = random_num & 255
|
||||
byte2 = (random_num >> 8) & 255
|
||||
|
||||
return [
|
||||
(byte1 & 170) | (option[0] & 85), # 偶数位与option[0]的奇数位合并
|
||||
(byte1 & 85) | (option[0] & 170), # 奇数位与option[0]的偶数位合并
|
||||
(byte2 & 170) | (option[1] & 85), # 偶数位与option[1]的奇数位合并
|
||||
(byte2 & 85) | (option[1] & 170), # 奇数位与option[1]的偶数位合并
|
||||
]
|
||||
|
||||
|
||||
def generate_random_str() -> str:
|
||||
"""
|
||||
生成随机字符串
|
||||
|
||||
Returns:
|
||||
随机字符串
|
||||
"""
|
||||
# 使用与JS版本相同的固定随机值
|
||||
random_values = [0.123456789, 0.987654321, 0.555555555]
|
||||
|
||||
# 生成三组随机字节并合并
|
||||
random_bytes = []
|
||||
random_bytes.extend(gener_random(int(random_values[0] * 10000), [3, 45]))
|
||||
random_bytes.extend(gener_random(int(random_values[1] * 10000), [1, 0]))
|
||||
random_bytes.extend(gener_random(int(random_values[2] * 10000), [1, 5]))
|
||||
|
||||
return ''.join(chr(b) for b in random_bytes)
|
||||
|
||||
|
||||
def generate_rc4_bb_str(url_search_params: str, user_agent: str, window_env_str: str,
|
||||
suffix: str = "cus", arguments: list[int] | None = None) -> str:
|
||||
if arguments is None:
|
||||
arguments = [0, 1, 14]
|
||||
|
||||
sm3 = SM3()
|
||||
start_time = int(time.time() * 1000)
|
||||
|
||||
# 三次加密处理
|
||||
# 1: url_search_params两次sm3之的结果
|
||||
url_search_params_list = sm3.sum(sm3.sum(url_search_params + suffix))
|
||||
# 2: 对后缀两次sm3之的结果
|
||||
cus = sm3.sum(sm3.sum(suffix))
|
||||
# 3: 对ua处理之后的结果
|
||||
ua_key = chr(0) + chr(1) + chr(14) # [1/256, 1, 14]
|
||||
ua = sm3.sum(result_encrypt(
|
||||
rc4_encrypt(user_agent, ua_key),
|
||||
"s3"
|
||||
))
|
||||
|
||||
end_time = start_time + 100
|
||||
|
||||
# 构建配置对象
|
||||
b = {
|
||||
8: 3,
|
||||
10: end_time,
|
||||
15: {
|
||||
"aid": 6383,
|
||||
"pageId": 110624,
|
||||
"boe": False,
|
||||
"ddrt": 7,
|
||||
"paths": {
|
||||
"include": [{} for _ in range(7)],
|
||||
"exclude": []
|
||||
},
|
||||
"track": {
|
||||
"mode": 0,
|
||||
"delay": 300,
|
||||
"paths": []
|
||||
},
|
||||
"dump": True,
|
||||
"rpU": "hwj"
|
||||
},
|
||||
16: start_time,
|
||||
18: 44,
|
||||
19: [1, 0, 1, 5],
|
||||
}
|
||||
|
||||
def split_to_bytes(num: int) -> list[int]:
|
||||
return [
|
||||
(num >> 24) & 255,
|
||||
(num >> 16) & 255,
|
||||
(num >> 8) & 255,
|
||||
num & 255
|
||||
]
|
||||
|
||||
# 处理时间戳
|
||||
start_time_bytes = split_to_bytes(b[16])
|
||||
b[20] = start_time_bytes[0]
|
||||
b[21] = start_time_bytes[1]
|
||||
b[22] = start_time_bytes[2]
|
||||
b[23] = start_time_bytes[3]
|
||||
b[24] = int(b[16] / 256 / 256 / 256 / 256) & 255
|
||||
b[25] = int(b[16] / 256 / 256 / 256 / 256 / 256) & 255
|
||||
|
||||
# 处理Arguments参数
|
||||
arg0_bytes = split_to_bytes(arguments[0])
|
||||
b[26] = arg0_bytes[0]
|
||||
b[27] = arg0_bytes[1]
|
||||
b[28] = arg0_bytes[2]
|
||||
b[29] = arg0_bytes[3]
|
||||
|
||||
b[30] = int(arguments[1] / 256) & 255
|
||||
b[31] = (arguments[1] % 256) & 255
|
||||
|
||||
arg1_bytes = split_to_bytes(arguments[1])
|
||||
b[32] = arg1_bytes[0]
|
||||
b[33] = arg1_bytes[1]
|
||||
|
||||
arg2_bytes = split_to_bytes(arguments[2])
|
||||
b[34] = arg2_bytes[0]
|
||||
b[35] = arg2_bytes[1]
|
||||
b[36] = arg2_bytes[2]
|
||||
b[37] = arg2_bytes[3]
|
||||
|
||||
# 处理加密结果
|
||||
b[38] = url_search_params_list[21]
|
||||
b[39] = url_search_params_list[22]
|
||||
b[40] = cus[21]
|
||||
b[41] = cus[22]
|
||||
b[42] = ua[23]
|
||||
b[43] = ua[24]
|
||||
|
||||
# 处理结束时间
|
||||
end_time_bytes = split_to_bytes(b[10])
|
||||
b[44] = end_time_bytes[0]
|
||||
b[45] = end_time_bytes[1]
|
||||
b[46] = end_time_bytes[2]
|
||||
b[47] = end_time_bytes[3]
|
||||
b[48] = b[8]
|
||||
b[49] = int(b[10] / 256 / 256 / 256 / 256) & 255
|
||||
b[50] = int(b[10] / 256 / 256 / 256 / 256 / 256) & 255
|
||||
|
||||
# 处理配置项
|
||||
b[51] = b[15]['pageId']
|
||||
|
||||
page_id_bytes = split_to_bytes(b[15]['pageId'])
|
||||
b[52] = page_id_bytes[0]
|
||||
b[53] = page_id_bytes[1]
|
||||
b[54] = page_id_bytes[2]
|
||||
b[55] = page_id_bytes[3]
|
||||
|
||||
b[56] = b[15]['aid']
|
||||
b[57] = b[15]['aid'] & 255
|
||||
b[58] = (b[15]['aid'] >> 8) & 255
|
||||
b[59] = (b[15]['aid'] >> 16) & 255
|
||||
b[60] = (b[15]['aid'] >> 24) & 255
|
||||
|
||||
# 处理环境信息
|
||||
window_env_list = [ord(char) for char in window_env_str]
|
||||
b[64] = len(window_env_list)
|
||||
b[65] = b[64] & 255
|
||||
b[66] = (b[64] >> 8) & 255
|
||||
|
||||
b[69] = 0
|
||||
b[70] = 0
|
||||
b[71] = 0
|
||||
|
||||
# 计算校验和
|
||||
b[72] = b[18] ^ b[20] ^ b[26] ^ b[30] ^ b[38] ^ b[40] ^ b[42] ^ b[21] ^ b[27] ^ b[31] ^ \
|
||||
b[35] ^ b[39] ^ b[41] ^ b[43] ^ b[22] ^ b[28] ^ b[32] ^ b[36] ^ b[23] ^ b[29] ^ \
|
||||
b[33] ^ b[37] ^ b[44] ^ b[45] ^ b[46] ^ b[47] ^ b[48] ^ b[49] ^ b[50] ^ b[24] ^ \
|
||||
b[25] ^ b[52] ^ b[53] ^ b[54] ^ b[55] ^ b[57] ^ b[58] ^ b[59] ^ b[60] ^ b[65] ^ \
|
||||
b[66] ^ b[70] ^ b[71]
|
||||
|
||||
# 构建最终字节数组
|
||||
bb = [
|
||||
b[18], b[20], b[52], b[26], b[30], b[34], b[58], b[38], b[40], b[53], b[42], b[21],
|
||||
b[27], b[54], b[55], b[31], b[35], b[57], b[39], b[41], b[43], b[22], b[28], b[32],
|
||||
b[60], b[36], b[23], b[29], b[33], b[37], b[44], b[45], b[59], b[46], b[47], b[48],
|
||||
b[49], b[50], b[24], b[25], b[65], b[66], b[70], b[71]
|
||||
]
|
||||
bb.extend(window_env_list)
|
||||
bb.append(b[72])
|
||||
|
||||
return rc4_encrypt(
|
||||
''.join(chr(byte) for byte in bb),
|
||||
chr(121)
|
||||
)
|
||||
|
||||
|
||||
def ab_sign(url_search_params: str, user_agent: str) -> str:
|
||||
window_env_str = "1920|1080|1920|1040|0|30|0|0|1872|92|1920|1040|1857|92|1|24|Win32"
|
||||
|
||||
# 1. 生成随机字符串前缀
|
||||
# 2. 生成RC4加密的主体部分
|
||||
# 3. 对结果进行最终加密并添加等号后缀
|
||||
return result_encrypt(
|
||||
generate_random_str() +
|
||||
generate_rc4_bb_str(url_search_params, user_agent, window_env_str),
|
||||
"s4"
|
||||
) + "="
|
||||
0
src/http_clients/__init__.py
Normal file
0
src/http_clients/__init__.py
Normal file
59
src/http_clients/async_http.py
Normal file
59
src/http_clients/async_http.py
Normal file
@ -0,0 +1,59 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
import httpx
|
||||
from typing import Dict, Any
|
||||
from .. import utils
|
||||
|
||||
OptionalStr = str | None
|
||||
OptionalDict = Dict[str, Any] | None
|
||||
|
||||
|
||||
async def async_req(
|
||||
url: str,
|
||||
proxy_addr: OptionalStr = None,
|
||||
headers: OptionalDict = None,
|
||||
data: dict | bytes | None = None,
|
||||
json_data: dict | list | None = None,
|
||||
timeout: int = 20,
|
||||
redirect_url: bool = False,
|
||||
return_cookies: bool = False,
|
||||
include_cookies: bool = False,
|
||||
abroad: bool = False,
|
||||
content_conding: str = 'utf-8',
|
||||
verify: bool = False,
|
||||
http2: bool = True
|
||||
) -> OptionalDict | OptionalStr | tuple:
|
||||
if headers is None:
|
||||
headers = {}
|
||||
try:
|
||||
proxy_addr = utils.handle_proxy_addr(proxy_addr)
|
||||
if data or json_data:
|
||||
async with httpx.AsyncClient(proxy=proxy_addr, timeout=timeout, verify=verify, http2=http2) as client:
|
||||
response = await client.post(url, data=data, json=json_data, headers=headers)
|
||||
else:
|
||||
async with httpx.AsyncClient(proxy=proxy_addr, timeout=timeout, verify=verify, http2=http2) as client:
|
||||
response = await client.get(url, headers=headers, follow_redirects=True)
|
||||
|
||||
if redirect_url:
|
||||
return str(response.url)
|
||||
elif return_cookies:
|
||||
cookies_dict = {name: value for name, value in response.cookies.items()}
|
||||
return (response.text, cookies_dict) if include_cookies else cookies_dict
|
||||
else:
|
||||
resp_str = response.text
|
||||
except Exception as e:
|
||||
resp_str = str(e)
|
||||
|
||||
return resp_str
|
||||
|
||||
|
||||
async def get_response_status(url: str, proxy_addr: OptionalStr = None, headers: OptionalDict = None,
|
||||
timeout: int = 10, abroad: bool = False, verify: bool = False, http2=False) -> bool:
|
||||
|
||||
try:
|
||||
proxy_addr = utils.handle_proxy_addr(proxy_addr)
|
||||
async with httpx.AsyncClient(proxy=proxy_addr, timeout=timeout, verify=verify) as client:
|
||||
response = await client.head(url, headers=headers, follow_redirects=True)
|
||||
return response.status_code == 200
|
||||
except Exception as e:
|
||||
print(e)
|
||||
return False
|
||||
88
src/http_clients/sync_http.py
Normal file
88
src/http_clients/sync_http.py
Normal file
@ -0,0 +1,88 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
import gzip
|
||||
import urllib.parse
|
||||
import urllib.error
|
||||
import requests
|
||||
import ssl
|
||||
import json
|
||||
import urllib.request
|
||||
|
||||
no_proxy_handler = urllib.request.ProxyHandler({})
|
||||
opener = urllib.request.build_opener(no_proxy_handler)
|
||||
|
||||
ssl_context = ssl.create_default_context()
|
||||
ssl_context.check_hostname = False
|
||||
ssl_context.verify_mode = ssl.CERT_NONE
|
||||
OptionalStr = str | None
|
||||
OptionalDict = dict | None
|
||||
|
||||
|
||||
def sync_req(
|
||||
url: str,
|
||||
proxy_addr: OptionalStr = None,
|
||||
headers: OptionalDict = None,
|
||||
data: dict | bytes | None = None,
|
||||
json_data: dict | list | None = None,
|
||||
timeout: int = 20,
|
||||
redirect_url: bool = False,
|
||||
abroad: bool = False,
|
||||
content_conding: str = 'utf-8'
|
||||
) -> str:
|
||||
if headers is None:
|
||||
headers = {}
|
||||
try:
|
||||
if proxy_addr:
|
||||
proxies = {
|
||||
'http': proxy_addr,
|
||||
'https': proxy_addr
|
||||
}
|
||||
if data or json_data:
|
||||
response = requests.post(
|
||||
url, data=data, json=json_data, headers=headers, proxies=proxies, timeout=timeout
|
||||
)
|
||||
else:
|
||||
response = requests.get(url, headers=headers, proxies=proxies, timeout=timeout)
|
||||
if redirect_url:
|
||||
return response.url
|
||||
resp_str = response.text
|
||||
else:
|
||||
if data and not isinstance(data, bytes):
|
||||
data = urllib.parse.urlencode(data).encode(content_conding)
|
||||
if json_data and isinstance(json_data, (dict, list)):
|
||||
data = json.dumps(json_data).encode(content_conding)
|
||||
|
||||
req = urllib.request.Request(url, data=data, headers=headers)
|
||||
|
||||
try:
|
||||
if abroad:
|
||||
response = urllib.request.urlopen(req, timeout=timeout)
|
||||
else:
|
||||
response = opener.open(req, timeout=timeout)
|
||||
if redirect_url:
|
||||
return response.url
|
||||
content_encoding = response.info().get('Content-Encoding')
|
||||
try:
|
||||
if content_encoding == 'gzip':
|
||||
with gzip.open(response, 'rt', encoding=content_conding) as gzipped:
|
||||
resp_str = gzipped.read()
|
||||
else:
|
||||
resp_str = response.read().decode(content_conding)
|
||||
finally:
|
||||
response.close()
|
||||
|
||||
except urllib.error.HTTPError as e:
|
||||
if e.code == 400:
|
||||
resp_str = e.read().decode(content_conding)
|
||||
else:
|
||||
raise
|
||||
except urllib.error.URLError as e:
|
||||
print(f"URL Error: {e}")
|
||||
raise
|
||||
except Exception as e:
|
||||
print(f"An error occurred: {e}")
|
||||
raise
|
||||
|
||||
except Exception as e:
|
||||
resp_str = str(e)
|
||||
|
||||
return resp_str
|
||||
33
src/javascript/laixiu.js
Normal file
33
src/javascript/laixiu.js
Normal file
@ -0,0 +1,33 @@
|
||||
|
||||
function generateUUID() {
|
||||
return 'xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx'.replace(/[xy]/g, function(c) {
|
||||
const r = Math.random() * 16 | 0, v = c === 'x' ? r : (r & 0x3 | 0x8);
|
||||
return v.toString(16);
|
||||
});
|
||||
}
|
||||
|
||||
function calculateSign() {
|
||||
const a = new Date().getTime();
|
||||
const s = generateUUID().replace(/-/g, "");
|
||||
const u = 'kk792f28d6ff1f34ec702c08626d454b39pro';
|
||||
|
||||
const input = "web" + s + a + u;
|
||||
|
||||
const hash = CryptoJS.MD5(input).toString();
|
||||
|
||||
return {
|
||||
timestamp: a,
|
||||
imei: s,
|
||||
requestId: hash,
|
||||
inputString: input
|
||||
};
|
||||
}
|
||||
|
||||
function sign(cryptoJSPath) {
|
||||
CryptoJS = require(cryptoJSPath);
|
||||
return calculateSign();
|
||||
}
|
||||
|
||||
module.exports = {
|
||||
sign
|
||||
};
|
||||
143
src/javascript/migu.js
Normal file
143
src/javascript/migu.js
Normal file
@ -0,0 +1,143 @@
|
||||
/**
|
||||
* Function to get the ddCalcu parameter value
|
||||
* @param {string} inputUrl - The original URL before encryption
|
||||
* @returns {Promise<string>} - Returns the calculated ddCalcu value
|
||||
*/
|
||||
async function getDdCalcu(inputUrl) {
|
||||
let wasmInstance = null;
|
||||
let memory_p = null; // Uint8Array view
|
||||
let memory_h = null; // Uint32Array view
|
||||
|
||||
// Fixed parameter
|
||||
const f = 'PBTxuWiTEbUPPFcpyxs0ww==';
|
||||
|
||||
// Utility function: Convert string to UTF-8 in memory
|
||||
function stringToUTF8(string, offset) {
|
||||
const encoder = new TextEncoder();
|
||||
const encoded = encoder.encode(string);
|
||||
for (let i = 0; i < encoded.length; i++) {
|
||||
memory_p[offset + i] = encoded[i];
|
||||
}
|
||||
memory_p[offset + encoded.length] = 0; // Null-terminate
|
||||
}
|
||||
|
||||
// Utility function: Read UTF-8 string from memory address
|
||||
function UTF8ToString(offset) {
|
||||
let s = '';
|
||||
let i = 0;
|
||||
while (memory_p[offset + i]) {
|
||||
s += String.fromCharCode(memory_p[offset + i]);
|
||||
i++;
|
||||
}
|
||||
return s;
|
||||
}
|
||||
|
||||
// WASM import function stubs
|
||||
function a(e, t, r, n) {
|
||||
let s = 0;
|
||||
for (let i = 0; i < r; i++) {
|
||||
const d = memory_h[t + 4 >> 2];
|
||||
t += 8;
|
||||
s += d;
|
||||
}
|
||||
memory_h[n >> 2] = s;
|
||||
return 0;
|
||||
}
|
||||
|
||||
function b() {}
|
||||
|
||||
function c() {}
|
||||
|
||||
// Step 1: Retrieve playerVersion
|
||||
const settingsResp = await fetch('https://app-sc.miguvideo.com/common/v1/settings/H5_DetailPage');
|
||||
const settingsData = await settingsResp.json();
|
||||
const playerVersion = JSON.parse(settingsData.body.paramValue).playerVersion;
|
||||
|
||||
// Step 2: Load WASM module
|
||||
const wasmUrl = `https://www.miguvideo.com/mgs/player/prd/${playerVersion}/dist/mgprtcl.wasm`;
|
||||
const wasmResp = await fetch(wasmUrl);
|
||||
if (!wasmResp.ok) throw new Error("Failed to download WASM");
|
||||
const wasmBuffer = await wasmResp.arrayBuffer();
|
||||
|
||||
const importObject = {
|
||||
a: { a, b, c }
|
||||
};
|
||||
|
||||
const { instance } = await WebAssembly.instantiate(wasmBuffer, importObject);
|
||||
wasmInstance = instance;
|
||||
|
||||
const memory = wasmInstance.exports.d;
|
||||
memory_p = new Uint8Array(memory.buffer);
|
||||
memory_h = new Uint32Array(memory.buffer);
|
||||
|
||||
const exports = {
|
||||
CallInterface1: wasmInstance.exports.h,
|
||||
CallInterface2: wasmInstance.exports.i,
|
||||
CallInterface3: wasmInstance.exports.j,
|
||||
CallInterface4: wasmInstance.exports.k,
|
||||
CallInterface6: wasmInstance.exports.m,
|
||||
CallInterface7: wasmInstance.exports.n,
|
||||
CallInterface8: wasmInstance.exports.o,
|
||||
CallInterface9: wasmInstance.exports.p,
|
||||
CallInterface10: wasmInstance.exports.q,
|
||||
CallInterface11: wasmInstance.exports.r,
|
||||
CallInterface14: wasmInstance.exports.t,
|
||||
malloc: wasmInstance.exports.u,
|
||||
};
|
||||
|
||||
const parsedUrl = new URL(inputUrl);
|
||||
const query = Object.fromEntries(parsedUrl.searchParams);
|
||||
|
||||
const o = query.userid || '';
|
||||
const a_val = query.timestamp || '';
|
||||
const s = query.ProgramID || '';
|
||||
const u = query.Channel_ID || '';
|
||||
const v = query.puData || '';
|
||||
|
||||
// Allocate memory
|
||||
const d = exports.malloc(o.length + 1);
|
||||
const h = exports.malloc(a_val.length + 1);
|
||||
const y = exports.malloc(s.length + 1);
|
||||
const m = exports.malloc(u.length + 1);
|
||||
const g = exports.malloc(v.length + 1);
|
||||
const b_val = exports.malloc(f.length + 1);
|
||||
const E = exports.malloc(128);
|
||||
const T = exports.malloc(128);
|
||||
|
||||
// Write data to memory
|
||||
stringToUTF8(o, d);
|
||||
stringToUTF8(a_val, h);
|
||||
stringToUTF8(s, y);
|
||||
stringToUTF8(u, m);
|
||||
stringToUTF8(v, g);
|
||||
stringToUTF8(f, b_val);
|
||||
|
||||
// Call interface functions
|
||||
const S = exports.CallInterface6(); // Create context
|
||||
exports.CallInterface1(S, y, s.length);
|
||||
exports.CallInterface10(S, h, a_val.length);
|
||||
exports.CallInterface9(S, d, o.length);
|
||||
exports.CallInterface3(S, 0, 0);
|
||||
exports.CallInterface11(S, 0, 0);
|
||||
exports.CallInterface8(S, g, v.length);
|
||||
exports.CallInterface2(S, m, u.length);
|
||||
exports.CallInterface14(S, b_val, f.length, T, 128);
|
||||
|
||||
const w = UTF8ToString(T);
|
||||
const I = exports.malloc(w.length + 1);
|
||||
stringToUTF8(w, I);
|
||||
|
||||
exports.CallInterface7(S, I, w.length);
|
||||
exports.CallInterface4(S, E, 128);
|
||||
|
||||
return UTF8ToString(E);
|
||||
}
|
||||
|
||||
const url = process.argv[2];
|
||||
|
||||
getDdCalcu(url).then(result => {
|
||||
console.log(result);
|
||||
}).catch(err => {
|
||||
console.error(err);
|
||||
process.exit(1);
|
||||
});
|
||||
78
src/javascript/taobao-sign.js
Normal file
78
src/javascript/taobao-sign.js
Normal file
@ -0,0 +1,78 @@
|
||||
function sign(e) {
|
||||
function t(e, t) {
|
||||
return e << t | e >>> 32 - t
|
||||
}
|
||||
function o(e, t) {
|
||||
var o, n, r, i, a;
|
||||
return r = 2147483648 & e,
|
||||
i = 2147483648 & t,
|
||||
a = (1073741823 & e) + (1073741823 & t),
|
||||
(o = 1073741824 & e) & (n = 1073741824 & t) ? 2147483648 ^ a ^ r ^ i : o | n ? 1073741824 & a ? 3221225472 ^ a ^ r ^ i : 1073741824 ^ a ^ r ^ i : a ^ r ^ i
|
||||
}
|
||||
function n(e, n, r, i, a, s, u) {
|
||||
return o(t(e = o(e, o(o(function(e, t, o) {
|
||||
return e & t | ~e & o
|
||||
}(n, r, i), a), u)), s), n)
|
||||
}
|
||||
function r(e, n, r, i, a, s, u) {
|
||||
return o(t(e = o(e, o(o(function(e, t, o) {
|
||||
return e & o | t & ~o
|
||||
}(n, r, i), a), u)), s), n)
|
||||
}
|
||||
function i(e, n, r, i, a, s, u) {
|
||||
return o(t(e = o(e, o(o(function(e, t, o) {
|
||||
return e ^ t ^ o
|
||||
}(n, r, i), a), u)), s), n)
|
||||
}
|
||||
function a(e, n, r, i, a, s, u) {
|
||||
return o(t(e = o(e, o(o(function(e, t, o) {
|
||||
return t ^ (e | ~o)
|
||||
}(n, r, i), a), u)), s), n)
|
||||
}
|
||||
function s(e) {
|
||||
var t, o = "", n = "";
|
||||
for (t = 0; 3 >= t; t++)
|
||||
o += (n = "0" + (e >>> 8 * t & 255).toString(16)).substr(n.length - 2, 2);
|
||||
return o
|
||||
}
|
||||
var u, l, d, c, p, f, h, m, y, g;
|
||||
for (g = function(e) {
|
||||
for (var t = e.length, o = t + 8, n = 16 * ((o - o % 64) / 64 + 1), r = Array(n - 1), i = 0, a = 0; t > a; )
|
||||
i = a % 4 * 8,
|
||||
r[(a - a % 4) / 4] |= e.charCodeAt(a) << i,
|
||||
a++;
|
||||
return i = a % 4 * 8,
|
||||
r[(a - a % 4) / 4] |= 128 << i,
|
||||
r[n - 2] = t << 3,
|
||||
r[n - 1] = t >>> 29,
|
||||
r
|
||||
}(e = function(e) {
|
||||
var t = String.fromCharCode;
|
||||
e = e.replace(/\r\n/g, "\n");
|
||||
for (var o, n = "", r = 0; r < e.length; r++)
|
||||
128 > (o = e.charCodeAt(r)) ? n += t(o) : o > 127 && 2048 > o ? (n += t(o >> 6 | 192),
|
||||
n += t(63 & o | 128)) : (n += t(o >> 12 | 224),
|
||||
n += t(o >> 6 & 63 | 128),
|
||||
n += t(63 & o | 128));
|
||||
return n
|
||||
}(e)),
|
||||
f = 1732584193,
|
||||
h = 4023233417,
|
||||
m = 2562383102,
|
||||
y = 271733878,
|
||||
u = 0; u < g.length; u += 16)
|
||||
l = f,
|
||||
d = h,
|
||||
c = m,
|
||||
p = y,
|
||||
h = a(h = a(h = a(h = a(h = i(h = i(h = i(h = i(h = r(h = r(h = r(h = r(h = n(h = n(h = n(h = n(h, m = n(m, y = n(y, f = n(f, h, m, y, g[u + 0], 7, 3614090360), h, m, g[u + 1], 12, 3905402710), f, h, g[u + 2], 17, 606105819), y, f, g[u + 3], 22, 3250441966), m = n(m, y = n(y, f = n(f, h, m, y, g[u + 4], 7, 4118548399), h, m, g[u + 5], 12, 1200080426), f, h, g[u + 6], 17, 2821735955), y, f, g[u + 7], 22, 4249261313), m = n(m, y = n(y, f = n(f, h, m, y, g[u + 8], 7, 1770035416), h, m, g[u + 9], 12, 2336552879), f, h, g[u + 10], 17, 4294925233), y, f, g[u + 11], 22, 2304563134), m = n(m, y = n(y, f = n(f, h, m, y, g[u + 12], 7, 1804603682), h, m, g[u + 13], 12, 4254626195), f, h, g[u + 14], 17, 2792965006), y, f, g[u + 15], 22, 1236535329), m = r(m, y = r(y, f = r(f, h, m, y, g[u + 1], 5, 4129170786), h, m, g[u + 6], 9, 3225465664), f, h, g[u + 11], 14, 643717713), y, f, g[u + 0], 20, 3921069994), m = r(m, y = r(y, f = r(f, h, m, y, g[u + 5], 5, 3593408605), h, m, g[u + 10], 9, 38016083), f, h, g[u + 15], 14, 3634488961), y, f, g[u + 4], 20, 3889429448), m = r(m, y = r(y, f = r(f, h, m, y, g[u + 9], 5, 568446438), h, m, g[u + 14], 9, 3275163606), f, h, g[u + 3], 14, 4107603335), y, f, g[u + 8], 20, 1163531501), m = r(m, y = r(y, f = r(f, h, m, y, g[u + 13], 5, 2850285829), h, m, g[u + 2], 9, 4243563512), f, h, g[u + 7], 14, 1735328473), y, f, g[u + 12], 20, 2368359562), m = i(m, y = i(y, f = i(f, h, m, y, g[u + 5], 4, 4294588738), h, m, g[u + 8], 11, 2272392833), f, h, g[u + 11], 16, 1839030562), y, f, g[u + 14], 23, 4259657740), m = i(m, y = i(y, f = i(f, h, m, y, g[u + 1], 4, 2763975236), h, m, g[u + 4], 11, 1272893353), f, h, g[u + 7], 16, 4139469664), y, f, g[u + 10], 23, 3200236656), m = i(m, y = i(y, f = i(f, h, m, y, g[u + 13], 4, 681279174), h, m, g[u + 0], 11, 3936430074), f, h, g[u + 3], 16, 3572445317), y, f, g[u + 6], 23, 76029189), m = i(m, y = i(y, f = i(f, h, m, y, g[u + 9], 4, 3654602809), h, m, g[u + 12], 11, 3873151461), f, h, g[u + 15], 16, 530742520), y, f, g[u + 2], 23, 3299628645), m = a(m, y = a(y, f = a(f, h, m, y, g[u + 0], 6, 4096336452), h, m, g[u + 7], 10, 1126891415), f, h, g[u + 14], 15, 2878612391), y, f, g[u + 5], 21, 4237533241), m = a(m, y = a(y, f = a(f, h, m, y, g[u + 12], 6, 1700485571), h, m, g[u + 3], 10, 2399980690), f, h, g[u + 10], 15, 4293915773), y, f, g[u + 1], 21, 2240044497), m = a(m, y = a(y, f = a(f, h, m, y, g[u + 8], 6, 1873313359), h, m, g[u + 15], 10, 4264355552), f, h, g[u + 6], 15, 2734768916), y, f, g[u + 13], 21, 1309151649), m = a(m, y = a(y, f = a(f, h, m, y, g[u + 4], 6, 4149444226), h, m, g[u + 11], 10, 3174756917), f, h, g[u + 2], 15, 718787259), y, f, g[u + 9], 21, 3951481745),
|
||||
f = o(f, l),
|
||||
h = o(h, d),
|
||||
m = o(m, c),
|
||||
y = o(y, p);
|
||||
return (s(f) + s(h) + s(m) + s(y)).toLowerCase()
|
||||
}
|
||||
|
||||
// 正确sign值：05748e8359cd3e6deaab02d15caafc11
|
||||
// var sg =sign('5655b7041ca049730330701082886efd&1719411639403&12574478&{"componentKey":"wp_pc_shop_basic_info","params":"{\\"memberId\\":\\"b2b-22133374292418351a\\"}"}')
|
||||
// console.log(sg)
|
||||
@ -2,15 +2,24 @@
|
||||
|
||||
import os
|
||||
import sys
|
||||
from loguru import logger
|
||||
|
||||
logger.remove()
|
||||
|
||||
custom_format = "<green>{time:YYYY-MM-DD HH:mm:ss.SSS}</green> | <level>{level: <8}</level> - <level>{message}</level>"
|
||||
os.environ["LOGURU_FORMAT"] = custom_format
|
||||
from loguru import logger
|
||||
|
||||
logger.add(
|
||||
sink=sys.stderr,
|
||||
format=custom_format,
|
||||
level="DEBUG",
|
||||
colorize=True,
|
||||
enqueue=True
|
||||
)
|
||||
|
||||
script_path = os.path.split(os.path.realpath(sys.argv[0]))[0]
|
||||
|
||||
logger.add(
|
||||
f"{script_path}/logs/DouyinLiveRecorder.log",
|
||||
f"{script_path}/logs/streamget.log",
|
||||
level="DEBUG",
|
||||
format="{time:YYYY-MM-DD HH:mm:ss.SSS} | {level: <8} | {name}:{function}:{line} - {message}",
|
||||
filter=lambda i: i["level"].name != "INFO",
|
||||
150
src/room.py
Normal file
150
src/room.py
Normal file
@ -0,0 +1,150 @@
|
||||
# -*- encoding: utf-8 -*-
|
||||
|
||||
"""
|
||||
Author: Hmily
|
||||
GitHub:https://github.com/ihmily
|
||||
Date: 2023-07-17 23:52:05
|
||||
Update: 2025-02-04 04:57:00
|
||||
Copyright (c) 2023 by Hmily, All Rights Reserved.
|
||||
"""
|
||||
import re
|
||||
import urllib.parse
|
||||
import execjs
|
||||
import httpx
|
||||
import urllib.request
|
||||
from . import JS_SCRIPT_PATH, utils
|
||||
|
||||
no_proxy_handler = urllib.request.ProxyHandler({})
|
||||
opener = urllib.request.build_opener(no_proxy_handler)
|
||||
|
||||
|
||||
class UnsupportedUrlError(Exception):
|
||||
pass
|
||||
|
||||
|
||||
HEADERS = {
|
||||
'User-Agent': 'Mozilla/5.0 (Linux; Android 11; SAMSUNG SM-G973U) AppleWebKit/537.36 (KHTML, like Gecko) '
|
||||
'SamsungBrowser/14.2 Chrome/87.0.4280.141 Mobile Safari/537.36',
|
||||
'Accept-Language': 'zh-CN,zh;q=0.8,zh-TW;q=0.7,zh-HK;q=0.5,en-US;q=0.3,en;q=0.2',
|
||||
'Cookie': 's_v_web_id=verify_lk07kv74_QZYCUApD_xhiB_405x_Ax51_GYO9bUIyZQVf'
|
||||
}
|
||||
|
||||
HEADERS_PC = {
|
||||
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) '
|
||||
'Chrome/121.0.0.0 Safari/537.36 Edg/121.0.0.0',
|
||||
'Cookie': 'sessionid=7494ae59ae06784454373ce25761e864; __ac_nonce=0670497840077ee4c9eb2; '
|
||||
'__ac_signature=_02B4Z6wo00f012DZczQAAIDCJJBb3EjnINdg-XeAAL8-db; '
|
||||
's_v_web_id=verify_m1ztgtjj_vuHnMLZD_iwZ9_4YO4_BdN1_7wLP3pyqXsf2; '
|
||||
}
|
||||
|
||||
|
||||
# X-bogus算法
|
||||
async def get_xbogus(url: str, headers: dict | None = None) -> str:
|
||||
if not headers or 'user-agent' not in (k.lower() for k in headers):
|
||||
headers = HEADERS
|
||||
query = urllib.parse.urlparse(url).query
|
||||
xbogus = execjs.compile(open(f'{JS_SCRIPT_PATH}/x-bogus.js').read()).call(
|
||||
'sign', query, headers.get("User-Agent", "user-agent"))
|
||||
return xbogus
|
||||
|
||||
|
||||
# 获取房间ID和用户secID
|
||||
async def get_sec_user_id(url: str, proxy_addr: str | None = None, headers: dict | None = None) -> tuple | None:
|
||||
if not headers or all(k.lower() not in ['user-agent', 'cookie'] for k in headers):
|
||||
headers = HEADERS
|
||||
|
||||
try:
|
||||
proxy_addr = utils.handle_proxy_addr(proxy_addr)
|
||||
async with httpx.AsyncClient(proxy=proxy_addr, timeout=15) as client:
|
||||
response = await client.get(url, headers=headers, follow_redirects=True)
|
||||
redirect_url = response.url
|
||||
if 'reflow/' in str(redirect_url):
|
||||
match = re.search(r'sec_user_id=([\w_\-]+)&', str(redirect_url))
|
||||
if match:
|
||||
sec_user_id = match.group(1)
|
||||
room_id = str(redirect_url).split('?')[0].rsplit('/', maxsplit=1)[1]
|
||||
return room_id, sec_user_id
|
||||
else:
|
||||
raise RuntimeError("Could not find sec_user_id in the URL.")
|
||||
else:
|
||||
raise UnsupportedUrlError("The redirect URL does not contain 'reflow/'.")
|
||||
except UnsupportedUrlError as e:
|
||||
raise e
|
||||
except Exception as e:
|
||||
raise RuntimeError(f"An error occurred: {e}")
|
||||
|
||||
|
||||
# 获取抖音号
|
||||
async def get_unique_id(url: str, proxy_addr: str | None = None, headers: dict | None = None) -> str | None:
|
||||
if not headers or all(k.lower() not in ['user-agent', 'cookie'] for k in headers):
|
||||
headers = HEADERS
|
||||
|
||||
try:
|
||||
proxy_addr = utils.handle_proxy_addr(proxy_addr)
|
||||
async with httpx.AsyncClient(proxy=proxy_addr, timeout=15) as client:
|
||||
response = await client.get(url, headers=headers, follow_redirects=True)
|
||||
redirect_url = str(response.url)
|
||||
if 'reflow/' in str(redirect_url):
|
||||
raise UnsupportedUrlError("Unsupported URL")
|
||||
sec_user_id = redirect_url.split('?')[0].rsplit('/', maxsplit=1)[1]
|
||||
headers['Cookie'] = ('ttwid=1%7C4ejCkU2bKY76IySQENJwvGhg1IQZrgGEupSyTKKfuyk%7C1740470403%7Cbc9a'
|
||||
'd2ee341f1a162f9e27f4641778030d1ae91e31f9df6553a8f2efa3bdb7b4; __ac_nonce=06'
|
||||
'83e59f3009cc48fbab0; __ac_signature=_02B4Z6wo00f01mG6waQAAIDB9JUCzFb6.TZhmsU'
|
||||
'AAPBf34; __ac_referer=__ac_blank')
|
||||
user_page_response = await client.get(f'https://www.iesdouyin.com/share/user/{sec_user_id}',
|
||||
headers=headers, follow_redirects=True)
|
||||
matches = re.findall(r'unique_id":"(.*?)","verification_type', user_page_response.text)
|
||||
if matches:
|
||||
unique_id = matches[-1]
|
||||
return unique_id
|
||||
else:
|
||||
raise RuntimeError("Could not find unique_id in the response.")
|
||||
except UnsupportedUrlError as e:
|
||||
raise e
|
||||
except Exception as e:
|
||||
raise RuntimeError(f"An error occurred: {e}")
|
||||
|
||||
|
||||
# 获取直播间webID
|
||||
async def get_live_room_id(room_id: str, sec_user_id: str, proxy_addr: str | None = None, params: dict | None = None,
|
||||
headers: dict | None = None) -> str:
|
||||
if not headers or all(k.lower() not in ['user-agent', 'cookie'] for k in headers):
|
||||
headers = HEADERS
|
||||
|
||||
if not params:
|
||||
params = {
|
||||
"verifyFp": "verify_lk07kv74_QZYCUApD_xhiB_405x_Ax51_GYO9bUIyZQVf",
|
||||
"type_id": "0",
|
||||
"live_id": "1",
|
||||
"room_id": room_id,
|
||||
"sec_user_id": sec_user_id,
|
||||
"app_id": "1128",
|
||||
"msToken": "wrqzbEaTlsxt52-vxyZo_mIoL0RjNi1ZdDe7gzEGMUTVh_HvmbLLkQrA_1HKVOa2C6gkxb6IiY6TY2z8enAkPEwGq--gM"
|
||||
"-me3Yudck2ailla5Q4osnYIHxd9dI4WtQ==",
|
||||
}
|
||||
|
||||
api = f'https://webcast.amemv.com/webcast/room/reflow/info/?{urllib.parse.urlencode(params)}'
|
||||
xbogus = await get_xbogus(api)
|
||||
api = api + "&X-Bogus=" + xbogus
|
||||
|
||||
try:
|
||||
proxy_addr = utils.handle_proxy_addr(proxy_addr)
|
||||
async with httpx.AsyncClient(proxy=proxy_addr,
|
||||
timeout=15) as client:
|
||||
response = await client.get(api, headers=headers)
|
||||
response.raise_for_status()
|
||||
json_data = response.json()
|
||||
return json_data['data']['room']['owner']['web_rid']
|
||||
except httpx.HTTPStatusError as e:
|
||||
print(f"HTTP status error occurred: {e.response.status_code}")
|
||||
raise
|
||||
except Exception as e:
|
||||
print(f"An exception occurred during get_live_room_id: {e}")
|
||||
raise
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
room_url = "https://v.douyin.com/iQLgKSj/"
|
||||
_room_id, sec_uid = get_sec_user_id(room_url)
|
||||
web_rid = get_live_room_id(_room_id, sec_uid)
|
||||
print("return web_rid:", web_rid)
|
||||
File diff suppressed because it is too large
Load Diff
@ -4,8 +4,8 @@
|
||||
Author: Hmily
|
||||
GitHub: https://github.com/ihmily
|
||||
Date: 2023-07-15 23:15:00
|
||||
Update: 2024-10-27 17:15:00
|
||||
Copyright (c) 2023-2024 by Hmily, All Rights Reserved.
|
||||
Update: 2025-02-06 02:28:00
|
||||
Copyright (c) 2023-2025 by Hmily, All Rights Reserved.
|
||||
Function: Get live stream data.
|
||||
"""
|
||||
import base64
|
||||
@ -21,10 +21,24 @@ from .utils import trace_error_decorator
|
||||
from .spider import (
|
||||
get_douyu_stream_data, get_bilibili_stream_data
|
||||
)
|
||||
from .http_clients.async_http import get_response_status
|
||||
|
||||
QUALITY_MAPPING = {"OD": 0, "BD": 0, "UHD": 1, "HD": 2, "SD": 3, "LD": 4}
|
||||
|
||||
|
||||
def get_quality_index(quality) -> tuple:
|
||||
if not quality:
|
||||
return list(QUALITY_MAPPING.items())[0]
|
||||
|
||||
quality_str = str(quality).upper()
|
||||
if quality_str.isdigit():
|
||||
quality_int = int(quality_str[0])
|
||||
quality_str = list(QUALITY_MAPPING.keys())[quality_int]
|
||||
return quality_str, QUALITY_MAPPING.get(quality_str, 0)
|
||||
|
||||
|
||||
@trace_error_decorator
|
||||
def get_douyin_stream_url(json_data: dict, video_quality: str) -> dict:
|
||||
async def get_douyin_stream_url(json_data: dict, video_quality: str, proxy_addr: str) -> dict:
|
||||
anchor_name = json_data.get('anchor_name')
|
||||
|
||||
result = {
|
||||
@ -32,7 +46,7 @@ def get_douyin_stream_url(json_data: dict, video_quality: str) -> dict:
|
||||
"is_live": False,
|
||||
}
|
||||
|
||||
status = json_data.get("status", 4) # 直播状态 2 是正在直播、4 是未开播
|
||||
status = json_data.get("status", 4)
|
||||
|
||||
if status == 2:
|
||||
stream_url = json_data['stream_url']
|
||||
@ -45,13 +59,18 @@ def get_douyin_stream_url(json_data: dict, video_quality: str) -> dict:
|
||||
flv_url_list.append(flv_url_list[-1])
|
||||
m3u8_url_list.append(m3u8_url_list[-1])
|
||||
|
||||
video_qualities = {"原画": 0, "蓝光": 0, "超清": 1, "高清": 2, "标清": 3, "流畅": 4}
|
||||
quality_index = video_qualities.get(video_quality)
|
||||
video_quality, quality_index = get_quality_index(video_quality)
|
||||
m3u8_url = m3u8_url_list[quality_index]
|
||||
flv_url = flv_url_list[quality_index]
|
||||
ok = await get_response_status(url=m3u8_url, proxy_addr=proxy_addr)
|
||||
if not ok:
|
||||
index = quality_index + 1 if quality_index < 4 else quality_index - 1
|
||||
m3u8_url = m3u8_url_list[index]
|
||||
flv_url = flv_url_list[index]
|
||||
result |= {
|
||||
'is_live': True,
|
||||
'title': json_data['title'],
|
||||
'quality': video_quality,
|
||||
'm3u8_url': m3u8_url,
|
||||
'flv_url': flv_url,
|
||||
'record_url': m3u8_url or flv_url,
|
||||
@ -60,7 +79,7 @@ def get_douyin_stream_url(json_data: dict, video_quality: str) -> dict:
|
||||
|
||||
|
||||
@trace_error_decorator
|
||||
def get_tiktok_stream_url(json_data: dict, video_quality: str) -> dict:
|
||||
async def get_tiktok_stream_url(json_data: dict, video_quality: str, proxy_addr: str) -> dict:
|
||||
if not json_data:
|
||||
return {"anchor_name": None, "is_live": False}
|
||||
|
||||
@ -68,10 +87,18 @@ def get_tiktok_stream_url(json_data: dict, video_quality: str) -> dict:
|
||||
play_list = []
|
||||
for key in stream:
|
||||
url_info = stream[key]['main']
|
||||
play_url = url_info[q_key]
|
||||
sdk_params = url_info['sdk_params']
|
||||
sdk_params = json.loads(sdk_params)
|
||||
vbitrate = int(sdk_params['vbitrate'])
|
||||
v_codec = sdk_params.get('VCodec', '')
|
||||
|
||||
play_url = ''
|
||||
if url_info.get(q_key):
|
||||
if url_info[q_key].endswith(".flv") or url_info[q_key].endswith(".m3u8"):
|
||||
play_url = url_info[q_key] + '?codec=' + v_codec
|
||||
else:
|
||||
play_url = url_info[q_key] + '&codec=' + v_codec
|
||||
|
||||
resolution = sdk_params['resolution']
|
||||
if vbitrate != 0 and resolution:
|
||||
width, height = map(int, resolution.split('x'))
|
||||
@ -101,13 +128,24 @@ def get_tiktok_stream_url(json_data: dict, video_quality: str) -> dict:
|
||||
flv_url_list.append(flv_url_list[-1])
|
||||
while len(m3u8_url_list) < 5:
|
||||
m3u8_url_list.append(m3u8_url_list[-1])
|
||||
video_qualities = {"原画": 0, "蓝光": 0, "超清": 1, "高清": 2, "标清": 3, '流畅': 4}
|
||||
quality_index = video_qualities.get(video_quality)
|
||||
flv_url = flv_url_list[quality_index]['url'].replace("https://", "http://")
|
||||
m3u8_url = m3u8_url_list[quality_index]['url'].replace("https://", "http://")
|
||||
video_quality, quality_index = get_quality_index(video_quality)
|
||||
flv_dict: dict = flv_url_list[quality_index]
|
||||
m3u8_dict: dict = m3u8_url_list[quality_index]
|
||||
|
||||
check_url = m3u8_dict.get('url') or flv_dict.get('url')
|
||||
ok = await get_response_status(url=check_url, proxy_addr=proxy_addr, http2=False)
|
||||
|
||||
if not ok:
|
||||
index = quality_index + 1 if quality_index < 4 else quality_index - 1
|
||||
flv_dict: dict = flv_url_list[index]
|
||||
m3u8_dict: dict = m3u8_url_list[index]
|
||||
|
||||
flv_url = flv_dict['url']
|
||||
m3u8_url = m3u8_dict['url']
|
||||
result |= {
|
||||
'is_live': True,
|
||||
'title': live_room['liveRoom']['title'],
|
||||
'quality': video_quality,
|
||||
'm3u8_url': m3u8_url,
|
||||
'flv_url': flv_url,
|
||||
'record_url': m3u8_url or flv_url,
|
||||
@ -116,7 +154,7 @@ def get_tiktok_stream_url(json_data: dict, video_quality: str) -> dict:
|
||||
|
||||
|
||||
@trace_error_decorator
|
||||
def get_kuaishou_stream_url(json_data: dict, video_quality: str) -> dict:
|
||||
async def get_kuaishou_stream_url(json_data: dict, video_quality: str) -> dict:
|
||||
if json_data['type'] == 1 and not json_data["is_live"]:
|
||||
return json_data
|
||||
live_status = json_data['is_live']
|
||||
@ -128,11 +166,10 @@ def get_kuaishou_stream_url(json_data: dict, video_quality: str) -> dict:
|
||||
}
|
||||
|
||||
if live_status:
|
||||
quality_mapping = {'原画': 0, '蓝光': 0, '超清': 1, '高清': 2, '标清': 3, '流畅': 4}
|
||||
quality_mapping_bitrate = {'原画': 99999, '蓝光': 4000, '超清': 2000, '高清': 1000, '标清': 800, '流畅': 600}
|
||||
if video_quality in quality_mapping:
|
||||
quality_mapping_bit = {'OD': 99999, 'BD': 4000, 'UHD': 2000, 'HD': 1000, 'SD': 800, 'LD': 600}
|
||||
if video_quality in QUALITY_MAPPING:
|
||||
|
||||
quality_index = quality_mapping[video_quality]
|
||||
quality, quality_index = get_quality_index(video_quality)
|
||||
if 'm3u8_url_list' in json_data:
|
||||
m3u8_url_list = json_data['m3u8_url_list'][::-1]
|
||||
while len(m3u8_url_list) < 5:
|
||||
@ -141,35 +178,36 @@ def get_kuaishou_stream_url(json_data: dict, video_quality: str) -> dict:
|
||||
result['m3u8_url'] = m3u8_url
|
||||
|
||||
if 'flv_url_list' in json_data:
|
||||
# checks if bitrate in flv_url_list
|
||||
if 'bitrate' in json_data['flv_url_list'][0]:
|
||||
flv_url_list = json_data['flv_url_list']
|
||||
flv_url_list = sorted(flv_url_list, key=lambda x: x['bitrate'], reverse=True)
|
||||
# uses quality_mapping_bitrate to get the index of the quality
|
||||
quality_index_bitrate_value = quality_mapping_bitrate[video_quality]
|
||||
# find the value below `quality_index_bitrate_value`, or else use the previous one.
|
||||
quality_str = str(video_quality).upper()
|
||||
if quality_str.isdigit():
|
||||
video_quality, quality_index_bitrate_value = list(quality_mapping_bit.items())[int(quality_str)]
|
||||
else:
|
||||
quality_index_bitrate_value = quality_mapping_bit.get(quality_str, 99999)
|
||||
video_quality = quality_str
|
||||
quality_index = next(
|
||||
(i for i, x in enumerate(flv_url_list) if x['bitrate'] <= quality_index_bitrate_value), None)
|
||||
if quality_index is None:
|
||||
# latest quality
|
||||
quality_index = len(flv_url_list) - 1
|
||||
flv_url = flv_url_list[quality_index]['url']
|
||||
|
||||
result['flv_url'] = flv_url
|
||||
result['record_url'] = flv_url
|
||||
else:
|
||||
# TODO: Old version which not working at 20241128, could be removed if not working confirmed,
|
||||
# please also clean the quality_mapping mapping
|
||||
flv_url_list = json_data['flv_url_list'][::-1]
|
||||
while len(flv_url_list) < 5:
|
||||
flv_url_list.append(flv_url_list[-1])
|
||||
flv_url = flv_url_list[quality_index]['url']
|
||||
result |= {'flv_url': flv_url, 'record_url': flv_url}
|
||||
result['is_live'] = True
|
||||
result['quality'] = video_quality
|
||||
return result
|
||||
|
||||
|
||||
@trace_error_decorator
|
||||
def get_huya_stream_url(json_data: dict, video_quality: str) -> dict:
|
||||
async def get_huya_stream_url(json_data: dict, video_quality: str) -> dict:
|
||||
game_live_info = json_data['data'][0]['gameLiveInfo']
|
||||
live_title = game_live_info['introduction']
|
||||
stream_info_list = json_data['data'][0]['gameStreamInfoList']
|
||||
@ -230,17 +268,17 @@ def get_huya_stream_url(json_data: dict, video_quality: str) -> dict:
|
||||
m3u8_url = f'{hls_url}/{stream_name}.{hls_url_suffix}?{new_anti_code}&ratio='
|
||||
|
||||
quality_list = flv_anti_code.split('&exsphd=')
|
||||
if len(quality_list) > 1 and video_quality not in ["原画", "蓝光"]:
|
||||
if len(quality_list) > 1 and video_quality not in ["OD", "BD"]:
|
||||
pattern = r"(?<=264_)\d+"
|
||||
quality_list = list(re.findall(pattern, quality_list[1]))[::-1]
|
||||
while len(quality_list) < 5:
|
||||
quality_list.append(quality_list[-1])
|
||||
|
||||
video_quality_options = {
|
||||
"超清": quality_list[0],
|
||||
"高清": quality_list[1],
|
||||
"标清": quality_list[2],
|
||||
"流畅": quality_list[3]
|
||||
"UHD": quality_list[0],
|
||||
"HD": quality_list[1],
|
||||
"SD": quality_list[2],
|
||||
"LD": quality_list[3]
|
||||
}
|
||||
|
||||
if video_quality not in video_quality_options:
|
||||
@ -253,6 +291,7 @@ def get_huya_stream_url(json_data: dict, video_quality: str) -> dict:
|
||||
result |= {
|
||||
'is_live': True,
|
||||
'title': live_title,
|
||||
'quality': video_quality,
|
||||
'm3u8_url': m3u8_url,
|
||||
'flv_url': flv_url,
|
||||
'record_url': flv_url or m3u8_url
|
||||
@ -261,33 +300,33 @@ def get_huya_stream_url(json_data: dict, video_quality: str) -> dict:
|
||||
|
||||
|
||||
@trace_error_decorator
|
||||
def get_douyu_stream_url(json_data: dict, video_quality: str, cookies: str, proxy_addr: str) -> dict:
|
||||
async def get_douyu_stream_url(json_data: dict, video_quality: str, cookies: str, proxy_addr: str) -> dict:
|
||||
if not json_data["is_live"]:
|
||||
return json_data
|
||||
|
||||
video_quality_options = {
|
||||
"原画": '0',
|
||||
"蓝光": '0',
|
||||
"超清": '3',
|
||||
"高清": '2',
|
||||
"标清": '1',
|
||||
"流畅": '1'
|
||||
"OD": '0',
|
||||
"BD": '0',
|
||||
"UHD": '3',
|
||||
"HD": '2',
|
||||
"SD": '1',
|
||||
"LD": '1'
|
||||
}
|
||||
|
||||
rid = str(json_data["room_id"])
|
||||
json_data.pop("room_id")
|
||||
rate = video_quality_options.get(video_quality, '0')
|
||||
flv_data = get_douyu_stream_data(rid, rate, cookies=cookies, proxy_addr=proxy_addr)
|
||||
flv_data = await get_douyu_stream_data(rid, rate, cookies=cookies, proxy_addr=proxy_addr)
|
||||
rtmp_url = flv_data['data'].get('rtmp_url')
|
||||
rtmp_live = flv_data['data'].get('rtmp_live')
|
||||
if rtmp_live:
|
||||
flv_url = f'{rtmp_url}/{rtmp_live}'
|
||||
json_data |= {'flv_url': flv_url, 'record_url': flv_url}
|
||||
json_data |= {'quality': video_quality, 'flv_url': flv_url, 'record_url': flv_url}
|
||||
return json_data
|
||||
|
||||
|
||||
@trace_error_decorator
|
||||
def get_yy_stream_url(json_data: dict) -> dict:
|
||||
async def get_yy_stream_url(json_data: dict) -> dict:
|
||||
anchor_name = json_data.get('anchor_name', '')
|
||||
result = {
|
||||
"anchor_name": anchor_name,
|
||||
@ -300,6 +339,7 @@ def get_yy_stream_url(json_data: dict) -> dict:
|
||||
result |= {
|
||||
'is_live': True,
|
||||
'title': json_data['title'],
|
||||
'quality': 'OD',
|
||||
'flv_url': flv_url,
|
||||
'record_url': flv_url
|
||||
}
|
||||
@ -307,7 +347,7 @@ def get_yy_stream_url(json_data: dict) -> dict:
|
||||
|
||||
|
||||
@trace_error_decorator
|
||||
def get_bilibili_stream_url(json_data: dict, video_quality: str, proxy_addr: str, cookies: str) -> dict:
|
||||
async def get_bilibili_stream_url(json_data: dict, video_quality: str, proxy_addr: str, cookies: str) -> dict:
|
||||
anchor_name = json_data["anchor_name"]
|
||||
if not json_data["live_status"]:
|
||||
return {
|
||||
@ -318,68 +358,89 @@ def get_bilibili_stream_url(json_data: dict, video_quality: str, proxy_addr: str
|
||||
room_url = json_data['room_url']
|
||||
|
||||
video_quality_options = {
|
||||
"原画": '10000',
|
||||
"蓝光": '400',
|
||||
"超清": '250',
|
||||
"高清": '150',
|
||||
"标清": '80',
|
||||
"流畅": '80'
|
||||
"OD": '10000',
|
||||
"BD": '400',
|
||||
"UHD": '250',
|
||||
"HD": '150',
|
||||
"SD": '80',
|
||||
"LD": '80'
|
||||
}
|
||||
|
||||
select_quality = video_quality_options[video_quality]
|
||||
play_url = get_bilibili_stream_data(
|
||||
play_url = await get_bilibili_stream_data(
|
||||
room_url, qn=select_quality, platform='web', proxy_addr=proxy_addr, cookies=cookies)
|
||||
return {
|
||||
'anchor_name': json_data['anchor_name'],
|
||||
'is_live': True,
|
||||
'title': json_data['title'],
|
||||
'quality': video_quality,
|
||||
'record_url': play_url
|
||||
}
|
||||
|
||||
|
||||
@trace_error_decorator
|
||||
def get_netease_stream_url(json_data: dict, video_quality: str) -> dict:
|
||||
async def get_netease_stream_url(json_data: dict, video_quality: str) -> dict:
|
||||
if not json_data['is_live']:
|
||||
return json_data
|
||||
stream_list = json_data['stream_list']['resolution']
|
||||
order = ['blueray', 'ultra', 'high', 'standard']
|
||||
sorted_keys = [key for key in order if key in stream_list]
|
||||
while len(sorted_keys) < 5:
|
||||
sorted_keys.append(sorted_keys[-1])
|
||||
quality_list = {'原画': 0, '蓝光': 0, '超清': 1, '高清': 2, '标清': 3, '流畅': 4}
|
||||
selected_quality = sorted_keys[quality_list[video_quality]]
|
||||
flv_url_list = stream_list[selected_quality]['cdn']
|
||||
selected_cdn = list(flv_url_list.keys())[0]
|
||||
flv_url = flv_url_list[selected_cdn]
|
||||
|
||||
m3u8_url = json_data['m3u8_url']
|
||||
flv_url = None
|
||||
if json_data.get('stream_list'):
|
||||
stream_list = json_data['stream_list']['resolution']
|
||||
order = ['blueray', 'ultra', 'high', 'standard']
|
||||
sorted_keys = [key for key in order if key in stream_list]
|
||||
while len(sorted_keys) < 5:
|
||||
sorted_keys.append(sorted_keys[-1])
|
||||
video_quality, quality_index = get_quality_index(video_quality)
|
||||
selected_quality = sorted_keys[quality_index]
|
||||
flv_url_list = stream_list[selected_quality]['cdn']
|
||||
selected_cdn = list(flv_url_list.keys())[0]
|
||||
flv_url = flv_url_list[selected_cdn]
|
||||
|
||||
return {
|
||||
"is_live": True,
|
||||
"anchor_name": json_data['anchor_name'],
|
||||
"title": json_data['title'],
|
||||
'quality': video_quality,
|
||||
"m3u8_url": m3u8_url,
|
||||
"flv_url": flv_url,
|
||||
"record_url": flv_url
|
||||
"record_url": flv_url or m3u8_url
|
||||
}
|
||||
|
||||
|
||||
def get_stream_url(json_data: dict, video_quality: str, url_type: str = 'm3u8', spec: bool = False,
|
||||
extra_key: str | int = None) -> dict:
|
||||
async def get_stream_url(json_data: dict, video_quality: str, url_type: str = 'm3u8', spec: bool = False,
|
||||
hls_extra_key: str | int = None, flv_extra_key: str | int = None) -> dict:
|
||||
if not json_data['is_live']:
|
||||
return json_data
|
||||
|
||||
play_url_list = json_data['play_url_list']
|
||||
quality_list = {'原画': 0, '蓝光': 0, '超清': 1, '高清': 2, '标清': 3, '流畅': 4}
|
||||
while len(play_url_list) < 5:
|
||||
play_url_list.append(play_url_list[-1])
|
||||
|
||||
selected_quality = quality_list[video_quality]
|
||||
video_quality, selected_quality = get_quality_index(video_quality)
|
||||
data = {
|
||||
"anchor_name": json_data['anchor_name'],
|
||||
"is_live": True
|
||||
}
|
||||
if url_type == 'm3u8':
|
||||
m3u8_url = play_url_list[selected_quality][extra_key] if extra_key else play_url_list[selected_quality]
|
||||
|
||||
def get_url(key):
|
||||
play_url = play_url_list[selected_quality]
|
||||
return play_url[key] if key else play_url
|
||||
|
||||
if url_type == 'all':
|
||||
m3u8_url = get_url(hls_extra_key)
|
||||
flv_url = get_url(flv_extra_key)
|
||||
data |= {
|
||||
"m3u8_url": json_data['m3u8_url'] if spec else m3u8_url,
|
||||
"flv_url": json_data['flv_url'] if spec else flv_url,
|
||||
"record_url": m3u8_url
|
||||
}
|
||||
elif url_type == 'm3u8':
|
||||
m3u8_url = get_url(hls_extra_key)
|
||||
data |= {"m3u8_url": json_data['m3u8_url'] if spec else m3u8_url, "record_url": m3u8_url}
|
||||
else:
|
||||
flv_url = play_url_list[selected_quality][extra_key] if extra_key else play_url_list[selected_quality]
|
||||
flv_url = get_url(flv_extra_key)
|
||||
data |= {"flv_url": flv_url, "record_url": flv_url}
|
||||
data['title'] = json_data.get('title')
|
||||
return data
|
||||
data['quality'] = video_quality
|
||||
return data
|
||||
@ -1,18 +1,24 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
import json
|
||||
import os
|
||||
import random
|
||||
import shutil
|
||||
import string
|
||||
from pathlib import Path
|
||||
import functools
|
||||
import hashlib
|
||||
import re
|
||||
import traceback
|
||||
from typing import Any
|
||||
from urllib.parse import parse_qs, urlparse
|
||||
from collections import OrderedDict
|
||||
import execjs
|
||||
from .logger import logger
|
||||
import configparser
|
||||
|
||||
OptionalStr = str | None
|
||||
OptionalDict = dict | None
|
||||
|
||||
|
||||
class Color:
|
||||
RED = "\033[31m"
|
||||
@ -151,3 +157,50 @@ def check_disk_capacity(file_path: str | Path, show: bool = False) -> float:
|
||||
f"Used: {disk_usage.used / (1024 ** 3):.2f} GB "
|
||||
f"Free: {free_space_gb:.2f} GB\n")
|
||||
return free_space_gb
|
||||
|
||||
|
||||
def handle_proxy_addr(proxy_addr):
|
||||
if proxy_addr:
|
||||
if not proxy_addr.startswith('http'):
|
||||
proxy_addr = 'http://' + proxy_addr
|
||||
else:
|
||||
proxy_addr = None
|
||||
return proxy_addr
|
||||
|
||||
|
||||
def generate_random_string(length: int) -> str:
|
||||
characters = string.ascii_uppercase + string.digits
|
||||
random_string = ''.join(random.choices(characters, k=length))
|
||||
return random_string
|
||||
|
||||
|
||||
def jsonp_to_json(jsonp_str: str) -> OptionalDict:
|
||||
pattern = r'(\w+)\((.*)\);?$'
|
||||
match = re.search(pattern, jsonp_str)
|
||||
|
||||
if match:
|
||||
_, json_str = match.groups()
|
||||
json_obj = json.loads(json_str)
|
||||
return json_obj
|
||||
else:
|
||||
raise Exception("No JSON data found in JSONP response.")
|
||||
|
||||
|
||||
def replace_url(file_path: str | Path, old: str, new: str) -> None:
|
||||
with open(file_path, 'r', encoding='utf-8-sig') as f:
|
||||
content = f.read()
|
||||
if old in content:
|
||||
with open(file_path, 'w', encoding='utf-8-sig') as f:
|
||||
f.write(content.replace(old, new))
|
||||
|
||||
|
||||
def get_query_params(url: str, param_name: OptionalStr) -> dict | list[str]:
|
||||
parsed_url = urlparse(url)
|
||||
query_params = parse_qs(parsed_url.query)
|
||||
|
||||
if param_name is None:
|
||||
return query_params
|
||||
else:
|
||||
values = query_params.get(param_name, [])
|
||||
return values
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user