1

backend/core/basic/character/calc_equip.py

@@ -1,4 +1,5 @@
 import sys
+from functools import cache
 from uuid import uuid1
 
 from core.basic.equipment import OathSuit, Suit, get_equipment
@@ -7,6 +8,16 @@ from ..roleinfo import CharacterEmblemInfo, CharacterEquipInfo, CharacterOathInf
 from .base import CharacterBase
 
 
+@cache
+def get_skill_classes(module_name: str) -> tuple[type, ...]:
+    module = sys.modules[module_name]
+    return tuple(
+        memberClass
+        for name in dir(module)
+        if isinstance((memberClass := getattr(module, name)), type) and memberClass.__name__.startswith('Skill')
+    )
+
+
 class CalcEquipMixin(CharacterBase):
     """装备计算、角色初始化相关方法"""
 
@@ -22,11 +33,12 @@ class CalcEquipMixin(CharacterBase):
             info['emblems'] = {}
         if info.get('oaths', None) is None:
             info['oaths'] = {}
+        equs = get_equipment(self.equVersion)
         for key in info['equips']:
-            self.charEquipInfo[key] = CharacterEquipInfo(info['equips'][key], self.equVersion, key)
+            self.charEquipInfo[key] = CharacterEquipInfo(info['equips'][key], self.equVersion, key, equs)
         for key in info['oaths']:
             # 导入誓约打造信息、誓约信息
-            self.charOathInfo.append(CharacterOathInfo(info['oaths'][key], self.equVersion, key))
+            self.charOathInfo.append(CharacterOathInfo(info['oaths'][key], self.equVersion, key, equs))
         try:
             # 异常处理 彩色等级自动取最小值
             minEmblemLv = min([info['emblems'][key]['lv'] for key in ['红色', '蓝色', '绿色', '黄色']], default=1)
@@ -36,7 +48,7 @@ class CalcEquipMixin(CharacterBase):
                 info['emblems'][key]['part'] = key
                 self.charEmblemInfo.append(CharacterEmblemInfo(info['emblems'][key]))
                 pass
-        except Exception as e:
+        except Exception:
             pass
         self.charOathSkillId = info.get('oathSkill', 1)
 
@@ -59,13 +71,12 @@ class CalcEquipMixin(CharacterBase):
 
     def load_skills(self) -> None:
         """加载技能"""
-        module = sys.modules[self.moduleName]
-        for name in dir(module):
-            memberClass = getattr(module, name)
-            if isinstance(memberClass, type) and memberClass.__name__.startswith('Skill'):
-                skill = memberClass(char=self)
-                self.skills_dict[skill.name] = skill
-                self.skills.append(skill)
+        for memberClass in get_skill_classes(self.moduleName):
+            skill = memberClass(char=self)
+            self.skills_dict[skill.name] = skill
+            self.skills_id_dict[str(skill.id)] = skill
+            self.skills_uuid_dict[str(skill.uuid)] = skill
+            self.skills.append(skill)
 
     # endregion
 
@@ -77,27 +88,27 @@ class CalcEquipMixin(CharacterBase):
         oathSuitInfo = {}
         correction = 0
         equs = get_equipment(self.equVersion)
-        for item in filter(lambda x: x.equInfo is not None, self.charEquipInfo.values()):
+        for item in self.charEquipInfo.values():
             partEqu = item.equInfo
+            if partEqu is None:
+                continue
             # partFusion = item.fusionInfo
-            if partEqu is not None:
-                for suit in partEqu.suit:
-                    if suitInfo.get(suit, None) is None:
-                        suitInfo[suit] = {'point': 0, 'count': 0}
-                    if partEqu.Point == 0:
-                        suitInfo[suit]['count'] += 1
-                    else:
-                        suitInfo[suit]['point'] += partEqu.Point
+            for suit in partEqu.suit:
+                suit_item = suitInfo.setdefault(suit, {'point': 0, 'count': 0})
+                if partEqu.Point == 0:
+                    suit_item['count'] += 1
+                else:
+                    suit_item['point'] += partEqu.Point
         for item in self.charOathInfo:
             partOath = item.oathInfo
-            if partOath is not None:
-                for suit in partOath.suit:
-                    if oathSuitInfo.get(suit, None) is None:
-                        oathSuitInfo[suit] = {'point': 0, 'count': 0}
-                    if partOath.Point == 0:
-                        oathSuitInfo[suit]['count'] += 1
-                    else:
-                        oathSuitInfo[suit]['point'] += partOath.Point
+            if partOath is None:
+                continue
+            for suit in partOath.suit:
+                oath_suit_item = oathSuitInfo.setdefault(suit, {'point': 0, 'count': 0})
+                if partOath.Point == 0:
+                    oath_suit_item['count'] += 1
+                else:
+                    oath_suit_item['point'] += partOath.Point
         max_point_suit = max((suit for suit in suitInfo if suitInfo[suit]['point'] > 0), key=lambda suit: suitInfo[suit]['point'], default=None)
         max_point_oath_suit = max((suit for suit in oathSuitInfo if oathSuitInfo[suit]['point'] > 0), key=lambda suit: oathSuitInfo[suit]['point'], default=None)
         zero_point_suits = [suit for suit in suitInfo if suitInfo[suit]['point'] == 0]
@@ -140,17 +151,19 @@ class CalcEquipMixin(CharacterBase):
                 oathSuitList += equs.get_oath_suit_info(suit, oathSuitInfo[suit]['point'], 0)
             if len(oathSuitList) == 0:
                 continue
+        oath_suit_by_suit_id = {item.suitId: item for item in oathSuitList}
         for suit in suitList:
             i = suit.id
             func = equs.funs.execture(f'suit_{i}')
-            suit = next((x for x in equs.suits if str(x.id) == str(i)), None)
+            suit = equs.suit_id_dict.get(str(i), None)
+            if suit is None:
+                continue
             # 誓约和套装进行匹配
             # 如果存在誓约和套装是同一个套装
             # 则直接不执行原套装效果，执行誓约套装增强的效果
-            oathEffect = next((x for x in oathSuitList if x.suitId == suit.suitId), None)
-            if suit is not None:
-                filtered_dict = {k: v for k, v in suit.__dict__.items() if k[0].isupper()}
-                self.SetStatus(**filtered_dict)
+            oathEffect = oath_suit_by_suit_id.get(suit.suitId, None)
+            filtered_dict = {k: v for k, v in suit.__dict__.items() if k[0].isupper()}
+            self.SetStatus(**filtered_dict)
             if func is not None and oathEffect is None:
                 # print(f'套装 {suit.name} 生效')
                 func(self)
@@ -160,15 +173,15 @@ class CalcEquipMixin(CharacterBase):
         self.max_point = 0 if res['suit'] is None else res.get('suit', {}).get('point', 0)
         try:
             self.oath_max_point = oathSuitInfo.get(max_point_oath_suit, {}).get('point', 0)
-        except:
+        except Exception:
             self.oath_max_point = 0
         self.oathSuitInfo = []
         self.oathSkillInfo = []
         # 当套装点数达到太初时，誓约技能生效
         if self.max_point >= 2550:
-            oathSuit = next((x for x in oathSuitList if x.suitId == max_point_oath_suit), None)
+            oathSuit = oath_suit_by_suit_id.get(max_point_oath_suit, None)
             if oathSuit is not None:
-                skills = list(filter(lambda x: x.suitId == oathSuit.suitId, oathSuit.skills))
+                skills = [x for x in oathSuit.skills if x.suitId == oathSuit.suitId]
                 skill = next((x for x in oathSuit.skills if str(x.skillId) == str(self.charOathSkillId)), None)
                 # 誓约套装效果生效
                 filtered_dict = {k: v for k, v in oathSuit.__dict__.items() if k[0].isupper()}
@@ -214,11 +227,11 @@ class CalcEquipMixin(CharacterBase):
         """计算装备基础效果、附魔"""
         effects = get_equipment(self.equVersion).funs
         # for detail in [(item.equInfo, item.fusionInfo, item.enchant, item.emblem_0, item.emblem_1) for item in filter(lambda x: x.equInfo is not None, self.charEquipInfo.values())]:
-        for detail in [(item.equInfo, item.enchant) for item in filter(lambda x: x.equInfo is not None, self.charEquipInfo.values())]:
-            equ = detail[0]
+        for item in self.charEquipInfo.values():
+            equ = item.equInfo
             if equ is None or equ.itemType == '副武器':
                 continue
-            enchat = detail[1]
+            enchat = item.enchant
             fun = effects.execture(f'equ_{equ.id}')
             if fun is None:
                 continue
@@ -316,20 +329,20 @@ class CalcEquipMixin(CharacterBase):
 
     def calc_basic(self):
         """计算基础属性:防具精通、增幅、强化等"""
+        char_equip_info = self.charEquipInfo
         for part in ['上衣', '头肩', '下装', '腰带', '鞋']:
-            cur = self.charEquipInfo[part]
+            cur = char_equip_info[part]
             if cur.equInfo is None:
                 continue
             equ = cur.equInfo
-            reinforce = self.charEquipInfo[part].reinforce
+            reinforce = cur.reinforce
             if equ is None:
                 continue
             for key in self.防具精通属性:
                 value = 精通计算(115, equ.rarity, reinforce, equ.itemDetailType, self.防具类型, self.buffer, key)
                 key = get_key_by_value(key)
                 setattr(self, key, getattr(self, key) + value)
-        for part in self.charEquipInfo.keys():
-            cur = self.charEquipInfo[part]
+        for part, cur in char_equip_info.items():
             if cur.equInfo is None:
                 continue
             if cur.reinforceType == 1: