added

pgbovine · pgbovine · commit fa5bfa51affb · 2013-12-29T00:14:57.000-05:00
diff --git a/v3/load_matrix_problem.py b/v3/load_matrix_problem.py
@@ -0,0 +1,24 @@
+#!/usr/bin/python
+
+import cgi
+import doctest
+import json
+import sys
+
+form = cgi.FieldStorage()
+prob_name = form['problem_name'].value
+assert type(prob_name) is str # prevent subclassing shenanigans
+# whitelist
+assert prob_name in ('python_comprehension-1',)
+
+fn = 'matrix-demo/' + prob_name + '.py'
+cod = open(fn).read()
+
+p = doctest.DocTestParser()
+examples = p.get_examples(cod)
+if len(examples):
+  first_ex = examples[0]
+  testCod = 'result = ' + first_ex.source
+
+print("Content-type: text/plain; charset=iso-8859-1\n")
+print(json.dumps(dict(code=cod, test=testCod)))
diff --git a/v3/submit_matrix_problem.py b/v3/submit_matrix_problem.py
@@ -0,0 +1,39 @@
+#!/usr/bin/python
+
+import cgi
+import doctest
+import json
+import urllib
+import urllib2
+
+form = cgi.FieldStorage()
+user_code = form['submitted_code'].value
+prob_name = form['problem_name'].value
+#assert type(prob_name) is str
+
+# whitelist
+assert prob_name in ('python_comprehension-1',)
+
+test_fn = 'matrix-demo/' + prob_name + '.test.py'
+test_cod = open(test_fn).read()
+
+# concatenate!
+script = test_cod + '\n' + user_code + \
+'''
+import doctest
+(n_fail, n_tests) = doctest.testmod(verbose=False)
+if n_fail == 0:
+  print("All %d tests passed!" % n_tests)
+'''
+
+# run on the EC2 sandbox
+url = 'http://ec2-107-20-94-197.compute-1.amazonaws.com/cgi-bin/run_code.py'
+values = {'user_script' : script}
+
+data = urllib.urlencode(values)
+req = urllib2.Request(url, data)
+response = urllib2.urlopen(req)
+the_page = response.read()
+
+print("Content-type: text/plain; charset=iso-8859-1\n")
+print(the_page)
