86 lines
1.9 KiB
Python
86 lines
1.9 KiB
Python
from abc import ABC, abstractmethod
|
|
from typing import Dict, Optional
|
|
|
|
from playwright.async_api import BrowserContext, BrowserType
|
|
|
|
|
|
class AbstractCrawler(ABC):
|
|
@abstractmethod
|
|
async def start(self):
|
|
"""
|
|
start crawler
|
|
"""
|
|
pass
|
|
|
|
@abstractmethod
|
|
async def search(self):
|
|
"""
|
|
search
|
|
"""
|
|
pass
|
|
|
|
@abstractmethod
|
|
async def launch_browser(self, chromium: BrowserType, playwright_proxy: Optional[Dict], user_agent: Optional[str],
|
|
headless: bool = True) -> BrowserContext:
|
|
"""
|
|
launch browser
|
|
:param chromium: chromium browser
|
|
:param playwright_proxy: playwright proxy
|
|
:param user_agent: user agent
|
|
:param headless: headless mode
|
|
:return: browser context
|
|
"""
|
|
pass
|
|
|
|
|
|
class AbstractLogin(ABC):
|
|
@abstractmethod
|
|
async def begin(self):
|
|
pass
|
|
|
|
@abstractmethod
|
|
async def login_by_qrcode(self):
|
|
pass
|
|
|
|
@abstractmethod
|
|
async def login_by_mobile(self):
|
|
pass
|
|
|
|
@abstractmethod
|
|
async def login_by_cookies(self):
|
|
pass
|
|
|
|
|
|
class AbstractStore(ABC):
|
|
@abstractmethod
|
|
async def store_content(self, content_item: Dict):
|
|
pass
|
|
|
|
@abstractmethod
|
|
async def store_comment(self, comment_item: Dict):
|
|
pass
|
|
|
|
# TODO support all platform
|
|
# only xhs is supported, so @abstractmethod is commented
|
|
@abstractmethod
|
|
async def store_creator(self, creator: Dict):
|
|
pass
|
|
|
|
|
|
class AbstractStoreImage(ABC):
|
|
# TODO: support all platform
|
|
# only weibo is supported
|
|
# @abstractmethod
|
|
async def store_image(self, image_content_item: Dict):
|
|
pass
|
|
|
|
|
|
class AbstractApiClient(ABC):
|
|
@abstractmethod
|
|
async def request(self, method, url, **kwargs):
|
|
pass
|
|
|
|
@abstractmethod
|
|
async def update_cookies(self, browser_context: BrowserContext):
|
|
pass
|