diff options
Diffstat (limited to 'users/models/domain.py')
-rw-r--r-- | users/models/domain.py | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/users/models/domain.py b/users/models/domain.py index a3815ee..d2b17e2 100644 --- a/users/models/domain.py +++ b/users/models/domain.py @@ -49,10 +49,7 @@ class Domain(models.Model): @classmethod def get_remote_domain(cls, domain: str) -> "Domain": - try: - return cls.objects.get(domain=domain, local=False) - except cls.DoesNotExist: - return cls.objects.create(domain=domain, local=False) + return cls.objects.get_or_create(domain=domain, local=False)[0] @classmethod def get_domain(cls, domain: str) -> Optional["Domain"]: @@ -93,3 +90,4 @@ class Domain(models.Model): raise ValueError( f"Service domain {self.service_domain} is already a domain elsewhere!" ) + super().save(*args, **kwargs) |