class AssignmentService:

assignment_service.py

@@ -0,0 +1,77 @@
+import uuid
+import json
+import datetime
+import gradio as gr
+from datetime import datetime
+
+
+class AssignmentService:
+    def __init__(self, gcs_service):
+        self.bucket_name = "ai_assignment_submission"
+        self.gcs_service = gcs_service
+
+    def create_assignment_id(self):
+        while True:
+            assignment_id = "assign_" + str(uuid.uuid4())
+            file_name = f"assignments/{assignment_id}.json"
+            if not self.gcs_service.check_file_exists(self.bucket_name, file_name):
+                return assignment_id
+
+    def create_assignment(self, user_data, assignment_type, metadata):
+        assignment_id = self.create_assignment_id()
+        timestamp_now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        new_assignment = {
+            "assignment_id": assignment_id,
+            "assigner_data": user_data,
+            "assignment_type": assignment_type,
+            "metadata": metadata,
+            "submission_ids": [],
+            "timestamp": timestamp_now
+        }
+        self.save_assignment_to_gcs(new_assignment)
+        self.save_user_assignment_to_gcs(user_data, assignment_id)
+        return assignment_id
+
+    def save_assignment_to_gcs(self, assignment):
+        file_name = f"assignments/{assignment['assignment_id']}.json"
+        self.gcs_service.upload_json_string(self.bucket_name, file_name, json.dumps(assignment))
+
+    def save_user_assignment_to_gcs(self, user_id, assignment_id):
+        user_assignments = self.get_user_assignments(user_id)
+        assignment_data = self.get_assignment(assignment_id)
+        user_assignments[assignment_id] = {
+            "assignment_name": assignment_data["metadata"]["topic"],
+            "assignment_type": assignment_data["assignment_type"],
+            "timestamp": assignment_data["timestamp"]
+        }
+        self.gcs_service.upload_json_string(self.bucket_name, f"users/{user_id}/assignments.json", json.dumps(user_assignments))
+
+    def get_assignment(self, assignment_id):
+        try:
+            file_name = f"assignments/{assignment_id}.json"
+            assignment_json = self.gcs_service.download_as_string(self.bucket_name, file_name)
+            assignment_data = json.loads(assignment_json)
+        except Exception as e:
+            print(f"Error: {e}")
+            return "無效的作業 ID"
+        return assignment_data
+
+    def get_user_assignments(self, user_id):
+        try:
+            user_assignments_json = self.gcs_service.download_as_string(self.bucket_name, f"users/{user_id}/assignments.json")
+            user_assignments = json.loads(user_assignments_json)
+        except Exception as e:
+            print(f"Error: {e}")
+            user_assignments = {}
+        return user_assignments
+
+    def update_assignment_list(self, assigner_data):
+        choices = []
+        user_assignments = self.get_user_assignments(assigner_data)
+        for assignment_id in user_assignments.keys():
+            assignment_data = self.get_assignment(assignment_id)
+            metadata = assignment_data.get("metadata", {})
+            topic = metadata.get("topic", "未命名作业")
+            choice = (topic, assignment_id)
+            choices.append(choice)
+        return gr.update(choices=choices)

submission_service.py

@@ -0,0 +1,120 @@
+import uuid
+import json
+import datetime
+import gradio as gr
+from datetime import datetime
+
+
+class SubmissionService:
+    def __init__(self, gcs_service):
+        self.gcs_service = gcs_service
+        self.bucket_name = "ai_assignment_submission"
+
+    def create_submission_id(self):
+        while True:
+            submission_id = str(uuid.uuid4())
+            file_name = f"submissions/{submission_id}.json"
+            if not self.gcs_service.check_file_exists(self.bucket_name, file_name):
+                return submission_id
+
+    def submit_assignment(self, assignment_id, student_id, student_name, submission_content, file_name=None, bucket_name=None):
+        submission_id = self.create_submission_id()
+        timestamp_now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        submission_data = {
+            "file_name": file_name,
+            "bucket_name": bucket_name,
+            "content": submission_content
+        }
+        submission_data = {
+            "submission_id": submission_id,
+            "assignment_id": assignment_id,
+            "student_id": student_id,
+            "student_name": student_name,
+            "submission_data": submission_data,
+            "timestamp": timestamp_now
+        }
+
+        self.save_submission_to_gcs(submission_data)
+        self.save_user_submissions_to_gcs(student_id, submission_data)
+
+        return f"作業已收到，學生ID：{student_id}"
+
+    def get_assignment_data(self, assignment_id):
+        try:
+            file_name = f"assignments/{assignment_id}.json"
+            assignment_json = self.gcs_service.download_as_string(self.bucket_name, file_name)
+            assignment_data = json.loads(assignment_json)
+        except Exception as e:
+            print(f"Error: {e}")
+            return None
+        return assignment_data
+
+    def save_submission_to_gcs(self, submission_data):
+        file_name = f"submissions/{submission_data['submission_id']}.json"
+        self.gcs_service.upload_json_string(self.bucket_name, file_name, json.dumps(submission_data))
+
+    def save_user_submissions_to_gcs(self, student_id, submission_data):
+        try:
+            user_submissions_json = self.gcs_service.download_as_string(self.bucket_name, f"users/{student_id}/submissions.json")
+            user_submissions = json.loads(user_submissions_json)
+        except Exception as e:
+            print(f"Error: {e}")
+            user_submissions = {}
+
+        submission_id = submission_data['submission_id']
+        assignment_id = submission_data['assignment_id']
+        assignment_data = self.get_assignment_data(assignment_id)
+        user_submissions[submission_id] = {
+            "assignment_id": submission_data['assignment_id'],
+            "assignment_name": assignment_data["metadata"]["topic"],
+            "assignment_type": assignment_data["assignment_type"],
+            "timestamp": submission_data['timestamp'],
+        }
+        self.gcs_service.upload_json_string(self.bucket_name, f"users/{student_id}/submissions.json", json.dumps(user_submissions))
+
+    def get_submission_from_gcs(self, submission_id):
+        try:
+            file_name = f"submissions/{submission_id}.json"
+            submission_json = self.gcs_service.download_as_string(self.bucket_name, file_name)
+            submission_data = json.loads(submission_json)
+        except Exception as e:
+            print(f"Error: {e}")
+            return None
+        return submission_data
+
+    def list_submissions(self):
+        # 列出所有提交的作业
+        try:
+            prefix = "submissions/"
+            blobs = self.gcs_service.list_files(self.bucket_name, prefix)
+            submission_ids = [blob.name.split("/")[-1].replace(".json", "") for blob in blobs]
+            return submission_ids
+        except Exception as e:
+            print(f"Error: {e}")
+            return []
+
+    def get_student_submissions_list(self, student_id):
+        try:
+            user_submissions_json = self.gcs_service.download_as_string(self.bucket_name, f"users/{student_id}/submissions.json")
+            user_submissions = json.loads(user_submissions_json)
+        except Exception as e:
+            print(f"Error: {e}")
+            return {}
+        return user_submissions
+    
+    def update_submission_list(self, student_id):
+        choices = []
+        user_submissions = self.get_student_submissions_list(student_id)
+        for submission_id, submission in user_submissions.items():
+            topic = submission.get("assignment_name", "未命名作业")
+            choice = (topic, submission_id)
+            choices.append(choice)
+        return gr.update(choices=choices)
+
+    def get_student_submission(self, student_id):
+        submission_ids = self.list_submissions()
+        for submission_id in submission_ids:
+            submission_data = self.get_submission_from_gcs(submission_id)
+            if submission_data and submission_data.get("student_id") == student_id:
+                return submission_data["submission"]
+        return "未找到學生提交的作業"