mirror of
https://github.com/RYGhub/royalnet.git
synced 2024-11-27 21:44:21 +00:00
35 lines
956 B
Python
35 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}")
|