refactor name logic

src/classes/name.py

@@ -0,0 +1,197 @@
+import random
+from typing import Optional
+from dataclasses import dataclass
+
+from src.utils.df import game_configs
+from src.classes.avatar import Gender
+
+
+@dataclass
+class LastName:
+    """姓氏"""
+    name: str
+    sect: Optional[str]
+    
+@dataclass
+class GivenName:
+    """名字"""
+    name: str
+    gender: Gender
+    sect: Optional[str]
+
+
+class NameManager:
+    """姓名管理器"""
+    
+    def __init__(self):
+        # 散修通用姓氏
+        self.common_last_names: list[str] = []
+        # 按宗门分类的姓氏 {宗门名: [姓氏列表]}
+        self.sect_last_names: dict[str, list[str]] = {}
+        
+        # 散修通用名字 {Gender: [名字列表]}
+        self.common_given_names: dict[Gender, list[str]] = {
+            Gender.MALE: [],
+            Gender.FEMALE: []
+        }
+        # 按宗门和性别分类的名字 {宗门名: {Gender: [名字列表]}}
+        self.sect_given_names: dict[str, dict[Gender, list[str]]] = {}
+        
+        self._load_names()
+    
+    def _load_names(self):
+        """从CSV加载姓名数据"""
+        # 加载姓氏
+        last_name_df = game_configs["last_name"]
+        for _, row in last_name_df.iterrows():
+            name = str(row["last_name"]).strip()
+            sect = str(row.get("sect", "")).strip() if row.get("sect") and str(row.get("sect")) != "nan" else None
+            
+            if sect:
+                if sect not in self.sect_last_names:
+                    self.sect_last_names[sect] = []
+                self.sect_last_names[sect].append(name)
+            else:
+                self.common_last_names.append(name)
+        
+        # 加载名字
+        given_name_df = game_configs["given_name"]
+        for _, row in given_name_df.iterrows():
+            name = str(row["given_name"]).strip()
+            gender_str = str(row["gender"]).strip()
+            gender = Gender.MALE if gender_str == "男" else Gender.FEMALE
+            sect = str(row.get("sect", "")).strip() if row.get("sect") and str(row.get("sect")) != "nan" else None
+            
+            if sect:
+                if sect not in self.sect_given_names:
+                    self.sect_given_names[sect] = {Gender.MALE: [], Gender.FEMALE: []}
+                self.sect_given_names[sect][gender].append(name)
+            else:
+                self.common_given_names[gender].append(name)
+    
+    def get_random_last_name(self, sect_name: Optional[str] = None) -> str:
+        """
+        获取随机姓氏
+        
+        Args:
+            sect_name: 宗门名称，如果为None则从散修姓氏中选择
+            
+        Returns:
+            姓氏字符串
+        """
+        if sect_name and sect_name in self.sect_last_names:
+            return random.choice(self.sect_last_names[sect_name])
+        return random.choice(self.common_last_names)
+    
+    def get_random_given_name(self, gender: Gender, sect_name: Optional[str] = None) -> str:
+        """
+        获取随机名字
+        
+        Args:
+            gender: 性别
+            sect_name: 宗门名称，如果为None则从散修名字中选择
+            
+        Returns:
+            名字字符串
+        """
+        if sect_name and sect_name in self.sect_given_names:
+            sect_names = self.sect_given_names[sect_name][gender]
+            if sect_names:
+                return random.choice(sect_names)
+        return random.choice(self.common_given_names[gender])
+    
+    def get_random_full_name(self, gender: Gender, sect_name: Optional[str] = None) -> str:
+        """
+        获取随机全名
+        
+        Args:
+            gender: 性别
+            sect_name: 宗门名称，如果为None则为散修
+            
+        Returns:
+            完整姓名
+        """
+        last_name = self.get_random_last_name(sect_name)
+        given_name = self.get_random_given_name(gender, sect_name)
+        return last_name + given_name
+    
+    def get_random_full_name_with_surname(
+        self, 
+        gender: Gender, 
+        surname: str, 
+        sect_name: Optional[str] = None
+    ) -> str:
+        """
+        使用指定姓氏生成随机全名
+        
+        Args:
+            gender: 性别
+            surname: 指定的姓氏
+            sect_name: 宗门名称，如果为None则为散修
+            
+        Returns:
+            完整姓名
+        """
+        if not surname:
+            return self.get_random_full_name(gender, sect_name)
+        given_name = self.get_random_given_name(gender, sect_name)
+        return surname + given_name
+
+
+# 全局单例
+_name_manager = NameManager()
+
+
+def get_random_name(gender: Gender, sect_name: Optional[str] = None) -> str:
+    """获取随机全名"""
+    return _name_manager.get_random_full_name(gender, sect_name)
+
+
+def get_random_name_for_sect(gender: Gender, sect) -> str:
+    """
+    基于宗门生成姓名（兼容旧接口）
+    
+    Args:
+        gender: 性别
+        sect: Sect对象或None
+        
+    Returns:
+        完整姓名
+    """
+    sect_name = sect.name if sect is not None else None
+    return _name_manager.get_random_full_name(gender, sect_name)
+
+
+def pick_surname_for_sect(sect) -> str:
+    """
+    从宗门常见姓或全局库中挑选一个姓氏（兼容旧接口）
+    
+    Args:
+        sect: Sect对象或None
+        
+    Returns:
+        姓氏
+    """
+    sect_name = sect.name if sect is not None else None
+    return _name_manager.get_random_last_name(sect_name)
+
+
+def get_random_name_with_surname(
+    gender: Gender, 
+    surname: str, 
+    sect
+) -> str:
+    """
+    使用指定姓氏生成随机全名（兼容旧接口）
+    
+    Args:
+        gender: 性别
+        surname: 指定的姓氏
+        sect: Sect对象或None
+        
+    Returns:
+        完整姓名
+    """
+    sect_name = sect.name if sect is not None else None
+    return _name_manager.get_random_full_name_with_surname(gender, surname, sect_name)
+