mirror of
https://github.com/PaiGramTeam/PamGram.git
synced 2024-11-26 15:59:32 +00:00
233e7ab58d
Co-authored-by: luoshuijs <luoshuijs@outlook.com> Co-authored-by: Karako <karakohear@gmail.com> Co-authored-by: xtaodada <xtao@xtaolink.cn>
102 lines
4.2 KiB
Python
102 lines
4.2 KiB
Python
from typing import List, Optional
|
|
|
|
from core.base_service import BaseService
|
|
from core.services.game.cache import GameCacheForMaterial, GameCacheForStrategy
|
|
from modules.apihelper.client.components.hyperion import Hyperion
|
|
|
|
__all__ = ("GameMaterialService", "GameStrategyService")
|
|
|
|
|
|
class GameStrategyService(BaseService):
|
|
def __init__(self, cache: GameCacheForStrategy, collections: Optional[List[int]] = None):
|
|
self._cache = cache
|
|
self._hyperion = Hyperion()
|
|
if collections is None:
|
|
self._collections = [839176, 839179, 839181, 1180811]
|
|
else:
|
|
self._collections = collections
|
|
self._special_posts = {"达达利亚": "21272578"}
|
|
|
|
async def _get_strategy_from_hyperion(self, collection_id: int, character_name: str) -> int:
|
|
if character_name in self._special_posts:
|
|
return self._special_posts[character_name]
|
|
post_id: int = -1
|
|
post_full_in_collection = await self._hyperion.get_post_full_in_collection(collection_id)
|
|
for post_data in post_full_in_collection["posts"]:
|
|
title = post_data["post"]["subject"]
|
|
topics = post_data["topics"]
|
|
for topic in topics:
|
|
if character_name == topic["name"]:
|
|
post_id = int(post_data["post"]["post_id"])
|
|
break
|
|
if post_id == -1 and title and character_name in title:
|
|
post_id = int(post_data["post"]["post_id"])
|
|
if post_id != -1:
|
|
break
|
|
return post_id
|
|
|
|
async def get_strategy(self, character_name: str) -> str:
|
|
cache = await self._cache.get_url_list(character_name)
|
|
if len(cache) >= 1:
|
|
return cache[0]
|
|
|
|
for collection_id in self._collections:
|
|
post_id = await self._get_strategy_from_hyperion(collection_id, character_name)
|
|
if post_id != -1:
|
|
break
|
|
else:
|
|
return ""
|
|
|
|
artwork_info = await self._hyperion.get_post_info(2, post_id)
|
|
await self._cache.set_url_list(character_name, artwork_info.image_urls)
|
|
return artwork_info.image_urls[0]
|
|
|
|
|
|
class GameMaterialService(BaseService):
|
|
def __init__(self, cache: GameCacheForMaterial, collections: Optional[List[int]] = None):
|
|
self._cache = cache
|
|
self._hyperion = Hyperion()
|
|
self._collections = [428421, 1164644] if collections is None else collections
|
|
self._special = ["雷电将军", "珊瑚宫心海", "菲谢尔", "托马", "八重神子", "九条裟罗", "辛焱", "神里绫华"]
|
|
|
|
async def _get_material_from_hyperion(self, collection_id: int, character_name: str) -> int:
|
|
post_id: int = -1
|
|
post_full_in_collection = await self._hyperion.get_post_full_in_collection(collection_id)
|
|
for post_data in post_full_in_collection["posts"]:
|
|
topics = post_data["topics"]
|
|
for topic in topics:
|
|
if character_name == topic["name"]:
|
|
post_id = int(post_data["post"]["post_id"])
|
|
break
|
|
if post_id != -1:
|
|
break
|
|
subject = post_data["post"]["subject"]
|
|
if character_name in subject:
|
|
post_id = int(post_data["post"]["post_id"])
|
|
if post_id != -1:
|
|
break
|
|
return post_id
|
|
|
|
async def get_material(self, character_name: str) -> str:
|
|
cache = await self._cache.get_url_list(character_name)
|
|
if len(cache) >= 1:
|
|
image_url_list = cache
|
|
else:
|
|
for collection_id in self._collections:
|
|
post_id = await self._get_material_from_hyperion(collection_id, character_name)
|
|
if post_id != -1:
|
|
break
|
|
else:
|
|
return ""
|
|
|
|
artwork_info = await self._hyperion.get_post_info(2, post_id)
|
|
image_url_list = artwork_info.image_urls
|
|
await self._cache.set_url_list(character_name, image_url_list)
|
|
if len(image_url_list) == 0:
|
|
return ""
|
|
if len(image_url_list) == 1:
|
|
return image_url_list[0]
|
|
if character_name in self._special:
|
|
return image_url_list[2]
|
|
return image_url_list[1]
|