upload: server/session_manager.py

server/session_manager.py

@@ -0,0 +1,40 @@
+"""
+server/session_manager.py
+
+Manages the Session 1 → Session 2 transition.
+
+Key responsibilities:
+  1. Wipe the filesystem: code written in Session 1 does NOT persist.
+  2. Preserve the task description and test suite (Session 2 must implement).
+  3. Randomize function/class names again so Session 2 cannot reconstruct
+     from Session 1 code memory (they share the handoff note only).
+"""
+
+import copy
+from server.task_generator import Task
+
+
+class SessionManager:
+    """
+    Handles the controlled transition between sessions.
+
+    After transition:
+      - task.files reset to starter_code (blank implementations)
+      - task description preserved (Session 2 sees the original task)
+      - test suites preserved (same tests run at submit)
+    """
+
+    def transition(self, task: Task) -> Task:
+        """
+        Wipe session 1 file state and return a fresh Task for session 2.
+
+        Args:
+            task: The task object at end of Session 1 (may have partial implementation).
+
+        Returns:
+            A new Task object with files reset to starter_code.
+        """
+        new_task = copy.deepcopy(task)
+        # Wipe all file contents back to starter (blank) state
+        new_task.files = copy.deepcopy(task.starter_code)
+        return new_task