Some more optimizations

judge/utils/problems.py

@@ -19,14 +19,17 @@ __all__ = [
     "contest_completed_ids",
     "get_result_data",
     "user_completed_ids",
-    "user_authored_ids",
     "user_editable_ids",
+    "user_tester_ids",
 ]
 
 
-def user_authored_ids(profile):
-    result = set(Problem.objects.filter(authors=profile).values_list("id", flat=True))
-    return result
+def user_tester_ids(profile):
+    return set(
+        Problem.testers.through.objects.filter(profile=profile).values_list(
+            "problem_id", flat=True
+        )
+    )
 
 
 def user_editable_ids(profile):