mirror of
https://github.com/RYGhub/royalnet.git
synced 2024-11-23 19:44:20 +00:00
36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
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!")
|