mirror of
https://github.com/RYGhub/royalnet.git
synced 2024-11-30 15:04:18 +00:00
40 lines
1.4 KiB
Python
40 lines
1.4 KiB
Python
|
from typing import *
|
||
|
|
import royalnet.commands as rc
|
||
|
|
import royalnet.utils as ru
|
||
|
|
|
||
|
|
from ..tables import DndBattleUnit
|
||
|
|
from .getactivebattle import get_active_battle
|
||
|
|
from .getactivechar import get_active_character
|
||
|
|
|
||
|
|
from sqlalchemy import func, and_
|
||
|
|
|
||
|
|
|
||
|
|
async def find_unit_in_current_battle(data: rc.CommandData, name: Optional[str]) -> Optional[DndBattleUnit]:
|
||
|
|
DndBattleUnitT = data._interface.alchemy.get(DndBattleUnit)
|
||
|
|
|
||
|
|
active_battle = await get_active_battle(data)
|
||
|
|
if active_battle is None:
|
||
|
|
raise rc.CommandError("No battle is active in this chat.")
|
||
|
|
|
||
|
|
if name is None:
|
||
|
|
active_character = await get_active_character(data)
|
||
|
|
if active_character is None:
|
||
|
|
raise rc.InvalidInputError("You currently have no active character.")
|
||
|
|
|
||
|
|
unit = await ru.asyncify(data.session.query(DndBattleUnitT).filter_by(
|
||
|
|
linked_character=active_character.character,
|
||
|
|
battle=active_battle.battle
|
||
|
|
).one_or_none)
|
||
|
|
if unit is None:
|
||
|
|
raise rc.InvalidInputError("Your active character is not fighting in this battle.")
|
||
|
|
|
||
|
|
else:
|
||
|
|
unit = await ru.asyncify(data.session.query(DndBattleUnitT).filter(and_(
|
||
|
|
func.lower(DndBattleUnitT.name) == func.lower(name),
|
||
|
|
DndBattleUnitT.battle == active_battle.battle
|
||
|
|
)).one_or_none)
|
||
|
|
if unit is None:
|
||
|
|
return None
|
||
|
|
|
||
|
|
return unit
|