import importlib from database.models import EmblemData, Session, EquData, StoneData, SuitData, EnchantData,JadeData from database.connect import get_db_engine as get_engine def parse_to_number_list(info: str, default: list[float] = [0]) -> list[float]: return default if not info else [float(i) for i in info.split(',')] class Equ: id: str name: str rarity: str itemType: str itemDetailType: str suit: list[str] Point: list[float] | float categorize: str imageUrl: str detail: str bufferDetail: str STR: list[float] | float INT: list[float] | float Vitality: list[float] | float Spirit: list[float] | float AtkP: list[float] | float AtkM: list[float] | float AtkI: list[float] | float SkillAttack: list[float] | float Attack: list[float] | float Buffer: list[float] | float max_adaptation: int def __init__(self, **kwargs): self.__dict__.update(kwargs) def adapt(self, adaptation: int = 0): """调适 Args: adaptation (int): 调适等级 """ temp = self.__dict__.copy() keys = [key for key in EquData.__dict__.keys() if key[0].isupper()] for attr in keys: value = getattr(self, attr) temp[attr] = value[min(adaptation, len(value) - 1)] if isinstance(value, list) else value return Equ(**temp) class Suit: id: int suitId: int suitName: str rarity: str name: str point: int level: int count: int imageUrl: str SkillAttack: float Attack: float def __init__(self, **kwargs): self.__dict__.update(kwargs) class Equipments: def __init__(self, version='0'): self.version = version self.engine = get_engine(version) self.equs: list[Equ] = [] self.stones: list[Equ] = [] self.funs = self.init_func() self.equ_dict: dict[str, Equ] = {} self.stone_dict: dict[str, Equ] = {} self.jades = [] self.enchants = [] self.emblems = [] self.suits: list[Suit] = [] self.suit_dict: dict[str, Suit] = {} self.init_equs() self.init_suits() self.init_enchants() self.init_stones() self.init_emblem() self.init_jades() self.engine.dispose() def init_equs(self): """从数据库中获取所有装备信息""" with Session(self.engine) as session: db_list = session.query(EquData).all() keys = [key for key in EquData.__dict__.keys() if key[0].isupper()] + ['suit'] for item in db_list: max_adaptation = 0 item.id = str(item.id) for attr in keys: if attr == 'suit': value = parse_to_number_list(getattr(item, attr), []) value = [str(int(i)) for i in value] else: value = parse_to_number_list(getattr(item, attr)) max_adaptation = max(max_adaptation, len(value) - 1) setattr(item, attr, value) equ_dict = {k: v for k, v in item.__dict__.items() if not k.startswith('_')} equ_dict['max_adaptation'] = max_adaptation equ = Equ(**equ_dict) self.equs.append(equ) self.equ_dict[equ.id] = equ def init_stones(self): """从数据库中获取所有装备信息""" with Session(self.engine) as session: db_list = session.query(StoneData).all() keys = [key for key in StoneData.__dict__.keys() if key[0].isupper()] + ['suit'] for item in db_list: max_adaptation = 0 item.id = str(item.id) for attr in keys: if attr == 'suit': value = parse_to_number_list(getattr(item, attr), []) value = [str(int(i)) for i in value] else: value = parse_to_number_list(getattr(item, attr)) max_adaptation = max(max_adaptation, len(value) - 1) setattr(item, attr, value) stone_dict = {k: v for k, v in item.__dict__.items() if not k.startswith('_')} stone_dict['max_adaptation'] = max_adaptation stone = Equ(**stone_dict) self.stones.append(stone) self.stone_dict[stone.id] = stone def init_suits(self): """从数据库中获取所有套装信息""" with Session(self.engine) as session: db_list = session.query(SuitData).all() for item in db_list: suit = Suit(**{k: v for k, v in item.__dict__.items() if not k.startswith('_')}) self.suits.append(suit) self.suit_dict.setdefault(str(suit.suitId), []).append(suit) pass def init_enchants(self): """从数据库中获取所有附魔信息""" with Session(self.engine) as session: db_list = session.query(EnchantData).all() for item in db_list: enchant = {k: v for k, v in item.__dict__.items() if not k.startswith('_')} enchant['position'] = [] if enchant['itemType'] is None else enchant['itemType'].split(',') enchant['categorize'] = [] if enchant['categorize'] is None else enchant['categorize'].split(',') del enchant['itemType'] self.enchants.append(enchant) pass pass def init_emblem(self): """从数据库中获取所有徽章信息""" with Session(self.engine) as session: db_list = session.query(EmblemData).all() for item in db_list: emblem = {k: v for k, v in item.__dict__.items() if not k.startswith('_')} emblem['position'] = [] if emblem['itemType'] is None else emblem['itemType'].split(',') emblem['categorize'] = [] if emblem['categorize'] is None else emblem['categorize'].split(',') del emblem['itemType'] self.emblems.append(emblem) pass pass def init_jades(self): """从数据库中获取所有辟邪玉信息""" with Session(self.engine) as session: db_list = session.query(JadeData).all() for item in db_list: jade = {k: v for k, v in item.__dict__.items() if not k.startswith('_')} self.jades.append(jade) pass pass def init_func(self): """根据装备版本加载装备效果""" try: funs = importlib.import_module(f"core.equipment.version_{self.version}") except Exception: funs = importlib.import_module("core.equipment.version_0") return funs def get_suit_info(self, suitId: str | int, point: int = 0, count: int = 0) -> list[Suit]: """根据套装点数返回对应适用的套装属性\n 新套装只需要传入suitID和point即可，count默认为0\n 老套装需要传入suitID和count即可，point默认为0 """ if suitId not in self.suit_dict: return [] # 先筛选出点数小于等于point和数量小于等于count的套装 suits: list[Suit] = [suit for suit in self.suit_dict[str(suitId)] if suit.point <= point and suit.count <= count] result: dict[int, Suit] = {} # 取每种count套装中点数最高的 for suit in suits: if suit.count not in result or suit.point > result[suit.count].point: result[suit.count] = suit return list(result.values()) class EquEffect: name: str = '' """装备技能名称""" icon: str = '' """技能图标""" cd: float = 0 """技能冷却时间""" data:float = 0 """伤害量""" def __init__(self, name: str, icon: str, cd: float, data: float): self.name = name self.icon = icon self.cd = cd self.data = data def get_equipment(version: str = '0') -> Equipments: # print('加载装备信息') return Equipments(version)