feat: add Powerup, UsedPowerup model and routes

Author: tnmkr

src/pwncore/models/__init__.py

@@ -29,6 +29,12 @@
     AttackDefTeam
 )
 
+from pwncore.models.powerups import (
+    Powerup,
+    PowerupType,
+    UsedPowerup,
+    Powerup_Pydantic,
+)
 from pwncore.models.pre_event import (
     PreEventProblem,
     PreEventSolvedProblem,
@@ -58,6 +64,9 @@
     "BaseProblem",
     "AttackDefProblem",
     "AttackDefTeam",
+    "Powerup",
+    "UsedPowerup",
+    "Powerup_Pydantic",
 )
 
 

src/pwncore/models/round2.py

@@ -1,10 +1,13 @@
 from __future__ import annotations
 from tortoise import fields
 from tortoise.models import Model
+from tortoise.expressions import F
+
 __all__ = (
     "AttackDefProblem",
     "AttackDefTeam",
     "Problem",
+    "Powerup",
 )
 #TODO: Actually implement this.
 # For now this is a dummy class for testing AttackDefTeam.
@@ -21,3 +24,5 @@ class AttackDefTeam(Model):
         "models.Team", null=False
     )
     assigned_attack_def_problem: fields.ReverseRelation(AttackDefProblem)
+    async def remaining_powerups(self):
+        return self.powerups.filter(used_count__lt=F('max_uses'))

src/pwncore/routes/__init__.py

@@ -1,6 +1,6 @@
 from fastapi import APIRouter
 
-from pwncore.routes import ctf, team, auth, admin, leaderboard
+from pwncore.routes import ctf, team, auth, admin, leaderboard, powerups
 
 # from pwncore.config import config
 
@@ -13,5 +13,6 @@
 router.include_router(team.router)
 router.include_router(leaderboard.router)
 router.include_router(admin.router)
+router.include_router(powerups.router)
 # if config.development:
 # router.include_router(admin.router)

src/pwncore/routes/admin.py

@@ -16,9 +16,15 @@
     User,
     AttackDefTeam,
     AttackDefProblem,
+    Powerup,
+    UsedPowerup
 )
 from pwncore.models.ctf import SolvedProblem
 from pwncore.models.pre_event import PreEventUser
+from pwncore.models.powerups import (
+    Sabotage,
+    Shield
+)
 
 metadata = {
     "name": "admin",
@@ -173,6 +179,18 @@ async def init_db(
         problem_id=(await Problem.get(name="GitGood2")).id,
         attack_def_team_id=(await AttackDefTeam.get(team_id=triple_b_battery.id)).id
     )
+    await Powerup.create(
+        attack_def_team = (await AttackDefTeam.get(team__id=triple_b_battery.id)),
+        powerup_type = Sabotage
+    )
+    shield = await Powerup.create(
+        attack_def_team = (await AttackDefTeam.get(team__id=triple_b_battery.id)),
+        powerup_type = Shield
+    )
+    await UsedPowerup.create(
+        powerup = shield
+    )
+
     await PreEventUser.create(tag="23BCE1000", email="[email protected]")
     await PreEventUser.create(tag="23BRS1000", email="[email protected]")
     await PreEventSolvedProblem.create(user_id="23BCE1000", problem_id="1")