models: add trade money columns

Ballsdex-Team · Feb 19, 2025 · 7b4dcb1 · 7b4dcb1
1 parent f401cbc
commit 7b4dcb1
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 1 deletion.
diff --git a/...bd_models/migrations/0006_player_money.py → ...rade_player1_money_trade_player2_money.py b/...bd_models/migrations/0006_player_money.py → ...rade_player1_money_trade_player2_money.py
@@ -1,4 +1,4 @@
-# Generated by Django 5.1.4 on 2025-02-19 13:46
+# Generated by Django 5.1.4 on 2025-02-19 15:06
 
 from django.db import migrations, models
 
@@ -17,4 +17,14 @@ class Migration(migrations.Migration):
                 default=0, help_text="Money posessed by the player"
             ),
         ),
+        migrations.AddField(
+            model_name="trade",
+            name="player1_money",
+            field=models.PositiveBigIntegerField(default=0),
+        ),
+        migrations.AddField(
+            model_name="trade",
+            name="player2_money",
+            field=models.PositiveBigIntegerField(default=0),
+        ),
     ]
diff --git a/admin_panel/bd_models/models.py b/admin_panel/bd_models/models.py
@@ -357,8 +357,10 @@ class Trade(models.Model):
     date = models.DateTimeField(auto_now_add=True, editable=False)
     player1 = models.ForeignKey(Player, on_delete=models.CASCADE)
     player1_id: int
+    player1_money = models.PositiveBigIntegerField(default=0)
     player2 = models.ForeignKey(Player, on_delete=models.CASCADE, related_name="trade_player2_set")
     player2_id: int
+    player2_money = models.PositiveBigIntegerField(default=0)
     tradeobject_set: models.QuerySet[TradeObject]
 
     def __str__(self) -> str:

diff --git a/ballsdex/core/models.py b/ballsdex/core/models.py
@@ -512,9 +512,17 @@ class Trade(models.Model):
     player1: fields.ForeignKeyRelation[Player] = fields.ForeignKeyField(
         "models.Player", related_name="trades"
     )
+    player1_money = fields.BigIntField(
+        default=0,
+        validators=[validators.MaxValueValidator((1 << 63) - 1), validators.MinValueValidator(0)],
+    )
     player2: fields.ForeignKeyRelation[Player] = fields.ForeignKeyField(
         "models.Player", related_name="trades2"
     )
+    player2_money = fields.BigIntField(
+        default=0,
+        validators=[validators.MaxValueValidator((1 << 63) - 1), validators.MinValueValidator(0)],
+    )
     date = fields.DatetimeField(auto_now_add=True)
     tradeobjects: fields.ReverseRelation[TradeObject]