Some progress on the initiative tracker

rpgpack/commands/__init__.py

@@ -12,6 +12,7 @@ from .testhealth import TesthealthCommand
 from .testfaction import TestfactionCommand
 from .dndnewbattle import DndnewbattleCommand
 from .dndactivebattle import DndactivebattleCommand
+from .dndaddunit import DndaddunitCommand
 
 # Enter the commands of your Pack here!
 available_commands = [
@@ -28,6 +29,7 @@ available_commands = [
     TestfactionCommand,
     DndnewbattleCommand,
     DndactivebattleCommand,
+    DndaddunitCommand
 ]
 
 # Don't change this, it should automatically generate __all__

rpgpack/commands/dndaddunit.py

@@ -0,0 +1,45 @@
+from typing import *
+import royalnet
+import royalnet.commands as rc
+import royalnet.utils as ru
+from ..types import Faction, Health
+from ..tables import DndBattleUnit
+from ..utils import get_active_battle
+
+
+class DndaddunitCommand(rc.Command):
+    name: str = "dndaddunit"
+
+    description: str = "Add an Unit to a Battle."
+
+    aliases = ["dau", "dndau", "daddunit", ""]
+
+    syntax: str = "{faction} {name} {initiative} {health} {armorclass}"
+
+    async def run(self, args: rc.CommandArgs, data: rc.CommandData) -> None:
+        faction = Faction[args[0].upper()]
+        name = args[1]
+        initiative = int(args[2])
+        health = args[3]
+        armor_class = int(args[4])
+
+        DndBattleUnitT = self.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.")
+
+        dbu = DndBattleUnitT(
+            initiative=initiative,
+            faction=faction,
+            name=name,
+            health_string=health,
+            armor_class=armor_class,
+            battle=active_battle.battle
+        )
+
+        data.session.add(dbu)
+        await data.session_commit()
+
+        await data.reply(f"✅ [b]{dbu.name}[/b] joined the battle!\n"
+                         f"{dbu}")

rpgpack/tables/dndactivebattle.py

@@ -6,9 +6,13 @@ from sqlalchemy.ext.declarative import *
 class DndActiveBattle:
     __tablename__ = "dndactivebattle"
 
+    @declared_attr
+    def active_battle_id(self):
+        return Column(Integer, primary_key=True)
+
     @declared_attr
     def battle_id(self):
-        return Column(Integer, ForeignKey("dndbattle.id"), primary_key=True)
+        return Column(Integer, ForeignKey("dndbattle.id"))
 
     @declared_attr
     def battle(self):

rpgpack/tables/dndactivecharacters.py

@@ -6,13 +6,17 @@ from sqlalchemy.ext.declarative import *
 class DndActiveCharacter:
     __tablename__ = "dndactivecharacters"
 
+    @declared_attr
+    def active_character_id(self):
+        return Column(Integer, primary_key=True)
+
     @declared_attr
     def character_id(self):
-        return Column(Integer, ForeignKey("dndcharacters.character_id"), primary_key=True)
+        return Column(Integer, ForeignKey("dndcharacters.character_id"))
 
     @declared_attr
     def user_id(self):
-        return Column(Integer, ForeignKey("users.uid"), primary_key=True)
+        return Column(Integer, ForeignKey("users.uid"))
 
     @declared_attr
     def character(self):

rpgpack/tables/dndbattle.py

@@ -28,4 +28,7 @@ class DndBattle:
         string = []
         string.append(f"⚔️ [b]{self.name}[/b]\n")
         string.append(f"{self.description}\n")
+        string.append("\n")
+        for unit in sorted(self.units, key=lambda u: -u.initiative):
+            string.append(f"{unit}\n")
         return "".join(string)

rpgpack/tables/dndbattleunit.py

@@ -54,3 +54,23 @@ class DndBattleUnit:
     @declared_attr
     def status(self):
         return Column(String)
+
+    def __str__(self):
+        string = [
+            f"{self.faction.value}",
+            f"[b]{self.name}[/b]",
+            f"[{self.initiative}]",
+        ]
+
+        if self.health:
+            string.append(f"{self.health}")
+        if self.armor_class:
+            string.append(f"🛡 {self.armor_class}")
+        if self.extra:
+            string.append(f"💠 {self.extra}")
+        if self.status:
+            string.append(f"💫 {self.status}")
+        return " ".join(string)
+
+    def __repr__(self):
+        return f"<DndBattleUnit {self.id} ({self.name})>"