mirror of
https://github.com/RYGhub/royalnet.git
synced 2024-12-17 23:24:20 +00:00
34 lines
956 B
Python
34 lines
956 B
Python
from typing import *
|
|
import royalnet
|
|
import royalnet.commands as rc
|
|
import royalnet.utils as ru
|
|
from ..tables import DndBattleUnit
|
|
from ..utils import find_unit_in_current_battle
|
|
|
|
|
|
class DndstatusCommand(rc.Command):
|
|
name: str = "dndstatus"
|
|
|
|
description: str = "Change the status for a unit in the current battle."
|
|
|
|
syntax: str = "[name] {status}"
|
|
|
|
aliases = ["status", "dstatus"]
|
|
|
|
async def run(self, args: rc.CommandArgs, data: rc.CommandData) -> None:
|
|
name = args.optional(0)
|
|
status = " ".join(args[1:])
|
|
|
|
if name is not None:
|
|
unit: Optional[DndBattleUnit] = await find_unit_in_current_battle(data, name)
|
|
else:
|
|
unit = None
|
|
|
|
if unit is None:
|
|
status = " ".join(args)
|
|
unit: Optional[DndBattleUnit] = await find_unit_in_current_battle(data, None)
|
|
|
|
|
|
unit.status = status
|
|
await data.session_commit()
|
|
await data.reply(f"{unit}")
|