mirror of
https://github.com/RYGhub/royalnet.git
synced 2024-11-27 13:34:28 +00:00
69 lines
2.7 KiB
Python
69 lines
2.7 KiB
Python
import re
|
||
# noinspection PyUnresolvedReferences
|
||
from royalnet.commands import *
|
||
from ..tables import DndCharacter
|
||
from ..utils import DndProficiencyType
|
||
|
||
|
||
class DndnewCommand(Command):
|
||
name: str = "dndnew"
|
||
|
||
description: str = "Create a new DnD character."
|
||
|
||
aliases = ["dn", "dndn", "new", "dnew"]
|
||
|
||
syntax = "{name}\n{character_sheet}"
|
||
|
||
def _search_value(self, name: str, string: str):
|
||
return re.search(r"\s*" + name + r"\s*([0-9.]+)\s*", string, re.IGNORECASE)
|
||
|
||
def _parse(self, character_sheet: str) -> dict:
|
||
columns = list(self.alchemy.get(DndCharacter).__table__.columns)
|
||
column_names = [column.name for column in columns if (not column.primary_key and
|
||
not column.foreign_keys and
|
||
column.name != "name")]
|
||
arguments = {}
|
||
for column_name in column_names:
|
||
match = self._search_value(column_name, character_sheet)
|
||
if match:
|
||
if column_name.endswith("_proficiency"):
|
||
arguments[column_name] = DndProficiencyType(float(match.group(1)))
|
||
else:
|
||
arguments[column_name] = match.group(1)
|
||
return arguments
|
||
|
||
def _syntax(self) -> str:
|
||
columns = list(self.alchemy.get(DndCharacter).__table__.columns)
|
||
column_names = [column.name for column in columns if (not column.primary_key and
|
||
not column.foreign_keys and
|
||
column.name != "name")]
|
||
message = "ℹ️ How to create a new character:\n[p]/dndnew YOUR_CHARACTER_NAME\n"
|
||
for column_name in column_names:
|
||
message += f"{column_name} _\n"
|
||
message += "[/p]"
|
||
return message
|
||
|
||
async def run(self, args: CommandArgs, data: CommandData) -> None:
|
||
character_sheet = args.joined()
|
||
|
||
if character_sheet == "":
|
||
await data.reply(self._syntax())
|
||
return
|
||
|
||
creator = await data.get_author()
|
||
|
||
name, rest = character_sheet.split("\n", 1)
|
||
|
||
character = self.alchemy.get(DndCharacter)(name=name, creator=creator, **self._parse(rest))
|
||
data.session.add(character)
|
||
|
||
try:
|
||
await data.session_commit()
|
||
except Exception as err:
|
||
# THIS IS INTENDED
|
||
if err.__class__.__name__ == "IntegrityError":
|
||
param_name = re.search(r'in column "(\S+)"', err.args[0]).group(1)
|
||
raise CommandError(f"Mandatory parameter '{param_name}' is missing.")
|
||
raise
|
||
|
||
await data.reply(f"✅ Character [b]{character.name}[/b] created!")
|