|
|
@@ -1,5 +1,6 @@
|
|
|
from django.contrib.contenttypes.fields import GenericRelation
|
|
|
from django.db import models
|
|
|
+from django.db.models import Q
|
|
|
from django.urls import reverse
|
|
|
|
|
|
from netbox.models import NestedGroupModel, PrimaryModel
|
|
|
@@ -36,12 +37,10 @@ class Tenant(PrimaryModel):
|
|
|
department.
|
|
|
"""
|
|
|
name = models.CharField(
|
|
|
- max_length=100,
|
|
|
- unique=True
|
|
|
+ max_length=100
|
|
|
)
|
|
|
slug = models.SlugField(
|
|
|
- max_length=100,
|
|
|
- unique=True
|
|
|
+ max_length=100
|
|
|
)
|
|
|
group = models.ForeignKey(
|
|
|
to='tenancy.TenantGroup',
|
|
|
@@ -62,6 +61,28 @@ class Tenant(PrimaryModel):
|
|
|
|
|
|
class Meta:
|
|
|
ordering = ['name']
|
|
|
+ constraints = (
|
|
|
+ models.UniqueConstraint(
|
|
|
+ fields=('group', 'name'),
|
|
|
+ name='%(app_label)s_%(class)s_unique_group_name',
|
|
|
+ violation_error_message="Tenant name must be unique per group."
|
|
|
+ ),
|
|
|
+ models.UniqueConstraint(
|
|
|
+ fields=('name',),
|
|
|
+ name='%(app_label)s_%(class)s_unique_name',
|
|
|
+ condition=Q(group__isnull=True)
|
|
|
+ ),
|
|
|
+ models.UniqueConstraint(
|
|
|
+ fields=('group', 'slug'),
|
|
|
+ name='%(app_label)s_%(class)s_unique_group_slug',
|
|
|
+ violation_error_message="Tenant slug must be unique per group."
|
|
|
+ ),
|
|
|
+ models.UniqueConstraint(
|
|
|
+ fields=('slug',),
|
|
|
+ name='%(app_label)s_%(class)s_unique_slug',
|
|
|
+ condition=Q(group__isnull=True)
|
|
|
+ ),
|
|
|
+ )
|
|
|
|
|
|
def __str__(self):
|
|
|
return self.name
|