Files
XHS_Downloader/source/application/image.py
JoeanAmier 3e8b69f8e1 fix: 修复项目功能异常
1. 更新 Cookie 参数处理
2. 优化作品数据返回格式
3. 更新用户脚本链接提取功能
4. 优化项目运行信息提示
5. 修复 record_data 参数无效的问题
6. 新增作品下载记录开关
7. 适配小红书平台规则
8. 默认开启局域网访问
9. 格式化项目代码
10. 更新英语翻译

Closes #127
Closes #128
Closes #130
Closes #132
2024-08-03 14:30:35 +08:00

49 lines
1.5 KiB
Python

from source.expansion import Namespace
from .request import Html
__all__ = ['Image']
class Image:
@classmethod
def get_image_link(cls, data: Namespace, format_: str) -> [list, list]:
images = data.safe_extract("imageList", [])
live_link = cls.__get_live_link(images)
token_list = [
cls.__extract_image_token(
Namespace.object_extract(
i, "urlDefault")) for i in images]
match format_:
case "png":
return [Html.format_url(cls.__generate_png_link(i))
for i in token_list], live_link
case "webp":
return [Html.format_url(cls.__generate_webp_link(i))
for i in token_list], live_link
case _:
raise ValueError
@staticmethod
def __generate_webp_link(token: str) -> str:
return f"https://sns-img-bd.xhscdn.com/{token}"
@staticmethod
def __generate_png_link(token: str) -> str:
return f"https://ci.xiaohongshu.com/{token}?imageView2/2/w/format/png"
@staticmethod
def __extract_image_token(url: str) -> str:
return "/".join(url.split("/")[5:]).split("!")[0]
@staticmethod
def __get_live_link(items: list) -> list:
return [
(
Html.format_url(
Namespace.object_extract(item, "stream.h264[0].masterUrl")
)
or None
)
for item in items
]