from typing import List, Optional from modules.apihelper.hyperion import Hyperion from .cache import GameCache class GameStrategyService: def __init__(self, cache: GameCache, 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 async def _get_strategy_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) if post_full_in_collection.error: return post_id for post_data in post_full_in_collection.data["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 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[-1] 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_artwork_info(2, post_id) await self._cache.set_url_list(character_name, artwork_info.results.image_url_list) return artwork_info.results.image_url_list[0] class GameMaterialService: def __init__(self, cache: GameCache, 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) if post_full_in_collection.error: return post_id for post_data in post_full_in_collection.data["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: return cache[-1] 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_artwork_info(2, post_id) await self._cache.set_url_list(character_name, artwork_info.results.image_url_list) image_url_list = artwork_info.results.image_url_list if len(image_url_list) == 0: return "" elif len(image_url_list) == 1: return image_url_list[0] elif character_name in self._special: return image_url_list[2] else: return image_url_list[1]