From acc9ee66e79e677472bf8ced5812da64a3409b0b Mon Sep 17 00:00:00 2001
From: Faraphel <rc60650@hotmail.com>
Date: Thu, 14 Dec 2023 20:57:08 +0100
Subject: [PATCH] added email to teaching unit

---
 Palto/Palto/admin.py     | 4 ++--
 Palto/Palto/factories.py | 6 ++----
 Palto/Palto/models.py    | 1 +
 3 files changed, 5 insertions(+), 6 deletions(-)

diff --git a/Palto/Palto/admin.py b/Palto/Palto/admin.py
index 86277a2..10d3d9f 100644
--- a/Palto/Palto/admin.py
+++ b/Palto/Palto/admin.py
@@ -34,8 +34,8 @@ class AdminStudentGroup(admin.ModelAdmin):
 
 @admin.register(models.TeachingUnit)
 class AdminTeachingUnit(admin.ModelAdmin):
-    list_display = ("id", "name")
-    search_fields = ("id", "name")
+    list_display = ("id", "name", "email")
+    search_fields = ("id", "name", "email")
 
 
 @admin.register(models.StudentCard)
diff --git a/Palto/Palto/factories.py b/Palto/Palto/factories.py
index 21a7baa..15e1079 100644
--- a/Palto/Palto/factories.py
+++ b/Palto/Palto/factories.py
@@ -90,10 +90,8 @@ class FakeTeachingUnitFactory(factory.django.DjangoModelFactory):
         model = models.TeachingUnit
 
     name: str = factory.Faker("administrative_unit")
-
-    department: models.Department = factory.SubFactory(
-        FakeDepartmentFactory
-    )
+    email: str = factory.Faker("company_email")
+    department: models.Department = factory.SubFactory(FakeDepartmentFactory)
 
     @factory.post_generation
     def managers(self, create, extracted, **kwargs):
diff --git a/Palto/Palto/models.py b/Palto/Palto/models.py
index 761ef10..ba54285 100644
--- a/Palto/Palto/models.py
+++ b/Palto/Palto/models.py
@@ -318,6 +318,7 @@ class TeachingUnit(models.Model, ModelPermissionHelper):
 
     id: uuid.UUID = models.UUIDField(default=uuid.uuid4, primary_key=True, editable=False, max_length=36)
     name: str = models.CharField(max_length=64)
+    email: str = models.EmailField(null=True, blank=True)
 
     department = models.ForeignKey(to=Department, on_delete=models.CASCADE, related_name="teaching_units")