Use database constraint to prevent relationships with yourself.

2020-03-14 22:20:01 +00:00
parent 1cdd7ea1fc
commit ffe81291ad
3 changed files with 33 additions and 2 deletions
--- a/fedireads/models/user.py
+++ b/fedireads/models/user.py
@ -95,6 +95,10 @@ class UserRelationship(FedireadsModel):
            models.UniqueConstraint(
                fields=['user_subject', 'user_object'],
                name='%(class)s_unique'
+            ),
+            models.CheckConstraint(
+                check=~models.Q(user_subject=models.F('user_object')),
+                name='%(class)s_no_self'
            )
        ]