Painstakingly added an endpoint to show messages from the current study group of the user

stude/studygroup_messages/admin.py

@@ -0,0 +1,5 @@
+from django.contrib import admin
+from .models import Message
+
+
+admin.site.register(Message)

stude/studygroup_messages/apps.py

@@ -0,0 +1,6 @@
+from django.apps import AppConfig
+
+
+class StudygroupMessagesConfig(AppConfig):
+    default_auto_field = 'django.db.models.BigAutoField'
+    name = 'studygroup_messages'

stude/studygroup_messages/migrations/0001_initial.py

@@ -0,0 +1,28 @@
+# Generated by Django 4.2.2 on 2023-06-27 15:24
+
+from django.conf import settings
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+        ('study_groups', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='Message',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('message_content', models.TextField(max_length=1024)),
+                ('timestamp', models.DateField(auto_now_add=True)),
+                ('study_group', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='study_groups.studygroup')),
+                ('user', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to=settings.AUTH_USER_MODEL)),
+            ],
+        ),
+    ]

stude/studygroup_messages/models.py

@@ -0,0 +1,11 @@
+from django.db import models
+from study_groups.models import StudyGroup
+
+# Create your models here.
+
+
+class Message(models.Model):
+    user = models.ForeignKey('accounts.CustomUser', on_delete=models.CASCADE)
+    study_group = models.ForeignKey(StudyGroup, on_delete=models.CASCADE)
+    message_content = models.TextField(max_length=1024)
+    timestamp = models.DateField(auto_now_add=True)

stude/studygroup_messages/serializers.py

@@ -0,0 +1,13 @@
+from rest_framework import serializers
+from .models import Message
+from accounts.models import CustomUser
+
+
+class MessageSerializer(serializers.ModelSerializer):
+    user = serializers.SlugRelatedField(
+        queryset=CustomUser.objects.all(), slug_field='full_name')
+    group = serializers.CharField(source='subject.Subject', read_only=True)
+
+    class Meta:
+        model = Message
+        fields = '__all__'

stude/studygroup_messages/tests.py

@@ -0,0 +1,3 @@
+from django.test import TestCase
+
+# Create your tests here.

stude/studygroup_messages/urls.py

@@ -0,0 +1,13 @@
+from django.urls import include, path
+from rest_framework import routers
+from . import views
+
+router = routers.DefaultRouter()
+router.register(r'group_messages', views.MessageViewSet,
+                basename='group_messages')
+
+# Wire up our API using automatic URL routing.
+# Additionally, we include login URLs for the browsable API.
+urlpatterns = [
+    path('', include(router.urls)),
+]

stude/studygroup_messages/views.py

@@ -0,0 +1,47 @@
+from django.shortcuts import render
+from .models import Message
+from .serializers import MessageSerializer
+from rest_framework.permissions import IsAuthenticated
+from rest_framework.exceptions import PermissionDenied
+from rest_framework import viewsets
+from student_status.models import StudentStatus
+from study_groups.models import StudyGroupMembership
+# Create your views here.
+
+
+class MessageViewSet(viewsets.ModelViewSet):
+    serializer_class = MessageSerializer
+    permission_classes = [IsAuthenticated]
+
+    def get_object(self):
+        user = self.request.user
+        return Message.objects.get(user=user)
+
+    def get_queryset(self):
+        user = self.request.user
+
+        if not user.is_student:
+            raise PermissionDenied(
+                "You must be a student to view messages of your current study group"
+            )
+
+        # Get student_status id of the current user
+        student_status = StudentStatus.objects.filter(
+            user=user.id
+        ).values_list('study_group', flat=True).first()
+
+        print("User ID:", user.id)
+        print("Student_Status ID:", student_status)
+
+        # Get the study group id
+        print(StudyGroupMembership.objects.all())
+        study_group_id_list = StudyGroupMembership.objects.filter(
+            user=user.id).values_list('study_group').first()
+
+        print("Study Group List:", study_group_id_list)
+
+        # Now fetch the Messages matching the study group id
+        messages = Message.objects.filter(study_group=study_group_id_list)
+        return messages
+
+    # To-do: only allow destroy of messages if message is by the same user