mirror of
https://github.com/PaiGramTeam/PamGram.git
synced 2024-11-22 22:37:21 +00:00
128 lines
3.7 KiB
Python
128 lines
3.7 KiB
Python
from datetime import datetime, timedelta
|
|
from io import BytesIO
|
|
from typing import Any, List, Optional
|
|
|
|
from PIL import Image, UnidentifiedImageError
|
|
from pydantic import BaseModel, PrivateAttr
|
|
|
|
__all__ = ("ArtworkImage", "PostInfo", "LiveInfo", "LiveCode", "LiveCodeHoYo")
|
|
|
|
|
|
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()
|
|
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) -> "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["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,
|
|
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]
|
|
|
|
|
|
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)
|