diff options
Diffstat (limited to 'Lib/rexec.py')
| -rw-r--r-- | Lib/rexec.py | 12 |
1 files changed, 11 insertions, 1 deletions
diff --git a/Lib/rexec.py b/Lib/rexec.py index 7a239bc..04ff405 100644 --- a/Lib/rexec.py +++ b/Lib/rexec.py @@ -22,6 +22,7 @@ import sys import __builtin__ import os import ihooks +import imp __all__ = ["RExec"] @@ -83,6 +84,9 @@ class RHooks(ihooks.Hooks): # Called by RExec instance to complete initialization self.rexec = rexec + def get_suffixes(self): + return self.rexec.get_suffixes() + def is_builtin(self, name): return self.rexec.is_builtin(name) @@ -144,6 +148,8 @@ class RExec(ihooks._Verbose): nok_builtin_names = ('open', 'file', 'reload', '__import__') + ok_file_types = (imp.C_EXTENSION, imp.PY_SOURCE) + def __init__(self, hooks = None, verbose = 0): """Returns an instance of the RExec class. @@ -203,7 +209,6 @@ class RExec(ihooks._Verbose): if sys.modules.has_key(name): src = sys.modules[name] else: - import imp src = imp.load_dynamic(name, filename, file) dst = self.copy_except(src, []) return dst @@ -214,6 +219,11 @@ class RExec(ihooks._Verbose): # Helpers for RHooks + def get_suffixes(self): + return [item # (suff, mode, type) + for item in imp.get_suffixes() + if item[2] in self.ok_file_types] + def is_builtin(self, mname): return mname in self.ok_builtin_modules |