70 lines
2.0 KiB
Python
70 lines
2.0 KiB
Python
from enum import Enum
|
||
from collections import defaultdict
|
||
|
||
|
||
class EssenceType(Enum):
|
||
"""
|
||
灵气类型
|
||
"""
|
||
GOLD = "gold" # 金
|
||
WOOD = "wood" # 木
|
||
WATER = "water" # 水
|
||
FIRE = "fire" # 火
|
||
EARTH = "earth" # 土
|
||
|
||
def __str__(self) -> str:
|
||
"""返回灵气类型的中文名称"""
|
||
return essence_names.get(self, self.value)
|
||
|
||
@classmethod
|
||
def from_str(cls, essence_str: str) -> 'EssenceType':
|
||
"""
|
||
从字符串创建EssenceType实例
|
||
|
||
Args:
|
||
essence_str: 灵气的字符串表示，如 "金", "木", "水", "火", "土"
|
||
|
||
Returns:
|
||
对应的EssenceType枚举值
|
||
|
||
Raises:
|
||
ValueError: 如果字符串不匹配任何已知的灵气类型
|
||
"""
|
||
# 首先尝试匹配中文名称
|
||
for essence_type, chinese_name in essence_names.items():
|
||
if chinese_name == essence_str:
|
||
return essence_type
|
||
|
||
# 然后尝试匹配英文值
|
||
for essence_type in cls:
|
||
if essence_type.value == essence_str:
|
||
return essence_type
|
||
|
||
raise ValueError(f"Unknown essence type: {essence_str}")
|
||
|
||
essence_names = {
|
||
EssenceType.GOLD: "金",
|
||
EssenceType.WOOD: "木",
|
||
EssenceType.WATER: "水",
|
||
EssenceType.FIRE: "火",
|
||
EssenceType.EARTH: "土"
|
||
}
|
||
|
||
class Essence():
|
||
"""
|
||
灵气，用来描述某个region的灵气情况。
|
||
灵气分为五种：金木水火土（先这些，之后加新的）
|
||
每个region有五种灵气，每种灵气有不同的浓度。
|
||
浓度从0~10。
|
||
"""
|
||
def __init__(self, density: dict[EssenceType, int]):
|
||
self.density = defaultdict(int)
|
||
for essence_type, density in density.items():
|
||
self.density[essence_type] = density
|
||
|
||
def get_density(self, essence_type: EssenceType) -> int:
|
||
return self.density[essence_type]
|
||
|
||
def set_density(self, essence_type: EssenceType, density: int):
|
||
self.density[essence_type] = density
|