feat: Update wheel

.gitignore

@@ -20,7 +20,6 @@ tmp
 
 # pip-related files
 *.egg-info
-dist
 pip-selfcheck.json
 
 # ok-related files

main.py

@@ -0,0 +1,13 @@
+import json
+from pytutor import generate_trace, server
+
+code = """
+num = input("Enter a number: ")
+print(num)
+"""
+trace = generate_trace.run(code, "[]")
+print(trace)
+
+print("=========================================")
+trace = generate_trace.run(code, "[10]")
+print(trace)

pytutor/generate_trace.py

@@ -40,3 +40,18 @@ def run_logger(source, setup, modules=None):
                                            custom_modules={'pg_setup': setup},
                                            extra_modules=modules)
 
+def run(source, raw_input, setup="", modules=None):
+    modules = modules or {}
+    # Add current directory to path to make sure that imports work consistently
+    sys.path.append(os.getcwd() + '/')
+
+    finalizer = lambda code,trace: json_finalizer(code, trace, modules)
+    return pg_logger.exec_script_str_local(source,
+                                           raw_input, # JSON list of strings for simulated raw_
+                                           True, # output cumulative trace (to display exited frames)
+                                           False, # render primitives as heap objects
+                                           finalizer,
+                                           separate_stdout_by_module=False,
+                                           disable_security_checks=True,
+                                           custom_modules={'pg_setup': setup},
+                                           extra_modules=modules)

pytutor/pg_logger.py

@@ -29,7 +29,6 @@
 # of all in-scope data structures after each executed instruction.
 
 
-import imp
 import sys
 import bdb # the KEY import here!
 import re
@@ -1353,7 +1352,7 @@ def _runscript(self, script_str):
         # which emulates "from <module> import *"
         for mn in self.custom_modules:
             # http://code.activestate.com/recipes/82234-importing-a-dynamically-generated-module/
-            new_m = imp.new_module(mn)
+            new_m = types.ModuleType(mn)
             exec(self.custom_modules[mn], new_m.__dict__) # exec in custom globals
             user_globals.update(new_m.__dict__)
 

setup.py

@@ -2,7 +2,7 @@
 setup(
   name='pytutor',
   packages=['pytutor'], # this must be the same as the name above
-  version='1.0.0',
+  version='1.0.1',
   description='An interface to run Phillip Guo\'s Python Tutor',
   author='Sumukh Sridhara',
   author_email='sumukh@berkeley.edu',