yeyinly/tvboxzt
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

在线之家

Browse Source 
短剧优选
This commit is contained in:
Harold
2026-04-20 11:34:26 +08:00
parent d9b216d3dd
commit 9ba826e2c4
4 changed files with 1402 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
py/tests/test_在线之家.py
+68
View File
@@ -1,6 +1,7 @@
import unittest
from importlib.machinery import SourceFileLoader
from pathlib import Path
from types import SimpleNamespace
from unittest.mock import patch
@@ -177,6 +178,19 @@ class TestZXZJSpider(unittest.TestCase):
"https://jx.zxzj.example/player?id=1",
)
def test_extract_iframe_url_reads_player_aaaa_target(self):
html = """
<script>
var player_aaaa = {
"url":"https://jx.zxzjys.com:9876/player-v2/common/demo/token"
};
</script>
"""
self.assertEqual(
self.spider._extract_iframe_url(html),
"https://jx.zxzjys.com:9876/player-v2/common/demo/token",
)
@patch.object(Spider, "_request_html")
def test_player_content_returns_decoded_direct_url_for_zxzj(self, mock_request_html):
mock_request_html.side_effect = [
@@ -200,6 +214,60 @@ class TestZXZJSpider(unittest.TestCase):
self.assertEqual(result["jx"], 1)
self.assertEqual(result["url"], "https://www.zxzjhd.com/vodplay/999-1-1.html")
@patch.object(Spider, "_request_html")
def test_player_content_falls_back_to_parser_page_when_new_line5_target_cannot_be_decoded(self, mock_request_html):
mock_request_html.side_effect = [
"""
<script>
var player_aaaa = {
"url":"https://jx.zxzjys.com:9876/player-v2/common/demo/token"
};
</script>
""",
"",
]
result = self.spider.playerContent("zxzj", "vodplay/4627-1-1.html", {})
self.assertEqual(result["parse"], 1)
self.assertEqual(result["jx"], 1)
self.assertEqual(result["url"], "https://jx.zxzjys.com:9876/player-v2/common/demo/token")
self.assertEqual(result["header"]["Referer"], "https://www.zxzjhd.com/vodplay/4627-1-1.html")
@patch.object(Spider, "fetch")
def test_player_content_uses_browser_like_headers_for_new_line5_parser_request(self, mock_fetch):
play_page = """
<script>
var player_aaaa = {
"url":"https://jx.zxzjys.com:9876/player-v2/common/demo/token"
};
</script>
"""
parser_page = """
<script>
var result_v2 = {
"data":"d6f636e256c607d6168756e2f67464544434241456469667f2f2a33707474786"
};
</script>
"""
def fake_fetch(url, headers=None, timeout=10):
if url == "https://www.zxzjhd.com/vodplay/4627-1-1.html":
return SimpleNamespace(status_code=200, text=play_page)
if url == "https://jx.zxzjys.com:9876/player-v2/common/demo/token":
self.assertEqual(headers["Referer"], "https://www.zxzjhd.com/vodplay/4627-1-1.html")
self.assertEqual(headers["Origin"], "https://www.zxzjhd.com")
self.assertEqual(headers["Sec-Fetch-Dest"], "iframe")
self.assertEqual(headers["Sec-Fetch-Mode"], "navigate")
self.assertEqual(headers["Sec-Fetch-Site"], "cross-site")
self.assertEqual(headers["Upgrade-Insecure-Requests"], "1")
return SimpleNamespace(status_code=200, text=parser_page)
raise AssertionError(url)
mock_fetch.side_effect = fake_fetch
result = self.spider.playerContent("zxzj", "vodplay/4627-1-1.html", {})
self.assertEqual(result["parse"], 0)
self.assertEqual(result["jx"], 0)
self.assertEqual(result["url"], "https://video.example.com")
def test_player_content_returns_direct_netdisk_link(self):
result = self.spider.playerContent("quark", "https://pan.quark.cn/s/demo", {})
self.assertEqual(
py/tests/test_短剧优选.py
+241
View File
@@ -0,0 +1,241 @@
import unittest
import json
from importlib.machinery import SourceFileLoader
from pathlib import Path
from unittest.mock import patch, MagicMock
ROOT = Path(__file__).resolve().parents[1]
MODULE = SourceFileLoader("duanjuyx_spider", str(ROOT / "短剧优选.py")).load_module()
Spider = MODULE.Spider
class TestDuanjuYouxuanSpider(unittest.TestCase):
def setUp(self):
Spider._instance = None
self.spider = Spider()
self.spider.init()
def test_home_content_has_active_platforms(self):
content = self.spider.homeContent(False)
ids = [c["type_id"] for c in content["class"]]
self.assertIn("七猫", ids)
self.assertIn("星芽", ids)
self.assertIn("西饭", ids)
self.assertIn("七星", ids)
self.assertIn("短剧网", ids)
def test_home_content_has_filters(self):
content = self.spider.homeContent(False)
self.assertIn("七猫", content["filters"])
self.assertIn("短剧网", content["filters"])
qm_filter = content["filters"]["七猫"][0]
self.assertEqual(qm_filter["key"], "area")
self.assertTrue(len(qm_filter["value"]) > 50)
def test_home_video_content_returns_empty(self):
self.assertEqual(self.spider.homeVideoContent(), {"list": []})
def test_player_tianquan(self):
result = self.spider.playerContent("甜圈短剧", "vid123", {})
self.assertEqual(result["parse"], 0)
self.assertIn("video_id=vid123", result["url"])
def test_player_xingya_passthrough(self):
result = self.spider.playerContent("星芽短剧", "https://example.com/video.m3u8", {})
self.assertEqual(result["url"], "https://example.com/video.m3u8")
def test_player_weiguan_selects_super(self):
play_setting = {"super": "s.mp4", "high": "h.mp4", "normal": "n.mp4"}
result = self.spider.playerContent("围观短剧", json.dumps(play_setting), {})
self.assertEqual(result["url"], "s.mp4")
def test_player_weiguan_falls_back_to_high(self):
play_setting = {"high": "h.mp4", "normal": "n.mp4"}
result = self.spider.playerContent("围观短剧", json.dumps(play_setting), {})
self.assertEqual(result["url"], "h.mp4")
def test_player_weiguan_invalid_json(self):
result = self.spider.playerContent("围观短剧", "not-json", {})
self.assertEqual(result["url"], "not-json")
def test_player_duanjuwang_passthrough(self):
result = self.spider.playerContent("短剧网", "push://pan.quark.cn/abc", {})
self.assertEqual(result["parse"], 0)
self.assertEqual(result["url"], "push://pan.quark.cn/abc")
@patch.object(Spider, "_get_text")
def test_player_juwang_extracts_m3u8(self, mock_text):
mock_text.return_value = 'var data = {"wwm3u8": "https:\\/\\/cdn.example.com\\/video.m3u8"};'
result = self.spider.playerContent("剧王短剧", "https://djw1.com/play/123", {})
self.assertEqual(result["url"], "https://cdn.example.com/video.m3u8")
self.assertEqual(result["parse"], 0)
@patch.object(Spider, "_get_json")
def test_category_tianquan(self, mock_json):
mock_json.return_value = {
"data": [
{"book_id": "b1", "title": "逆袭人生", "cover": "http://pic/1.jpg", "sub_title": "60集"},
{"book_id": "b2", "title": "霸总归来", "cover": "http://pic/2.jpg", "sub_title": "80集"},
]
}
result = self.spider.categoryContent("甜圈", "1", False, {"area": "逆袭"})
self.assertEqual(len(result["list"]), 2)
self.assertEqual(result["list"][0]["vod_id"], "甜圈@b1")
self.assertEqual(result["list"][0]["vod_name"], "逆袭人生")
self.assertEqual(result["pagecount"], 2)
@patch.object(Spider, "_post_json")
def test_category_weiguan(self, mock_json):
mock_json.return_value = {
"data": [
{"oneId": "wg1", "title": "都市情缘", "vertPoster": "http://pic/wg.jpg", "episodeCount": 50},
]
}
result = self.spider.categoryContent("围观", "1", False, {"area": "都市"})
self.assertEqual(len(result["list"]), 1)
self.assertEqual(result["list"][0]["vod_id"], "围观@wg1")
@patch.object(Spider, "_get_text")
def test_category_duanjuwang(self, mock_text):
mock_text.return_value = """
<li class="col-6">
<h3 class="f-14"><a href="/detail/123.html">短剧名（60集完结）</a></h3>
<img class="lazy" data-original="/cover.jpg" />
</li>
"""
result = self.spider.categoryContent("短剧网", "1", False, {"area": "1"})
self.assertEqual(len(result["list"]), 1)
self.assertEqual(result["list"][0]["vod_id"], "短剧网@https://sm3.cc/detail/123.html")
self.assertEqual(result["list"][0]["vod_name"], "短剧名（60集完结）")
self.assertEqual(result["list"][0]["vod_remarks"], "")
self.assertEqual(result["pagecount"], 2)
@patch.object(Spider, "_get_json")
def test_category_empty_result(self, mock_json):
mock_json.return_value = {"data": []}
result = self.spider.categoryContent("甜圈", "2", False, {"area": "逆袭"})
self.assertEqual(result["list"], [])
self.assertEqual(result["pagecount"], 2)
@patch.object(Spider, "_get_json")
def test_detail_tianquan(self, mock_json):
mock_json.return_value = {
"book_name": "甜剧1号", "book_pic": "http://pic/t.jpg", "desc": "剧情简介",
"duration": "60分钟", "author": "作者", "time": "2025-01-01",
"data": [
{"title": "第1集", "video_id": "v1"},
{"title": "第2集", "video_id": "v2"},
],
}
result = self.spider.detailContent(["甜圈@b1"])
vod = result["list"][0]
self.assertEqual(vod["vod_name"], "甜剧1号")
self.assertEqual(vod["vod_play_from"], "甜圈短剧")
self.assertIn("第1集$v1", vod["vod_play_url"])
self.assertIn("第2集$v2", vod["vod_play_url"])
@patch.object(Spider, "_get_json")
def test_detail_weiguan(self, mock_json):
mock_json.return_value = {
"data": [
{"title": "逆袭", "playOrder": 1, "playSetting": {"super": "s1.mp4"}, "vertPoster": "p.jpg", "collectionCount": 100},
{"title": "逆袭", "playOrder": 2, "playSetting": {"super": "s2.mp4"}, "vertPoster": "p.jpg", "collectionCount": 100},
]
}
result = self.spider.detailContent(["围观@wg1"])
vod = result["list"][0]
self.assertEqual(vod["vod_name"], "逆袭")
self.assertEqual(vod["vod_remarks"], "共2集")
self.assertIn("围观短剧", vod["vod_play_from"])
@patch.object(Spider, "_get_text")
def test_detail_juwang(self, mock_text):
mock_text.return_value = """
<html><body>
<h1>剧王短剧名</h1>
<img src="/poster.jpg" />
<div class="info-detail">剧情简介</div>
<div class="ep-list">
<a href="/play/1-1.html">第1集</a>
<a href="/play/1-2.html">第2集</a>
</div>
</body></html>
"""
result = self.spider.detailContent(["剧王@/show/123"])
vod = result["list"][0]
self.assertEqual(vod["vod_name"], "剧王短剧名")
self.assertEqual(vod["vod_content"], "剧情简介")
self.assertEqual(vod["vod_play_from"], "剧王短剧")
self.assertIn("第1集$/play/1-1.html", vod["vod_play_url"])
@patch.object(Spider, "_get_text")
def test_detail_duanjuwang(self, mock_text):
mock_text.return_value = """
<html><body>
<h1>网盘短剧名</h1>
<img class="lazy" data-original="/poster.jpg" />
<div class="content">
<a href="https://pan.quark.cn/s/abc123">夸克链接</a>
<a href="https://pan.baidu.com/s/xyz789">百度链接</a>
</div>
</body></html>
"""
result = self.spider.detailContent(["短剧网@https://sm3.cc/detail/123"])
vod = result["list"][0]
self.assertEqual(vod["vod_name"], "网盘短剧名")
self.assertEqual(vod["vod_content"], "此为推送网盘规则")
self.assertIn("quark", vod["vod_play_from"])
self.assertIn("baidu", vod["vod_play_from"])
self.assertIn("quark$https://pan.quark.cn/s/abc123", vod["vod_play_url"])
@patch.object(Spider, "_get_text")
def test_detail_duanjuwang_no_netdisk_links(self, mock_text):
mock_text.return_value = """
<html><body>
<h1>无链接短剧</h1>
<div class="content"><p>暂无链接</p></div>
</body></html>
"""
result = self.spider.detailContent(["短剧网@https://sm3.cc/detail/456"])
vod = result["list"][0]
self.assertEqual(vod["vod_name"], "无链接短剧")
self.assertEqual(vod["vod_play_from"], "短剧网")
self.assertEqual(vod["vod_play_url"], "")
@patch.object(Spider, "_get_json")
def test_detail_failure_returns_error_vod(self, mock_json):
mock_json.side_effect = Exception("network error")
result = self.spider.detailContent(["甜圈@missing"])
vod = result["list"][0]
self.assertIn("详情加载失败", vod["vod_name"])
@patch.object(Spider, "_aggregate_search")
def test_search_filters_by_keyword(self, mock_search):
mock_search.return_value = [
{"vod_id": "甜圈@1", "vod_name": "逆袭人生", "vod_pic": "", "vod_remarks": ""},
{"vod_id": "七猫@2", "vod_name": "霸道总裁", "vod_pic": "", "vod_remarks": ""},
]
result = self.spider.searchContent("逆袭", False, "1")
self.assertEqual(len(result["list"]), 1)
self.assertEqual(result["list"][0]["vod_name"], "逆袭人生")
@patch.object(Spider, "_aggregate_search")
def test_search_empty_keyword(self, mock_search):
result = self.spider.searchContent("", False, "1")
self.assertEqual(result["list"], [])
mock_search.assert_not_called()
def test_filter_defaults(self):
self.assertEqual(self.spider.filter_defaults["七猫"]["area"], "0")
self.assertEqual(self.spider.filter_defaults["围观"]["area"], "都市")
self.assertEqual(self.spider.filter_defaults["短剧网"]["area"], "1")
def test_identify_disk(self):
self.assertEqual(self.spider._identify_disk("https://pan.quark.cn/s/abc"), "quark")
self.assertEqual(self.spider._identify_disk("https://pan.baidu.com/s/xyz"), "baidu")
self.assertEqual(self.spider._identify_disk("https://drive.uc.cn/s/123"), "uc")
self.assertEqual(self.spider._identify_disk("https://example.com/video.mp4"), "")
if __name__ == "__main__":
unittest.main()