iShotaBot/models/apis/splash.py
2023-07-16 17:49:21 +08:00

71 lines
1.9 KiB
Python
Raw Permalink Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

import re
from datetime import datetime
from typing import Optional
from pydantic import BaseModel
GAME_ID_MAP = {1: "bh3", 2: "ys", 3: "bh2", 4: "wd", 5: "dby", 6: "sr", 8: "zzz"}
class Splash(BaseModel):
id: int
splash_image: str
app_path: str
online_ts: int
offline_ts: int
game_id: Optional[int] = 0
file_id: Optional[str] = ""
@property
def online_time(self) -> datetime:
return datetime.fromtimestamp(int(self.online_ts))
@property
def online_time_str(self) -> str:
return self.online_time.strftime("%Y-%m-%d %H:%M:%S")
@property
def offline_time(self) -> datetime:
return datetime.fromtimestamp(int(self.offline_ts))
@property
def offline_time_str(self) -> str:
return self.offline_time.strftime("%Y-%m-%d %H:%M:%S")
@property
def article_id(self) -> int:
try:
return int(re.search(r"article/(\d+)", self.app_path).group(1))
except AttributeError:
return 0
@property
def game_short_name(self) -> str:
if not self.game_id:
return ""
return GAME_ID_MAP.get(self.game_id)
@property
def article_url(self) -> str:
if self.app_path.startswith("http"):
return self.app_path
if not self.article_id:
return ""
if not self.game_short_name:
return ""
return (
f"https://www.miyoushe.com/{self.game_short_name}/article/{self.article_id}"
)
@property
def text(self) -> str:
return (
f"#id{self.id} \n"
f"ID<code>{self.id}</code>\n"
f"所属分区:<code>{self.game_id} - {self.game_short_name}</code>\n"
f"开始时间:<code>{self.online_time_str}</code>\n"
f"结束时间:<code>{self.offline_time_str}</code>\n"
f"链接: {self.splash_image}\n"
f"文章链接: {self.article_url}"
)