Files
KianaBH3/game_server/game/chat/command/max_skill.py
2024-11-18 16:22:42 +08:00

49 lines
1.9 KiB
Python

from database import mongo
from lib.proto import GetAvatarDataReq, AvatarSubSkill
from game_server.net.session import Session
from game_server.resource import ResourceManager
from game_server.game.enum.data_type import DataType
from game_server.game.chat.decorators import Command
from game_server.resource.configdb.avatar_sub_skill_data import AvatarSubSkillData
@Command(
prefix="maxskill",
usage="/maxskill {avatar_id | all} - Maximize the skills of a specific avatar. Use 'all' to maximize the skills of all avatars.",
)
async def execute(session: Session, avatar_id):
desc = getattr(execute, "usage", "No description available.")
if avatar_id == "all":
for avatar in session.player.avatars.values():
await max_out_skills_for_avatar(avatar, session)
return "All avatars' skills have been maxed out!"
if not avatar_id.isdigit():
return f"Usage: {desc}"
avatar_id = int(avatar_id)
avatar = session.player.avatars.get(avatar_id)
if not avatar:
return f"Avatar with ID {avatar_id} does not exist."
await max_out_skills_for_avatar(avatar, session)
return f"All skills for avatar {avatar_id} have been maxed out!"
async def max_out_skills_for_avatar(avatar, session:Session):
resource_manager = ResourceManager.instance()
for skill_id, skill in avatar.skill_lists.items():
sub_skills = [
data for data in resource_manager.values(AvatarSubSkillData)
if data.skillId == skill_id
]
for sub_skill_data in sub_skills:
skill.sub_skill_lists[sub_skill_data.avatarSubSkillId] = AvatarSubSkill(
sub_skill_id=sub_skill_data.avatarSubSkillId,
level=sub_skill_data.maxLv
)
mongo.save(session, DataType.AVATAR, [avatar.avatar_id])
await session.process_packet(session.create_packet(GetAvatarDataReq(avatar_id_list=[avatar.avatar_id])))