minor update
This commit is contained in:
+14
-4
@@ -1,5 +1,6 @@
|
||||
from django.contrib.auth.models import AbstractUser
|
||||
from django.core.validators import FileExtensionValidator
|
||||
from django.conf import settings
|
||||
from django.db import models
|
||||
|
||||
class User(AbstractUser):
|
||||
@@ -98,15 +99,24 @@ class Department(models.Model):
|
||||
description = models.TextField(blank=True, null=True)
|
||||
created_at = models.DateTimeField(auto_now_add=True)
|
||||
|
||||
# Each department can have many users
|
||||
users = models.ManyToManyField(
|
||||
User,
|
||||
settings.AUTH_USER_MODEL,
|
||||
related_name="departments",
|
||||
blank=True,
|
||||
limit_choices_to={'is_staff': True}
|
||||
blank=True
|
||||
)
|
||||
|
||||
# One admin per department
|
||||
admin = models.OneToOneField(
|
||||
settings.AUTH_USER_MODEL,
|
||||
on_delete=models.SET_NULL,
|
||||
related_name="admin_of_department",
|
||||
null=True,
|
||||
blank=True
|
||||
)
|
||||
|
||||
class Meta:
|
||||
ordering = ["name"]
|
||||
|
||||
def __str__(self):
|
||||
return self.name.name
|
||||
return self.name
|
||||
Reference in New Issue
Block a user