sg and cw access query changed

supervisor/methods.py

@@ -1,3 +1,4 @@
+from supervisor.models import TA
 from studentportal.models import Project
 from supervisor.decorators import supervisor_logged_in
 from credentials import SG_USERS, CW_USERS, FULL_USERS
@@ -6,12 +7,15 @@
 @supervisor_logged_in
 def filtered_projects(request, **kwargs):
     queryset = None
-    if request.user.email in FULL_USERS:
+    user=TA.objects.filter(email=request.user.email).first()
+    if user==None:
+        queryset = Project.objects.none()
+    elif user.sg_access and user.cw_access: 
         queryset = Project.objects.filter(**kwargs)
-    elif request.user.email in SG_USERS:
+    elif user.sg_access:
         queryset = Project.objects.filter(category__name='SG', **kwargs)
-    elif request.user.email in CW_USERS:
+    elif user.cw_access:
         queryset = Project.objects.filter(category__name='CW', **kwargs)
     else:
-        queryset = Project.objects.filter(**kwargs)
+        queryset = Project.objects.none()
     return queryset

supervisor/migrations/0002_auto_20210630_0316.py

@@ -0,0 +1,23 @@
+# Generated by Django 3.0.8 on 2021-06-30 03:16
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('supervisor', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='ta',
+            name='cw_access',
+            field=models.BooleanField(default=False),
+        ),
+        migrations.AddField(
+            model_name='ta',
+            name='sg_access',
+            field=models.BooleanField(default=False),
+        ),
+    ]

supervisor/models.py

@@ -116,6 +116,8 @@ def get_priority(self):
 class TA(models.Model):
     email = models.CharField(max_length=100)
     instructor = models.BooleanField(default=False)
+    sg_access = models.BooleanField(default=False)
+    cw_access = models.BooleanField(default=False)
 
 
 class Diff(models.Model):