Add Python push&pull services to wsgi_server

1 files changed, 21 insertions, 0 deletions

diff --git a/wsgi_server.py b/wsgi_server.py
index fda576b..324f338 100644
--- a/wsgi_server.py
+++ b/wsgi_server.py
@@ -143,6 +143,24 @@ class Query(CodeqService):
                 session.update_solution(problem_id, trace, program)
             return result
 
+# Pull stdout/stderr from the session's Python interpreter.
+class PythonPull(CodeqService):
+    def process(self, js, session):
+        python = session.get_python()
+        output = python.pull()
+        return {'code': 0, 'message': 'ok', 'terminal': {'text': output if output else ''}}
+
+# Push stdin to the session's Python interpreter.
+class PythonPush(CodeqService):
+    def process(self, js, session):
+        text = js.get('text')
+        if text is None:
+            return {'code': 1, 'message': 'No input specified'}
+
+        python = session.get_python()
+        python.push(text)
+        return {'code': 0, 'message': 'ok'}
+
 class Hint(CodeqService):
     def process(self, js, session):
         language = js.get('language')
@@ -211,6 +229,9 @@ api.add_route('/activity', activity)
 query = Query()
 api.add_route('/query', query)
 
+api.add_route('/python_pull', PythonPull())
+api.add_route('/python_push', PythonPush())
+
 hint = Hint()
 api.add_route('/hint', hint)