royalnet/rpgpack/commands/dndedit.py

import re
from royalnet.commands import *
import royalnet.utils as ru
from .dndnew import DndnewCommand
from ..tables import DndCharacter
from ..utils import get_active_character


class DndeditCommand(DndnewCommand):
    name: str = "dndedit"

    description: str = "Edit the active DnD character."

    aliases = ["de", "dnde", "edit", "dedit"]

    async def run(self, args: CommandArgs, data: CommandData) -> None:
        character_sheet = args.joined()

        async with data.session_acm() as session:
            active_character = await get_active_character(session=session, data=data)

            if active_character is None:
                raise CommandError("You don't have an active character.")
            char: DndCharacter = active_character.character

            if character_sheet == "":
                await data.reply(char.to_edit_string())
                return

            arguments = self._parse(character_sheet)
            for key in arguments:
                char.__setattr__(key, arguments[key])

            await ru.asyncify(session.commit)

            await data.reply(f"✅ Edit successful!")