fix: avoid error when creating same model names with different users

PauloUbirajara · Apr 28, 2024 · 751b6a0 · 751b6a0
1 parent 4a29307
commit 751b6a0
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/apps/category/models.py b/apps/category/models.py
@@ -8,10 +8,11 @@ class Category(models.Model):
 
     user = models.ForeignKey(to="auth.User", on_delete=models.CASCADE)
 
-    name = models.TextField(unique=True)
+    name = models.TextField()
 
     created_at = models.DateTimeField(auto_now_add=True)
     updated_at = models.DateTimeField(auto_now=True)
 
     class Meta:
+        unique_together = ["name", "user"]
         ordering = ["-updated_at"]
diff --git a/apps/transaction/models.py b/apps/transaction/models.py
@@ -19,4 +19,5 @@ class Transaction(models.Model):
     updated_at = models.DateTimeField(auto_now=True)
 
     class Meta:
+        unique_together = ["name", "user"]
         ordering = ["-updated_at"]
diff --git a/apps/wallet/models.py b/apps/wallet/models.py
@@ -9,7 +9,7 @@ class Wallet(models.Model):
     user = models.ForeignKey(to="auth.User", on_delete=models.CASCADE)
     currency = models.ForeignKey(to="currency.Currency", on_delete=models.CASCADE)
 
-    name = models.TextField(unique=True)
+    name = models.TextField()
     description = models.TextField(blank=True, default="")
 
     created_at = models.DateTimeField(auto_now_add=True)
@@ -19,4 +19,5 @@ def __str__(self) -> str:
         return "{} ({})".format(self.name, self.id)
 
     class Meta:
+        unique_together = ["name", "user"]
         ordering = ["-updated_at"]