mirror of
https://github.com/JoeanAmier/XHS-Downloader.git
synced 2025-12-26 04:48:05 +08:00
62 lines
1.5 KiB
Python
62 lines
1.5 KiB
Python
from contextlib import suppress
|
|
|
|
from rookiepy import (
|
|
arc,
|
|
brave,
|
|
chrome,
|
|
chromium,
|
|
edge,
|
|
firefox,
|
|
librewolf,
|
|
opera,
|
|
# opera_gx,
|
|
# safari,
|
|
vivaldi,
|
|
)
|
|
|
|
__all__ = ["BrowserCookie"]
|
|
|
|
|
|
class BrowserCookie:
|
|
SUPPORT_BROWSER = {
|
|
"arc": arc,
|
|
"chrome": chrome,
|
|
"chromium": chromium,
|
|
"opera": opera,
|
|
# "opera_gx": opera_gx,
|
|
"brave": brave,
|
|
"edge": edge,
|
|
"vivaldi": vivaldi,
|
|
"firefox": firefox,
|
|
"librewolf": librewolf,
|
|
# "safari": safari,
|
|
}
|
|
|
|
@classmethod
|
|
def get(cls, browser: str | int, domains: list[str]) -> str:
|
|
if not (browser := cls.__browser_object(browser)):
|
|
print("浏览器名称或序号输入错误！")
|
|
return ""
|
|
try:
|
|
cookies = browser(domains=domains)
|
|
return "; ".join(f"{i["name"]}={i["value"]}" for i in cookies)
|
|
except RuntimeError:
|
|
print("获取 Cookie 失败，未找到 Cookie 数据！")
|
|
return ""
|
|
|
|
@classmethod
|
|
def __browser_object(cls, browser: str | int):
|
|
with suppress(ValueError):
|
|
browser = int(browser) - 1
|
|
if isinstance(browser, int):
|
|
try:
|
|
return list(cls.SUPPORT_BROWSER.values())[browser - 1]
|
|
except IndexError:
|
|
return None
|
|
if isinstance(browser, str):
|
|
try:
|
|
return cls.SUPPORT_BROWSER[browser.lower()]
|
|
except KeyError:
|
|
return None
|
|
raise TypeError
|