mirror of
https://github.com/PaiGramTeam/PamGram.git
synced 2024-11-24 15:19:20 +00:00
176 lines
4.8 KiB
Python
176 lines
4.8 KiB
Python
from datetime import datetime, timedelta
|
|
from enum import Enum
|
|
from io import BytesIO
|
|
from typing import Any, List, Optional, Dict
|
|
|
|
from PIL import Image, UnidentifiedImageError
|
|
from pydantic import BaseModel, PrivateAttr
|
|
|
|
__all__ = (
|
|
"ArtworkImage",
|
|
"PostInfo",
|
|
"LiveInfo",
|
|
"LiveCode",
|
|
"LiveCodeHoYo",
|
|
"PostTypeEnum",
|
|
"PostRecommend",
|
|
"HoYoPostMultiLang",
|
|
)
|
|
|
|
|
|
class ArtworkImage(BaseModel):
|
|
art_id: int
|
|
page: int = 0
|
|
data: bytes = b""
|
|
file_name: Optional[str] = None
|
|
file_extension: Optional[str] = None
|
|
is_error: bool = False
|
|
|
|
@property
|
|
def format(self) -> Optional[str]:
|
|
if not self.is_error:
|
|
try:
|
|
with BytesIO(self.data) as stream, Image.open(stream) as im:
|
|
return im.format
|
|
except UnidentifiedImageError:
|
|
pass
|
|
return None
|
|
|
|
@staticmethod
|
|
def gen(*args, **kwargs) -> List["ArtworkImage"]:
|
|
data = [ArtworkImage(*args, **kwargs)]
|
|
if data[0].file_extension and data[0].file_extension in ["gif", "mp4"]:
|
|
return data
|
|
try:
|
|
with BytesIO(data[0].data) as stream, Image.open(stream) as image:
|
|
width, height = image.size
|
|
crop_height = height
|
|
crop_num = 1
|
|
max_height = 10000 - width
|
|
while crop_height > max_height:
|
|
crop_num += 1
|
|
crop_height = height / crop_num
|
|
new_data = []
|
|
for i in range(crop_num):
|
|
slice_image = image.crop((0, crop_height * i, width, crop_height * (i + 1)))
|
|
bio = BytesIO()
|
|
slice_image.save(bio, "png")
|
|
kwargs["data"] = bio.getvalue()
|
|
kwargs["file_extension"] = "png"
|
|
new_data.append(ArtworkImage(*args, **kwargs))
|
|
return new_data
|
|
except UnidentifiedImageError:
|
|
return data
|
|
|
|
|
|
class PostInfo(BaseModel):
|
|
_data: dict = PrivateAttr()
|
|
hoyolab: bool
|
|
post_id: int
|
|
user_uid: int
|
|
subject: str
|
|
image_urls: List[str]
|
|
created_at: int
|
|
video_urls: List[str]
|
|
|
|
def __init__(self, _data: dict, **data: Any):
|
|
super().__init__(**data)
|
|
self._data = _data
|
|
|
|
@classmethod
|
|
def paste_data(cls, data: dict, hoyolab: bool = False) -> "PostInfo":
|
|
_data_post = data["post"]
|
|
post = _data_post["post"]
|
|
post_id = post["post_id"]
|
|
subject = post["subject"]
|
|
image_list = _data_post["image_list"]
|
|
image_urls = [image["url"] for image in image_list]
|
|
vod_list = _data_post.get("vod_list", [])
|
|
video_urls = [vod["resolutions"][-1]["url"] for vod in vod_list]
|
|
created_at = post["created_at"]
|
|
user = _data_post["user"] # 用户数据
|
|
user_uid = user["uid"] # 用户ID
|
|
return PostInfo(
|
|
_data=data,
|
|
hoyolab=hoyolab,
|
|
post_id=post_id,
|
|
user_uid=user_uid,
|
|
subject=subject,
|
|
image_urls=image_urls,
|
|
video_urls=video_urls,
|
|
created_at=created_at,
|
|
)
|
|
|
|
def __getitem__(self, item):
|
|
return self._data[item]
|
|
|
|
@property
|
|
def type_enum(self) -> "PostTypeEnum":
|
|
return PostTypeEnum.CN if not self.hoyolab else PostTypeEnum.OS
|
|
|
|
def get_url(self, short_name: str) -> str:
|
|
if not self.hoyolab:
|
|
return f"https://www.miyoushe.com/{short_name}/article/{self.post_id}"
|
|
return f"https://www.hoyolab.com/article/{self.post_id}"
|
|
|
|
def get_fix_url(self, short_name: str) -> str:
|
|
url = self.get_url(short_name)
|
|
return url.replace(".com/", ".pp.ua/")
|
|
|
|
|
|
class LiveInfo(BaseModel):
|
|
act_type: str
|
|
title: str
|
|
live_time: str
|
|
start: datetime
|
|
end: datetime
|
|
remain: int
|
|
now: datetime
|
|
is_end: bool
|
|
code_ver: str
|
|
|
|
|
|
class LiveCode(BaseModel):
|
|
code: str
|
|
to_get_time: datetime
|
|
|
|
@property
|
|
def text(self) -> str:
|
|
return self.code if self.code else "XXXXXXXXXXXX"
|
|
|
|
|
|
class LiveCodeHoYo(BaseModel):
|
|
exchange_code: str
|
|
offline_at: datetime
|
|
|
|
@property
|
|
def text(self) -> str:
|
|
return self.exchange_code if self.exchange_code else "XXXXXXXXXXXX"
|
|
|
|
@staticmethod
|
|
def guess_offline_at() -> datetime:
|
|
return datetime.now().replace(hour=23, minute=59, second=59, microsecond=999999) + timedelta(days=1)
|
|
|
|
|
|
class PostTypeEnum(str, Enum):
|
|
"""社区类型枚举"""
|
|
|
|
NULL = "null"
|
|
CN = "cn"
|
|
OS = "os"
|
|
|
|
|
|
class HoYoPostMultiLang(BaseModel):
|
|
lang_subject: dict
|
|
|
|
|
|
class PostRecommend(BaseModel):
|
|
hoyolab: bool = False
|
|
post_id: int
|
|
|
|
@staticmethod
|
|
def parse(data: Dict, hoyolab: bool = False):
|
|
_post = data.get("post")
|
|
post_id = _post.get("post_id")
|
|
return PostRecommend(hoyolab=hoyolab, post_id=post_id)
|