save project to group if group session is active

backface · Aug 5, 2024 · 58fdc92 · 58fdc92
1 parent 103e68d
commit 58fdc92
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 24 deletions.
diff --git a/api/api_legacy.py b/api/api_legacy.py
@@ -15,7 +15,7 @@
 import base64
 
 from apps.projects.models import Project
-from apps.classrooms.models import Group, Unit
+from apps.classrooms.models import Group, SelectedProject
 
 
 # TODO:
@@ -239,20 +239,19 @@ def save_project(
 
     if "group" in request.session:
         group = Group.objects.get(id=request.session["group"])
-        if group.current_unit:
-            unit = Unit.objects.get(id=group.current_unit)
-            unit.projects.add(project)
-            unit.save()
-        else:
-            group.projects.add(project)
-            group.save()
+        newproject = SelectedProject(
+            group=group,
+            project=project,
+            is_starter=False,
+            unit_id=group.current_unit or "",
+        )
+        newproject.save()
+        # if group.current_unit:
+        #     unit = Unit.objects.get(id=group.current_unit)
+        #     unit.projects.add(project)
+        #     unit.save()
+        # else:
+        #     group.projects.add(project)
+        #     group.save()
 
     return {"text": f"project {projectname} {'created' if created else 'updated'}"}
-
-
-@api.get("hello")
-def hello_world(request):
-    """
-    Just a Test
-    """
-    return {"msg": "Hello, World!"}
diff --git a/api/api_v1.py b/api/api_v1.py
@@ -22,7 +22,7 @@
 from allauth.account.utils import send_email_confirmation
 
 from apps.projects.models import Project
-from apps.classrooms.models import Group, Unit
+from apps.classrooms.models import Group, SelectedProject
 
 
 api = NinjaAPI(
@@ -414,13 +414,20 @@ def save_project(request, username: str, projectname: str):
 
     if "group" in request.session:
         group = Group.objects.get(id=request.session["group"])
-        if group.current_unit:
-            unit = Unit.objects.get(id=group.current_unit)
-            unit.projects.add(project)
-            unit.save()
-        else:
-            group.projects.add(project)
-            group.save()
+        newproject = SelectedProject(
+            group=group,
+            project=project,
+            is_starter=False,
+            unit_id=group.current_unit or "",
+        )
+        newproject.save()
+        # if group.current_unit:
+        #     unit = Unit.objects.get(id=group.current_unit)
+        #     unit.projects.add(project)
+        #     unit.save()
+        # else:
+        #     group.projects.add(project)
+        #     group.save()
 
     return Message(message=f"project {projectname} saved")