feat: Wheel update

.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

@@ -1354,6 +1354,7 @@ def _runscript(self, script_str):
         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__)