create lectures and attendence models

Author: Imafe

api/migrations/0003_courselecture_attendenceacknowledgement.py

@@ -0,0 +1,35 @@
+# Generated by Django 5.0.2 on 2024-03-13 13:57
+
+import django.db.models.deletion
+from django.conf import settings
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('api', '0002_alter_user_role_usercourse'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='CourseLecture',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('start_time', models.DateTimeField()),
+                ('end_time', models.DateTimeField()),
+                ('lecture_type', models.CharField(choices=[('lecture', 'Lecture'), ('practical', 'Practical'), ('workshop', 'Workshop'), ('exam', 'Exam')], default='lecture')),
+                ('course', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='course_lecture', to='api.course')),
+            ],
+        ),
+        migrations.CreateModel(
+            name='AttendenceAcknowledgement',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('attended_student', models.BooleanField(default=False)),
+                ('attended_teacher', models.BooleanField(default=False)),
+                ('student', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='user_ack', to=settings.AUTH_USER_MODEL)),
+                ('lecture', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='lecture', to='api.courselecture')),
+            ],
+        ),
+    ]

api/models.py

@@ -1,3 +1,4 @@
+import datetime
 from django.contrib.auth.models import AbstractUser, AbstractBaseUser, UserManager
 from django.db import models
 from django.contrib.auth.validators import UnicodeUsernameValidator
@@ -52,6 +53,11 @@ def clean(self):
     def get_classes(self):
         return [uc.course for uc in UserCourse.objects.filter(user__id=self.id)]
 
+class LectureTypes(models.TextChoices):
+    LECTURE = "lecture"
+    PRACTICAL = "practical"
+    WORKSHOP = "workshop"
+    EXAM = "exam"
 
 class Course(models.Model):
     course_name = models.CharField(max_length=50)
@@ -68,11 +74,51 @@ def get_enrolled_students(self):
     def get_teachers(self):
         return [uc.user for uc in UserCourse.objects.filter(user__role=AccountRoles.TEACHER)]
 
+    def get_lectures(self):
+        return [lecture for lecture in CourseLecture.objects.filter(course=self)]
+
     def add_user_to_course(self, user: User):
-        UserCourse.objects.create(user=user, course=self)
+        UserCourse.objects.create(user=user, course=self).save()
+
+    def add_lecture_to_course(self, start_time: datetime.datetime, end_time: datetime.datetime, lecture_type : LectureTypes = LectureTypes.LECTURE):
+        CourseLecture.objects.create(start_time=start_time, end_time=end_time, lecture_type=lecture_type, course=self).save()
 
 # This is for storing users (teachers/students) in courses
 class UserCourse(models.Model):
     user = models.ForeignKey(User, null=False, related_name='user', on_delete=models.CASCADE)
     course = models.ForeignKey(Course, null=False, related_name='course', on_delete=models.CASCADE)
 
+class CourseLecture(models.Model):
+    course = models.ForeignKey(Course, null=False, related_name='course_lecture', on_delete=models.CASCADE)
+    start_time = models.DateTimeField()
+    end_time = models.DateTimeField()
+    lecture_type = models.CharField(
+        choices=LectureTypes.choices,
+        default=LectureTypes.LECTURE,
+    )
+
+    def set_attendence_user(self, student : User, teacher=False):
+        queryset = AttendenceAcknowledgement.objects.filter(lecture=self, student=student)
+        if not queryset: ack = AttendenceAcknowledgement.objects.create(lecture=self, student=student)
+        else: ack = queryset[0]
+
+        if teacher: ack.attended_teacher = True
+        else: ack.attended_student = True
+        ack.save()
+
+    def get_attendence_user(self, student : User):
+        queryset = AttendenceAcknowledgement.objects.filter(lecture=self, student=student)
+        return None if not queryset else queryset[0]
+    
+    def get_attendence(self): 
+        queryset = AttendenceAcknowledgement.objects.filter(lecture=self)
+        return [] if not queryset else queryset[0]
+
+class AttendenceAcknowledgement(models.Model):
+    attended_student = models.BooleanField(default=False)
+    attended_teacher = models.BooleanField(default=False)
+    student = models.ForeignKey(User, null=False, related_name='user_ack', on_delete=models.CASCADE)
+    lecture = models.ForeignKey(CourseLecture, null=False, related_name='lecture', on_delete=models.CASCADE)
+
+
+