mirror of
https://github.com/NanmiCoder/MediaCrawler.git
synced 2025-11-25 03:15:17 +08:00
56 lines
2.0 KiB
Python
56 lines
2.0 KiB
Python
import json
|
|
import time
|
|
import base64
|
|
import random
|
|
import hashlib
|
|
from io import BytesIO
|
|
from typing import Optional, Dict, List, Tuple
|
|
|
|
from PIL import Image
|
|
from playwright.async_api import Cookie
|
|
from playwright.async_api import Page
|
|
|
|
|
|
async def find_login_qrcode(page: Page, selector: str) -> str:
|
|
"""find login qrcode image from target selector"""
|
|
try:
|
|
elements = await page.wait_for_selector(
|
|
selector=selector,
|
|
)
|
|
login_qrcode_img = await elements.get_property("src")
|
|
return str(login_qrcode_img)
|
|
|
|
except Exception as e:
|
|
print(e)
|
|
return ""
|
|
|
|
|
|
def show_qrcode(qr_code: str):
|
|
"""parse base64 encode qrcode image and show it"""
|
|
qr_code = qr_code.split(",")[1]
|
|
qr_code = base64.b64decode(qr_code)
|
|
image = Image.open(BytesIO(qr_code))
|
|
image.show()
|
|
|
|
|
|
def get_user_agent() -> str:
|
|
ua_list = [
|
|
"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/114.0.0.0 Safari/537.36",
|
|
"Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/104.0.5112.79 Safari/537.36",
|
|
"Mozilla/5.0 (Windows NT 10.0) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/104.0.0.0 Safari/537.36",
|
|
"Mozilla/5.0 (Macintosh; Intel Mac OS X 10_14_6) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/104.0.0.0 Safari/537.36",
|
|
"Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/103.0.5060.53 Safari/537.36",
|
|
"Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_3) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/99.0.4844.84 Safari/537.36"
|
|
]
|
|
return random.choice(ua_list)
|
|
|
|
|
|
def convert_cookies(cookies: Optional[List[Cookie]]) -> Tuple[str, Dict]:
|
|
if not cookies:
|
|
return "", {}
|
|
cookies_str = ";".join([f"{cookie.get('name')}={cookie.get('value')}" for cookie in cookies])
|
|
cookie_dict = dict()
|
|
for cookie in cookies:
|
|
cookie_dict[cookie.get('name')] = cookie.get('value')
|
|
return cookies_str, cookie_dict
|