mirror of
https://github.com/RYDE-WORK/MediaCrawler.git
synced 2026-01-30 04:13:13 +08:00
57 lines
1.8 KiB
Python
57 lines
1.8 KiB
Python
import asyncio
|
|
import functools
|
|
import sys
|
|
from typing import Optional
|
|
|
|
import redis
|
|
from playwright.async_api import BrowserContext, Page
|
|
from tenacity import (RetryError, retry, retry_if_result, stop_after_attempt,
|
|
wait_fixed)
|
|
|
|
import config
|
|
from base.base_crawler import AbstractLogin
|
|
from tools import utils
|
|
|
|
|
|
class KuaishouLogin(AbstractLogin):
|
|
def __init__(self,
|
|
login_type: str,
|
|
browser_context: BrowserContext,
|
|
context_page: Page,
|
|
login_phone: Optional[str] = "",
|
|
cookie_str: str = ""
|
|
):
|
|
self.login_type = login_type
|
|
self.browser_context = browser_context
|
|
self.context_page = context_page
|
|
self.login_phone = login_phone
|
|
self.cookie_str = cookie_str
|
|
|
|
async def begin(self):
|
|
"""Start login xiaohongshu"""
|
|
utils.logger.info("Begin login kuaishou ...")
|
|
if self.login_type == "qrcode":
|
|
await self.login_by_qrcode()
|
|
elif self.login_type == "phone":
|
|
await self.login_by_mobile()
|
|
elif self.login_type == "cookie":
|
|
await self.login_by_cookies()
|
|
else:
|
|
raise ValueError("Invalid Login Type Currently only supported qrcode or phone or cookie ...")
|
|
|
|
async def login_by_qrcode(self):
|
|
pass
|
|
|
|
async def login_by_mobile(self):
|
|
pass
|
|
|
|
async def login_by_cookies(self):
|
|
utils.logger.info("Begin login kuaishou by cookie ...")
|
|
for key, value in utils.convert_str_cookie_to_dict(self.cookie_str).items():
|
|
await self.browser_context.add_cookies([{
|
|
'name': key,
|
|
'value': value,
|
|
'domain': ".douyin.com",
|
|
'path': "/"
|
|
}])
|