summaryrefslogtreecommitdiff
path: root/server
diff options
context:
space:
mode:
Diffstat (limited to 'server')
-rw-r--r--server/handlers.py10
-rw-r--r--server/user_session.py7
2 files changed, 17 insertions, 0 deletions
diff --git a/server/handlers.py b/server/handlers.py
index f427056..d1ad634 100644
--- a/server/handlers.py
+++ b/server/handlers.py
@@ -288,6 +288,15 @@ class GetAttempts(CodeqService):
request.reply({'code': 0, 'data': request.session.get_attempts(language)})
+class GetSolutions(CodeqService):
+ def process(self, request):
+ problem_ids = request.data.get('problem_ids')
+ if problem_ids is None:
+ request.reply({'code': 1, 'message': 'No problem IDs specified'})
+ else:
+ request.reply({'code': 0, 'data': request.session.get_solutions(problem_ids)})
+
+
class GetCurrentSolution(CodeqService):
def process(self, request):
js = request.data
@@ -354,6 +363,7 @@ incoming_handlers = {
'signup': Signup(),
'change_password': ChangePassword(),
'get_attempts': GetAttempts(),
+ 'get_solutions': GetSolutions(),
'get_current_solution': GetCurrentSolution(),
'logout': Logout(),
'activity': Activity(),
diff --git a/server/user_session.py b/server/user_session.py
index 63d3459..af28046 100644
--- a/server/user_session.py
+++ b/server/user_session.py
@@ -433,6 +433,13 @@ class UserSession(object):
results.extend(group_results)
return results
+ def get_solutions(self, problem_ids):
+ uid = self.get_uid()
+ solutions = {s.problem_id: s.content for s in Solution.filter(codeq_user_id=uid)
+ if s.problem_id in problem_ids}
+ # return solutions in the same order as in [problem_pids]
+ return [solutions[pid] for pid in problem_ids if pid in solutions]
+
def send(self, json_obj):
"""Sends a message to the user.