Merge pull request #174 from diogomatoschaves/fix-database-migrations

Delete referencing null objects
diogomatoschaves · Mar 3, 2024 · 15becea · 15becea
2 parents 5cddd86 + a03c1d8
commit 15becea
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/database/model/migrations/0093_auto_20240227_1753.py b/database/model/migrations/0093_auto_20240227_1753.py
@@ -4,13 +4,20 @@
 import django.db.models.deletion
 
 
+def delete_referecing_null(apps, _):
+    Position = apps.get_model('model', 'Position')
+
+    Position.objects.filter(pipeline__id=None).delete()
+
+
 class Migration(migrations.Migration):
 
     dependencies = [
         ('model', '0092_run_trade_pnl'),
     ]
 
     operations = [
+        migrations.RunPython(delete_referecing_null),
         migrations.RemoveField(
             model_name='position',
             name='exchange',

diff --git a/database/model/migrations/0094_auto_20240227_1808.py b/database/model/migrations/0094_auto_20240227_1808.py
@@ -4,13 +4,20 @@
 import django.db.models.deletion
 
 
+def delete_referecing_null(apps, _):
+    Trade = apps.get_model('model', 'Trade')
+
+    Trade.objects.filter(pipeline__id=None).delete()
+
+
 class Migration(migrations.Migration):
 
     dependencies = [
         ('model', '0093_auto_20240227_1753'),
     ]
 
     operations = [
+        migrations.RunPython(delete_referecing_null),
         migrations.RemoveField(
             model_name='trade',
             name='exchange',