Thomas Wouters | a977329 | 2006-04-21 09:43:23 +0000 | [diff] [blame] | 1 | """runpy.py - locating and running Python code using the module namespace |
| 2 | |
| 3 | Provides support for locating and running Python scripts using the Python |
| 4 | module namespace instead of the native filesystem. |
| 5 | |
| 6 | This allows Python code to play nicely with non-filesystem based PEP 302 |
| 7 | importers when locating support scripts as well as when importing modules. |
| 8 | """ |
| 9 | # Written by Nick Coghlan <ncoghlan at gmail.com> |
| 10 | # to implement PEP 338 (Executing Modules as Scripts) |
| 11 | |
Brett Cannon | aa93642 | 2012-04-27 15:30:58 -0400 | [diff] [blame] | 12 | |
Thomas Wouters | a977329 | 2006-04-21 09:43:23 +0000 | [diff] [blame] | 13 | import sys |
Nick Coghlan | be7e49f | 2012-07-20 23:40:09 +1000 | [diff] [blame] | 14 | import importlib.machinery # importlib first so we can test #15386 via -m |
Eric Snow | 6029e08 | 2014-01-25 15:32:46 -0700 | [diff] [blame] | 15 | import importlib.util |
jsnklln | e243bae | 2019-11-18 14:11:13 -0500 | [diff] [blame] | 16 | import io |
Brett Cannon | 82d2107 | 2013-06-15 14:27:21 -0400 | [diff] [blame] | 17 | import types |
Maor Kleinberger | 0911ea5 | 2020-03-08 22:43:17 +0200 | [diff] [blame] | 18 | import os |
Nick Coghlan | 720c7e2 | 2013-12-15 20:33:02 +1000 | [diff] [blame] | 19 | from pkgutil import read_code, get_importer |
Thomas Wouters | a977329 | 2006-04-21 09:43:23 +0000 | [diff] [blame] | 20 | |
| 21 | __all__ = [ |
Nick Coghlan | 260bd3e | 2009-11-16 06:49:25 +0000 | [diff] [blame] | 22 | "run_module", "run_path", |
Thomas Wouters | a977329 | 2006-04-21 09:43:23 +0000 | [diff] [blame] | 23 | ] |
| 24 | |
Nick Coghlan | 260bd3e | 2009-11-16 06:49:25 +0000 | [diff] [blame] | 25 | class _TempModule(object): |
| 26 | """Temporarily replace a module in sys.modules with an empty namespace""" |
| 27 | def __init__(self, mod_name): |
| 28 | self.mod_name = mod_name |
Brett Cannon | 82d2107 | 2013-06-15 14:27:21 -0400 | [diff] [blame] | 29 | self.module = types.ModuleType(mod_name) |
Nick Coghlan | 260bd3e | 2009-11-16 06:49:25 +0000 | [diff] [blame] | 30 | self._saved_module = [] |
| 31 | |
| 32 | def __enter__(self): |
| 33 | mod_name = self.mod_name |
| 34 | try: |
| 35 | self._saved_module.append(sys.modules[mod_name]) |
| 36 | except KeyError: |
| 37 | pass |
| 38 | sys.modules[mod_name] = self.module |
| 39 | return self |
| 40 | |
| 41 | def __exit__(self, *args): |
| 42 | if self._saved_module: |
| 43 | sys.modules[self.mod_name] = self._saved_module[0] |
| 44 | else: |
| 45 | del sys.modules[self.mod_name] |
| 46 | self._saved_module = [] |
| 47 | |
| 48 | class _ModifiedArgv0(object): |
| 49 | def __init__(self, value): |
| 50 | self.value = value |
| 51 | self._saved_value = self._sentinel = object() |
| 52 | |
| 53 | def __enter__(self): |
| 54 | if self._saved_value is not self._sentinel: |
| 55 | raise RuntimeError("Already preserving saved value") |
| 56 | self._saved_value = sys.argv[0] |
| 57 | sys.argv[0] = self.value |
| 58 | |
| 59 | def __exit__(self, *args): |
| 60 | self.value = self._sentinel |
| 61 | sys.argv[0] = self._saved_value |
Thomas Wouters | a977329 | 2006-04-21 09:43:23 +0000 | [diff] [blame] | 62 | |
Eric Snow | 32439d6 | 2015-05-02 19:15:18 -0600 | [diff] [blame] | 63 | # TODO: Replace these helpers with importlib._bootstrap_external functions. |
Thomas Wouters | ed03b41 | 2007-08-28 21:37:11 +0000 | [diff] [blame] | 64 | def _run_code(code, run_globals, init_globals=None, |
Nick Coghlan | 720c7e2 | 2013-12-15 20:33:02 +1000 | [diff] [blame] | 65 | mod_name=None, mod_spec=None, |
| 66 | pkg_name=None, script_name=None): |
Benjamin Peterson | f6489f9 | 2009-11-25 17:46:26 +0000 | [diff] [blame] | 67 | """Helper to run code in nominated namespace""" |
Thomas Wouters | a977329 | 2006-04-21 09:43:23 +0000 | [diff] [blame] | 68 | if init_globals is not None: |
| 69 | run_globals.update(init_globals) |
Nick Coghlan | 720c7e2 | 2013-12-15 20:33:02 +1000 | [diff] [blame] | 70 | if mod_spec is None: |
| 71 | loader = None |
| 72 | fname = script_name |
| 73 | cached = None |
| 74 | else: |
| 75 | loader = mod_spec.loader |
| 76 | fname = mod_spec.origin |
| 77 | cached = mod_spec.cached |
| 78 | if pkg_name is None: |
| 79 | pkg_name = mod_spec.parent |
Thomas Wouters | a977329 | 2006-04-21 09:43:23 +0000 | [diff] [blame] | 80 | run_globals.update(__name__ = mod_name, |
Nick Coghlan | 720c7e2 | 2013-12-15 20:33:02 +1000 | [diff] [blame] | 81 | __file__ = fname, |
| 82 | __cached__ = cached, |
Nick Coghlan | 761bb11 | 2012-07-14 23:59:22 +1000 | [diff] [blame] | 83 | __doc__ = None, |
Nick Coghlan | 720c7e2 | 2013-12-15 20:33:02 +1000 | [diff] [blame] | 84 | __loader__ = loader, |
| 85 | __package__ = pkg_name, |
| 86 | __spec__ = mod_spec) |
Georg Brandl | 7cae87c | 2006-09-06 06:51:57 +0000 | [diff] [blame] | 87 | exec(code, run_globals) |
Thomas Wouters | a977329 | 2006-04-21 09:43:23 +0000 | [diff] [blame] | 88 | return run_globals |
| 89 | |
| 90 | def _run_module_code(code, init_globals=None, |
Nick Coghlan | 720c7e2 | 2013-12-15 20:33:02 +1000 | [diff] [blame] | 91 | mod_name=None, mod_spec=None, |
| 92 | pkg_name=None, script_name=None): |
Benjamin Peterson | f6489f9 | 2009-11-25 17:46:26 +0000 | [diff] [blame] | 93 | """Helper to run code in new namespace with sys modified""" |
Nick Coghlan | 720c7e2 | 2013-12-15 20:33:02 +1000 | [diff] [blame] | 94 | fname = script_name if mod_spec is None else mod_spec.origin |
| 95 | with _TempModule(mod_name) as temp_module, _ModifiedArgv0(fname): |
Nick Coghlan | 260bd3e | 2009-11-16 06:49:25 +0000 | [diff] [blame] | 96 | mod_globals = temp_module.module.__dict__ |
Thomas Wouters | ed03b41 | 2007-08-28 21:37:11 +0000 | [diff] [blame] | 97 | _run_code(code, mod_globals, init_globals, |
Nick Coghlan | 720c7e2 | 2013-12-15 20:33:02 +1000 | [diff] [blame] | 98 | mod_name, mod_spec, pkg_name, script_name) |
Thomas Wouters | ed03b41 | 2007-08-28 21:37:11 +0000 | [diff] [blame] | 99 | # Copy the globals of the temporary module, as they |
| 100 | # may be cleared when the temporary module goes away |
| 101 | return mod_globals.copy() |
Thomas Wouters | a977329 | 2006-04-21 09:43:23 +0000 | [diff] [blame] | 102 | |
Brett Cannon | fdcdd9e | 2016-07-08 11:00:00 -0700 | [diff] [blame] | 103 | # Helper to get the full name, spec and code for a module |
Martin Panter | 657257e | 2015-12-03 01:23:10 +0000 | [diff] [blame] | 104 | def _get_module_details(mod_name, error=ImportError): |
Martin Panter | 7dda421 | 2015-12-10 06:47:06 +0000 | [diff] [blame] | 105 | if mod_name.startswith("."): |
| 106 | raise error("Relative module names not supported") |
| 107 | pkg_name, _, _ = mod_name.rpartition(".") |
| 108 | if pkg_name: |
| 109 | # Try importing the parent to avoid catching initialization errors |
| 110 | try: |
| 111 | __import__(pkg_name) |
| 112 | except ImportError as e: |
| 113 | # If the parent or higher ancestor package is missing, let the |
| 114 | # error be raised by find_spec() below and then be caught. But do |
| 115 | # not allow other errors to be caught. |
| 116 | if e.name is None or (e.name != pkg_name and |
| 117 | not pkg_name.startswith(e.name + ".")): |
| 118 | raise |
Martin Panter | 9c8aa9b | 2016-08-21 04:07:58 +0000 | [diff] [blame] | 119 | # Warn if the module has already been imported under its normal name |
| 120 | existing = sys.modules.get(mod_name) |
| 121 | if existing is not None and not hasattr(existing, "__path__"): |
| 122 | from warnings import warn |
| 123 | msg = "{mod_name!r} found in sys.modules after import of " \ |
| 124 | "package {pkg_name!r}, but prior to execution of " \ |
| 125 | "{mod_name!r}; this may result in unpredictable " \ |
| 126 | "behaviour".format(mod_name=mod_name, pkg_name=pkg_name) |
| 127 | warn(RuntimeWarning(msg)) |
Martin Panter | 7dda421 | 2015-12-10 06:47:06 +0000 | [diff] [blame] | 128 | |
Nick Coghlan | 720c7e2 | 2013-12-15 20:33:02 +1000 | [diff] [blame] | 129 | try: |
Eric Snow | 6029e08 | 2014-01-25 15:32:46 -0700 | [diff] [blame] | 130 | spec = importlib.util.find_spec(mod_name) |
Nick Coghlan | 720c7e2 | 2013-12-15 20:33:02 +1000 | [diff] [blame] | 131 | except (ImportError, AttributeError, TypeError, ValueError) as ex: |
| 132 | # This hack fixes an impedance mismatch between pkgutil and |
| 133 | # importlib, where the latter raises other errors for cases where |
| 134 | # pkgutil previously raised ImportError |
Martin Panter | 9c8aa9b | 2016-08-21 04:07:58 +0000 | [diff] [blame] | 135 | msg = "Error while finding module specification for {!r} ({}: {})" |
Pablo Galindo | ef67512 | 2020-03-31 12:23:55 +0100 | [diff] [blame] | 136 | if mod_name.endswith(".py"): |
| 137 | msg += (f". Try using '{mod_name[:-3]}' instead of " |
| 138 | f"'{mod_name}' as the module name.") |
Martin Panter | 7dda421 | 2015-12-10 06:47:06 +0000 | [diff] [blame] | 139 | raise error(msg.format(mod_name, type(ex).__name__, ex)) from ex |
Nick Coghlan | 720c7e2 | 2013-12-15 20:33:02 +1000 | [diff] [blame] | 140 | if spec is None: |
Martin Panter | 657257e | 2015-12-03 01:23:10 +0000 | [diff] [blame] | 141 | raise error("No module named %s" % mod_name) |
Nick Coghlan | 720c7e2 | 2013-12-15 20:33:02 +1000 | [diff] [blame] | 142 | if spec.submodule_search_locations is not None: |
Nick Coghlan | 3f48ae3 | 2009-02-08 01:58:26 +0000 | [diff] [blame] | 143 | if mod_name == "__main__" or mod_name.endswith(".__main__"): |
Martin Panter | 657257e | 2015-12-03 01:23:10 +0000 | [diff] [blame] | 144 | raise error("Cannot use package as __main__ module") |
Nick Coghlan | 3f48ae3 | 2009-02-08 01:58:26 +0000 | [diff] [blame] | 145 | try: |
| 146 | pkg_main_name = mod_name + ".__main__" |
Martin Panter | 7dda421 | 2015-12-10 06:47:06 +0000 | [diff] [blame] | 147 | return _get_module_details(pkg_main_name, error) |
| 148 | except error as e: |
Martin Panter | dda5843 | 2015-12-12 06:58:55 +0000 | [diff] [blame] | 149 | if mod_name not in sys.modules: |
| 150 | raise # No module loaded; being a package is irrelevant |
Martin Panter | 657257e | 2015-12-03 01:23:10 +0000 | [diff] [blame] | 151 | raise error(("%s; %r is a package and cannot " + |
Nick Coghlan | 3f48ae3 | 2009-02-08 01:58:26 +0000 | [diff] [blame] | 152 | "be directly executed") %(e, mod_name)) |
Nick Coghlan | 720c7e2 | 2013-12-15 20:33:02 +1000 | [diff] [blame] | 153 | loader = spec.loader |
| 154 | if loader is None: |
Martin Panter | 657257e | 2015-12-03 01:23:10 +0000 | [diff] [blame] | 155 | raise error("%r is a namespace package and cannot be executed" |
Nick Coghlan | 720c7e2 | 2013-12-15 20:33:02 +1000 | [diff] [blame] | 156 | % mod_name) |
Martin Panter | 657257e | 2015-12-03 01:23:10 +0000 | [diff] [blame] | 157 | try: |
| 158 | code = loader.get_code(mod_name) |
| 159 | except ImportError as e: |
| 160 | raise error(format(e)) from e |
Thomas Wouters | a977329 | 2006-04-21 09:43:23 +0000 | [diff] [blame] | 161 | if code is None: |
Martin Panter | 657257e | 2015-12-03 01:23:10 +0000 | [diff] [blame] | 162 | raise error("No code object available for %s" % mod_name) |
Nick Coghlan | 720c7e2 | 2013-12-15 20:33:02 +1000 | [diff] [blame] | 163 | return mod_name, spec, code |
Thomas Wouters | ed03b41 | 2007-08-28 21:37:11 +0000 | [diff] [blame] | 164 | |
Martin Panter | 657257e | 2015-12-03 01:23:10 +0000 | [diff] [blame] | 165 | class _Error(Exception): |
| 166 | """Error that _run_module_as_main() should report without a traceback""" |
| 167 | |
Benjamin Peterson | f6489f9 | 2009-11-25 17:46:26 +0000 | [diff] [blame] | 168 | # XXX ncoghlan: Should this be documented and made public? |
| 169 | # (Current thoughts: don't repeat the mistake that lead to its |
| 170 | # creation when run_module() no longer met the needs of |
| 171 | # mainmodule.c, but couldn't be changed because it was public) |
Nick Coghlan | 260bd3e | 2009-11-16 06:49:25 +0000 | [diff] [blame] | 172 | def _run_module_as_main(mod_name, alter_argv=True): |
Thomas Wouters | ed03b41 | 2007-08-28 21:37:11 +0000 | [diff] [blame] | 173 | """Runs the designated module in the __main__ namespace |
| 174 | |
Benjamin Peterson | f6489f9 | 2009-11-25 17:46:26 +0000 | [diff] [blame] | 175 | Note that the executed module will have full access to the |
| 176 | __main__ namespace. If this is not desirable, the run_module() |
R. David Murray | 445448c | 2009-12-20 17:28:31 +0000 | [diff] [blame] | 177 | function should be used to run the module code in a fresh namespace. |
Benjamin Peterson | f6489f9 | 2009-11-25 17:46:26 +0000 | [diff] [blame] | 178 | |
| 179 | At the very least, these variables in __main__ will be overwritten: |
| 180 | __name__ |
Thomas Wouters | ed03b41 | 2007-08-28 21:37:11 +0000 | [diff] [blame] | 181 | __file__ |
Barry Warsaw | 28a691b | 2010-04-17 00:19:56 +0000 | [diff] [blame] | 182 | __cached__ |
Thomas Wouters | ed03b41 | 2007-08-28 21:37:11 +0000 | [diff] [blame] | 183 | __loader__ |
Benjamin Peterson | f6489f9 | 2009-11-25 17:46:26 +0000 | [diff] [blame] | 184 | __package__ |
Thomas Wouters | ed03b41 | 2007-08-28 21:37:11 +0000 | [diff] [blame] | 185 | """ |
Christian Heimes | c3f30c4 | 2008-02-22 16:37:40 +0000 | [diff] [blame] | 186 | try: |
Nick Coghlan | 260bd3e | 2009-11-16 06:49:25 +0000 | [diff] [blame] | 187 | if alter_argv or mod_name != "__main__": # i.e. -m switch |
Martin Panter | 657257e | 2015-12-03 01:23:10 +0000 | [diff] [blame] | 188 | mod_name, mod_spec, code = _get_module_details(mod_name, _Error) |
Nick Coghlan | 260bd3e | 2009-11-16 06:49:25 +0000 | [diff] [blame] | 189 | else: # i.e. directory or zipfile execution |
Martin Panter | 657257e | 2015-12-03 01:23:10 +0000 | [diff] [blame] | 190 | mod_name, mod_spec, code = _get_main_module_details(_Error) |
| 191 | except _Error as exc: |
| 192 | msg = "%s: %s" % (sys.executable, exc) |
Christian Heimes | c3f30c4 | 2008-02-22 16:37:40 +0000 | [diff] [blame] | 193 | sys.exit(msg) |
Thomas Wouters | ed03b41 | 2007-08-28 21:37:11 +0000 | [diff] [blame] | 194 | main_globals = sys.modules["__main__"].__dict__ |
Nick Coghlan | 260bd3e | 2009-11-16 06:49:25 +0000 | [diff] [blame] | 195 | if alter_argv: |
Nick Coghlan | 720c7e2 | 2013-12-15 20:33:02 +1000 | [diff] [blame] | 196 | sys.argv[0] = mod_spec.origin |
Thomas Wouters | ed03b41 | 2007-08-28 21:37:11 +0000 | [diff] [blame] | 197 | return _run_code(code, main_globals, None, |
Nick Coghlan | 720c7e2 | 2013-12-15 20:33:02 +1000 | [diff] [blame] | 198 | "__main__", mod_spec) |
Thomas Wouters | ed03b41 | 2007-08-28 21:37:11 +0000 | [diff] [blame] | 199 | |
| 200 | def run_module(mod_name, init_globals=None, |
| 201 | run_name=None, alter_sys=False): |
| 202 | """Execute a module's code without importing it |
| 203 | |
| 204 | Returns the resulting top level namespace dictionary |
| 205 | """ |
Nick Coghlan | 720c7e2 | 2013-12-15 20:33:02 +1000 | [diff] [blame] | 206 | mod_name, mod_spec, code = _get_module_details(mod_name) |
Thomas Wouters | a977329 | 2006-04-21 09:43:23 +0000 | [diff] [blame] | 207 | if run_name is None: |
| 208 | run_name = mod_name |
Thomas Wouters | ed03b41 | 2007-08-28 21:37:11 +0000 | [diff] [blame] | 209 | if alter_sys: |
Nick Coghlan | 720c7e2 | 2013-12-15 20:33:02 +1000 | [diff] [blame] | 210 | return _run_module_code(code, init_globals, run_name, mod_spec) |
Thomas Wouters | ed03b41 | 2007-08-28 21:37:11 +0000 | [diff] [blame] | 211 | else: |
| 212 | # Leave the sys module alone |
Nick Coghlan | 720c7e2 | 2013-12-15 20:33:02 +1000 | [diff] [blame] | 213 | return _run_code(code, {}, init_globals, run_name, mod_spec) |
Thomas Wouters | a977329 | 2006-04-21 09:43:23 +0000 | [diff] [blame] | 214 | |
Martin Panter | 657257e | 2015-12-03 01:23:10 +0000 | [diff] [blame] | 215 | def _get_main_module_details(error=ImportError): |
Benjamin Peterson | f6489f9 | 2009-11-25 17:46:26 +0000 | [diff] [blame] | 216 | # Helper that gives a nicer error message when attempting to |
| 217 | # execute a zipfile or directory by invoking __main__.py |
Nick Coghlan | 85e729e | 2012-07-15 18:09:52 +1000 | [diff] [blame] | 218 | # Also moves the standard __main__ out of the way so that the |
| 219 | # preexisting __loader__ entry doesn't cause issues |
Benjamin Peterson | f6489f9 | 2009-11-25 17:46:26 +0000 | [diff] [blame] | 220 | main_name = "__main__" |
Nick Coghlan | 85e729e | 2012-07-15 18:09:52 +1000 | [diff] [blame] | 221 | saved_main = sys.modules[main_name] |
| 222 | del sys.modules[main_name] |
Benjamin Peterson | f6489f9 | 2009-11-25 17:46:26 +0000 | [diff] [blame] | 223 | try: |
| 224 | return _get_module_details(main_name) |
| 225 | except ImportError as exc: |
| 226 | if main_name in str(exc): |
Martin Panter | 657257e | 2015-12-03 01:23:10 +0000 | [diff] [blame] | 227 | raise error("can't find %r module in %r" % |
Nick Coghlan | 85e729e | 2012-07-15 18:09:52 +1000 | [diff] [blame] | 228 | (main_name, sys.path[0])) from exc |
Benjamin Peterson | f6489f9 | 2009-11-25 17:46:26 +0000 | [diff] [blame] | 229 | raise |
Nick Coghlan | 85e729e | 2012-07-15 18:09:52 +1000 | [diff] [blame] | 230 | finally: |
| 231 | sys.modules[main_name] = saved_main |
Benjamin Peterson | f6489f9 | 2009-11-25 17:46:26 +0000 | [diff] [blame] | 232 | |
Thomas Wouters | a977329 | 2006-04-21 09:43:23 +0000 | [diff] [blame] | 233 | |
Nick Coghlan | 85e729e | 2012-07-15 18:09:52 +1000 | [diff] [blame] | 234 | def _get_code_from_file(run_name, fname): |
Nick Coghlan | 260bd3e | 2009-11-16 06:49:25 +0000 | [diff] [blame] | 235 | # Check for a compiled file first |
Maor Kleinberger | 0911ea5 | 2020-03-08 22:43:17 +0200 | [diff] [blame] | 236 | decoded_path = os.path.abspath(os.fsdecode(fname)) |
| 237 | with io.open_code(decoded_path) as f: |
Nick Coghlan | 260bd3e | 2009-11-16 06:49:25 +0000 | [diff] [blame] | 238 | code = read_code(f) |
| 239 | if code is None: |
| 240 | # That didn't work, so try it as normal source code |
Maor Kleinberger | 0911ea5 | 2020-03-08 22:43:17 +0200 | [diff] [blame] | 241 | with io.open_code(decoded_path) as f: |
Nick Coghlan | 260bd3e | 2009-11-16 06:49:25 +0000 | [diff] [blame] | 242 | code = compile(f.read(), fname, 'exec') |
Nick Coghlan | 720c7e2 | 2013-12-15 20:33:02 +1000 | [diff] [blame] | 243 | return code, fname |
Nick Coghlan | 260bd3e | 2009-11-16 06:49:25 +0000 | [diff] [blame] | 244 | |
| 245 | def run_path(path_name, init_globals=None, run_name=None): |
| 246 | """Execute code located at the specified filesystem location |
| 247 | |
| 248 | Returns the resulting top level namespace dictionary |
| 249 | |
| 250 | The file path may refer directly to a Python script (i.e. |
| 251 | one that could be directly executed with execfile) or else |
| 252 | it may refer to a zipfile or directory containing a top |
| 253 | level __main__.py script. |
| 254 | """ |
| 255 | if run_name is None: |
| 256 | run_name = "<run_path>" |
Nick Coghlan | 761bb11 | 2012-07-14 23:59:22 +1000 | [diff] [blame] | 257 | pkg_name = run_name.rpartition(".")[0] |
Nick Coghlan | 85e729e | 2012-07-15 18:09:52 +1000 | [diff] [blame] | 258 | importer = get_importer(path_name) |
Brett Cannon | e4f41de | 2013-06-16 13:13:40 -0400 | [diff] [blame] | 259 | # Trying to avoid importing imp so as to not consume the deprecation warning. |
| 260 | is_NullImporter = False |
| 261 | if type(importer).__module__ == 'imp': |
| 262 | if type(importer).__name__ == 'NullImporter': |
| 263 | is_NullImporter = True |
| 264 | if isinstance(importer, type(None)) or is_NullImporter: |
Nick Coghlan | 260bd3e | 2009-11-16 06:49:25 +0000 | [diff] [blame] | 265 | # Not a valid sys.path entry, so run the code directly |
| 266 | # execfile() doesn't help as we want to allow compiled files |
Nick Coghlan | 720c7e2 | 2013-12-15 20:33:02 +1000 | [diff] [blame] | 267 | code, fname = _get_code_from_file(run_name, path_name) |
| 268 | return _run_module_code(code, init_globals, run_name, |
| 269 | pkg_name=pkg_name, script_name=fname) |
Nick Coghlan | 260bd3e | 2009-11-16 06:49:25 +0000 | [diff] [blame] | 270 | else: |
Brett Cannon | fdcdd9e | 2016-07-08 11:00:00 -0700 | [diff] [blame] | 271 | # Finder is defined for path, so add it to |
Nick Coghlan | 260bd3e | 2009-11-16 06:49:25 +0000 | [diff] [blame] | 272 | # the start of sys.path |
| 273 | sys.path.insert(0, path_name) |
| 274 | try: |
| 275 | # Here's where things are a little different from the run_module |
| 276 | # case. There, we only had to replace the module in sys while the |
| 277 | # code was running and doing so was somewhat optional. Here, we |
| 278 | # have no choice and we have to remove it even while we read the |
| 279 | # code. If we don't do this, a __loader__ attribute in the |
| 280 | # existing __main__ module may prevent location of the new module. |
Nick Coghlan | 720c7e2 | 2013-12-15 20:33:02 +1000 | [diff] [blame] | 281 | mod_name, mod_spec, code = _get_main_module_details() |
Nick Coghlan | 260bd3e | 2009-11-16 06:49:25 +0000 | [diff] [blame] | 282 | with _TempModule(run_name) as temp_module, \ |
| 283 | _ModifiedArgv0(path_name): |
| 284 | mod_globals = temp_module.module.__dict__ |
| 285 | return _run_code(code, mod_globals, init_globals, |
Nick Coghlan | 720c7e2 | 2013-12-15 20:33:02 +1000 | [diff] [blame] | 286 | run_name, mod_spec, pkg_name).copy() |
Nick Coghlan | 260bd3e | 2009-11-16 06:49:25 +0000 | [diff] [blame] | 287 | finally: |
| 288 | try: |
| 289 | sys.path.remove(path_name) |
| 290 | except ValueError: |
| 291 | pass |
| 292 | |
| 293 | |
Thomas Wouters | a977329 | 2006-04-21 09:43:23 +0000 | [diff] [blame] | 294 | if __name__ == "__main__": |
| 295 | # Run the module specified as the next command line argument |
| 296 | if len(sys.argv) < 2: |
Guido van Rossum | be19ed7 | 2007-02-09 05:37:30 +0000 | [diff] [blame] | 297 | print("No module specified for execution", file=sys.stderr) |
Thomas Wouters | a977329 | 2006-04-21 09:43:23 +0000 | [diff] [blame] | 298 | else: |
| 299 | del sys.argv[0] # Make the requested module sys.argv[0] |
Thomas Wouters | ed03b41 | 2007-08-28 21:37:11 +0000 | [diff] [blame] | 300 | _run_module_as_main(sys.argv[0]) |