make invoice_id unique for AppUserTransaction

GooeyAI · Oct 13, 2023 · a0444ab · a0444ab
1 parent 65e7159
commit a0444ab
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/app_users/models.py b/app_users/models.py
@@ -212,13 +212,12 @@ class AppUserTransaction(models.Model):
     user = models.ForeignKey(
         "AppUser", on_delete=models.CASCADE, related_name="transactions"
     )
-    invoice_id = models.CharField(max_length=255)
+    invoice_id = models.CharField(max_length=255, unique=True)
     amount = models.IntegerField()
     end_balance = models.IntegerField()
     created_at = models.DateTimeField(editable=False, blank=True, default=timezone.now)
 
     class Meta:
-        unique_together = ["user", "invoice_id"]
         verbose_name = "Transaction"
 
     def __str__(self):

diff --git a/scripts/migrate_txns.py b/scripts/migrate_txns.py
@@ -7,9 +7,9 @@
 
 
 def run():
-    with ThreadPoolExecutor(1000) as pool:
+    with ThreadPoolExecutor(500) as pool:
         for user in AppUser.objects.all():
-            step1(pool, user.id, user.uid)
+            pool.submit(step1, pool, user.id, user.uid)
 
 
 def step1(pool: ThreadPoolExecutor, user_id: int, uid: str):