From 434b30711dd7e83490dc9c5073a0d13c3a560691 Mon Sep 17 00:00:00 2001 From: Harold <8866033@gmail.com> Date: Fri, 24 Apr 2026 18:19:58 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20finish=20=E5=9B=9B=E4=B8=87=E5=BD=B1?= =?UTF-8?q?=E8=A7=86=20player=20handling?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- py/tests/test_四万影视.py | 40 +++++++++++++++++++++++++++++++++++++++ py/四万影视.py | 12 ++++++++++++ 2 files changed, 52 insertions(+) diff --git a/py/tests/test_四万影视.py b/py/tests/test_四万影视.py index 5f6f775..5fd903c 100644 --- a/py/tests/test_四万影视.py +++ b/py/tests/test_四万影视.py @@ -163,3 +163,43 @@ class TestSiWanSpider(unittest.TestCase): vod["vod_play_url"], "第1集$https://cdn.example/1.m3u8#第2集$play-2$$$正片$play-3", ) + + @patch.object(Spider, "_api_get") + def test_category_content_returns_empty_payload_on_api_error(self, mock_api_get): + mock_api_get.side_effect = ValueError("boom") + self.assertEqual( + self.spider.categoryContent("20", "1", False, {}), + {"page": 1, "limit": 0, "total": 0, "list": []}, + ) + + @patch.object(Spider, "_api_get") + def test_detail_content_returns_empty_list_on_api_error(self, mock_api_get): + mock_api_get.side_effect = ValueError("boom") + self.assertEqual(self.spider.detailContent(["9"]), {"list": []}) + + def test_player_content_returns_direct_url_for_http_play_id(self): + result = self.spider.playerContent("量子", "https://cdn.example/test.m3u8", {}) + self.assertEqual(result["parse"], 0) + self.assertEqual(result["jx"], 0) + self.assertEqual(result["url"], "https://cdn.example/test.m3u8") + self.assertEqual(result["header"]["Referer"], "https://40000.me/") + + def test_player_content_marks_non_http_play_id_for_parse(self): + result = self.spider.playerContent("量子", "play-2", {}) + self.assertEqual(result["parse"], 1) + self.assertEqual(result["jx"], 1) + self.assertEqual(result["url"], "play-2") + + def test_player_content_returns_empty_parse_payload_for_blank_id(self): + self.assertEqual( + self.spider.playerContent("量子", "", {}), + { + "parse": 1, + "jx": 1, + "url": "", + "header": { + "User-Agent": self.spider.headers["User-Agent"], + "Referer": "https://40000.me/", + }, + }, + ) diff --git a/py/四万影视.py b/py/四万影视.py index 2c615c4..6baadb4 100644 --- a/py/四万影视.py +++ b/py/四万影视.py @@ -281,3 +281,15 @@ class Spider(BaseSpider): } ] } + + def playerContent(self, flag, id, vipFlags): + play_id = str(id or "").strip() + header = { + "User-Agent": self.headers["User-Agent"], + "Referer": self.host + "/", + } + if not play_id: + return {"parse": 1, "jx": 1, "url": "", "header": header} + if play_id.startswith("http://") or play_id.startswith("https://"): + return {"parse": 0, "jx": 0, "url": play_id, "header": header} + return {"parse": 1, "jx": 1, "url": play_id, "header": header}