fix: Fix minor issues of backend

Author: pinecone28

backend/problem/serializers.py

@@ -131,7 +131,7 @@ class BaseProblemSerializer(serializers.ModelSerializer):
     def get_contest_name(self, obj):
         if obj.contest:
             return obj.contest.title
-    
+
     def get_assignment_name(self, obj):
         if obj.assignment:
             return obj.assignment.title

backend/problem/tests.py

@@ -275,12 +275,12 @@ def setUp(self):
 
     def test_get_assignment_problem_list(self):
         self.test_create_assignment_problem()
-        res = self.client.get(f"{self.url}?assignment_id={self.assignment_id}")
+        res = self.client.get(f"{self.url}?course_id={self.course_id}&assignment_id={self.assignment_id}")
         self.assertSuccess(res)
 
     def test_get_assignment_problem(self):
         problem_id = self.test_create_assignment_problem()["id"]
-        res = self.client.get(f"{self.url}?assignment_id={self.assignment_id}&problem_id={problem_id}")
+        res = self.client.get(f"{self.url}?course_id={self.course_id}&assignment_id={self.assignment_id}&problem_id={problem_id}")
         self.assertSuccess(res)
 
     def test_create_assignment_problem(self):

backend/problem/views/professor.py

@@ -283,7 +283,7 @@ def get(self, request):
     @admin_role_required
     def post(self, request):
         data = request.data
-        
+
         try:
             course = Course.objects.get(id=data.pop("course_id"))
             ensure_created_by(course, request.user)

backend/submission/views.py

@@ -4,7 +4,6 @@
 from drf_yasg import openapi
 
 from assignment.models import Assignment
-from course.models import Course
 from contest.models import ContestStatus, ContestRuleType
 from judge.tasks import judge_task
 from options.options import SysOptions
@@ -78,7 +77,11 @@ def post(self, request):
             return self.error(error)
 
         try:
-            problem = Problem.objects.get(id=data["problem_id"], contest_id=data.get("contest_id"), assignment_id=data.get("assignment_id"), course_id=data.get("course_id"), visible=True)
+            problem = Problem.objects.get(id=data["problem_id"],
+                                          contest_id=data.get("contest_id"),
+                                          assignment_id=data.get("assignment_id"),
+                                          course_id=data.get("course_id"),
+                                          visible=True)
         except Problem.DoesNotExist:
             return self.error("Problem not exist")
         if data["language"] not in problem.languages: