diff --git a/py/tests/test_飞快TV.py b/py/tests/test_飞快TV.py new file mode 100644 index 0000000..6d6f103 --- /dev/null +++ b/py/tests/test_飞快TV.py @@ -0,0 +1,30 @@ +import unittest +from importlib.machinery import SourceFileLoader +from pathlib import Path + + +ROOT = Path(__file__).resolve().parents[1] +MODULE = SourceFileLoader("feikuai_spider", str(ROOT / "飞快TV.py")).load_module() +Spider = MODULE.Spider + + +class TestFeikuaiSpider(unittest.TestCase): + def setUp(self): + Spider._instance = None + self.spider = Spider() + self.spider.init() + + def test_home_content_returns_expected_classes(self): + result = self.spider.homeContent(False) + self.assertEqual( + result["class"], + [ + {"type_id": "1", "type_name": "电影"}, + {"type_id": "2", "type_name": "剧集"}, + {"type_id": "3", "type_name": "综艺"}, + {"type_id": "4", "type_name": "动漫"}, + ], + ) + + def test_home_video_content_returns_empty_list(self): + self.assertEqual(self.spider.homeVideoContent(), {"list": []}) diff --git a/py/飞快TV.py b/py/飞快TV.py new file mode 100644 index 0000000..5ed55b1 --- /dev/null +++ b/py/飞快TV.py @@ -0,0 +1,40 @@ +# coding=utf-8 +import sys + +from base.spider import Spider as BaseSpider + +sys.path.append("..") + + +class Spider(BaseSpider): + def __init__(self): + self.name = "飞快TV" + self.host = "https://feikuai.tv" + self.user_agent = ( + "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) " + "AppleWebKit/537.36 (KHTML, like Gecko) " + "Chrome/130.0.0.0 Safari/537.36" + ) + self.headers = { + "User-Agent": self.user_agent, + "Referer": self.host + "/", + "Origin": self.host, + } + self.classes = [ + {"type_id": "1", "type_name": "电影"}, + {"type_id": "2", "type_name": "剧集"}, + {"type_id": "3", "type_name": "综艺"}, + {"type_id": "4", "type_name": "动漫"}, + ] + + def init(self, extend=""): + return None + + def getName(self): + return self.name + + def homeContent(self, filter): + return {"class": self.classes} + + def homeVideoContent(self): + return {"list": []}