import abc from typing import * import royalnet import royalnet.commands as rc import royalnet.utils as ru from ...tables import DndBattleUnit from ...utils import get_targets class DndBattleTargetCommand(rc.Command, abc.ABC): @abc.abstractmethod async def _change(self, unit: DndBattleUnit, args: List[str]): ... async def run(self, args: rc.CommandArgs, data: rc.CommandData) -> None: target = args[0] units = await get_targets(data, target) if len(units) == 0: raise rc.InvalidInputError(f"No targets found matching [c]{target}[/c].") for unit in units: await self._change(unit, args[1:]) await data.session_commit() message = [] for unit in units: message.append(f"{unit}") await data.reply("\n\n".join(message))