Record solved problems in database

1 files changed, 4 insertions, 0 deletions

diff --git a/server/python_session.py b/server/python_session.py
index f7ec53f..4fcc3b5 100644
--- a/server/python_session.py
+++ b/server/python_session.py
@@ -13,6 +13,7 @@ import threading
 import time
 
 import server
+import server.user_session
 from db.models import Problem
 
 __all__ = ['PythonSession']
@@ -110,6 +111,9 @@ class PythonSession(server.LanguageSession):
 
         try:
             passed, hints = problem_module.test(self.run, program)
+            if passed:
+                session = server.user_session.get_session_by_id(sid)
+                session.update_solution(problem_id, done=True)
         except AttributeError as ex:
             hints = [{'id': 'system_error', 'args': {'message': 'test function does not exist'}}]