diff options
Diffstat (limited to 'server/user_session.py')
-rw-r--r-- | server/user_session.py | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/server/user_session.py b/server/user_session.py index c9d21b4..0abc421 100644 --- a/server/user_session.py +++ b/server/user_session.py @@ -6,6 +6,7 @@ import hashlib import base64 import random from . import prolog_session +from . import python_session from . import problems import db from errors.session import NoSuchSession, AuthenticationFailed @@ -30,6 +31,7 @@ class UserSession(object): self.uid = uid self.username = username self.prolog_session = None + self.python_session = None def destroy(self): """Destroys the session.""" @@ -39,6 +41,9 @@ class UserSession(object): if self.prolog_session is not None: self.prolog_session.end() self.prolog_session = None + if self.python_session is not None: + self.python_session.end() + self.python_session = None # TODO: add any cleanups as features are added! def get_sid(self): @@ -53,6 +58,12 @@ class UserSession(object): self.prolog_session = prolog_session.PrologSession() # lazy init return self.prolog_session + def get_python(self): + with self._access_lock: + if self.python_session is None: + self.python_session = python_session.PythonSession() # lazy init + return self.python_session + def get_problem_data(self, language, problem_group, problem): mod = problems.load_problem(language, problem_group, problem, 'en') conn = db.get_connection() |