2009-01-17 20:24:28 -04:00
|
|
|
"""Core implementation of import.
|
|
|
|
|
|
|
|
This module is NOT meant to be directly imported! It has been designed such
|
|
|
|
that it can be bootstrapped into Python as the implementation of import. As
|
|
|
|
such it requires the injection of specific modules and attributes in order to
|
|
|
|
work. One should use importlib as the public-facing version of this module.
|
|
|
|
|
|
|
|
"""
|
2012-04-25 05:54:48 -03:00
|
|
|
#
|
2019-02-13 19:21:17 -04:00
|
|
|
# IMPORTANT: Whenever making changes to this module, be sure to run a top-level
|
|
|
|
# `make regen-importlib` followed by `make` in order to get the frozen version
|
|
|
|
# of the module updated. Not doing so will result in the Makefile to fail for
|
2012-04-25 05:54:48 -03:00
|
|
|
# all others who don't have a ./python around to freeze the module
|
|
|
|
# in the early stages of compilation.
|
|
|
|
#
|
2009-01-17 20:24:28 -04:00
|
|
|
|
2012-04-20 22:19:53 -03:00
|
|
|
# See importlib._setup() for what is injected into the global namespace.
|
|
|
|
|
2009-01-17 20:24:28 -04:00
|
|
|
# When editing this code be aware that code executed at import time CANNOT
|
|
|
|
# reference any injected objects! This includes not only global code but also
|
|
|
|
# anything specified at the class level.
|
|
|
|
|
2020-12-04 19:39:21 -04:00
|
|
|
def _object_name(obj):
|
|
|
|
try:
|
|
|
|
return obj.__qualname__
|
|
|
|
except AttributeError:
|
|
|
|
return type(obj).__qualname__
|
|
|
|
|
2009-03-12 19:28:55 -03:00
|
|
|
# Bootstrap-related code ######################################################
|
|
|
|
|
2020-11-19 08:43:43 -04:00
|
|
|
# Modules injected manually by _setup()
|
|
|
|
_thread = None
|
|
|
|
_warnings = None
|
|
|
|
_weakref = None
|
|
|
|
|
|
|
|
# Import done by _install_external_importers()
|
2015-05-16 00:54:59 -03:00
|
|
|
_bootstrap_external = None
|
|
|
|
|
2020-11-19 08:43:43 -04:00
|
|
|
|
2009-03-12 19:07:17 -03:00
|
|
|
def _wrap(new, old):
|
2012-07-07 08:16:44 -03:00
|
|
|
"""Simple substitute for functools.update_wrapper."""
|
2011-12-15 00:53:13 -04:00
|
|
|
for replace in ['__module__', '__name__', '__qualname__', '__doc__']:
|
2012-02-08 19:44:14 -04:00
|
|
|
if hasattr(old, replace):
|
|
|
|
setattr(new, replace, getattr(old, replace))
|
2009-02-06 22:13:28 -04:00
|
|
|
new.__dict__.update(old.__dict__)
|
|
|
|
|
|
|
|
|
2013-11-22 12:05:39 -04:00
|
|
|
def _new_module(name):
|
|
|
|
return type(sys)(name)
|
|
|
|
|
|
|
|
|
2012-05-17 13:55:59 -03:00
|
|
|
# Module-level locking ########################################################
|
|
|
|
|
|
|
|
# A dict mapping module names to weakrefs of _ModuleLock instances
|
2017-07-10 17:52:32 -03:00
|
|
|
# Dictionary protected by the global import lock
|
2012-05-17 13:55:59 -03:00
|
|
|
_module_locks = {}
|
2023-01-20 20:00:39 -04:00
|
|
|
|
|
|
|
# A dict mapping thread IDs to lists of _ModuleLock instances. This maps a
|
|
|
|
# thread to the module locks it is blocking on acquiring. The values are
|
|
|
|
# lists because a single thread could perform a re-entrant import and be "in
|
|
|
|
# the process" of blocking on locks for more than one module. A thread can
|
|
|
|
# be "in the process" because a thread cannot actually block on acquiring
|
|
|
|
# more than one lock but it can have set up bookkeeping that reflects that
|
|
|
|
# it intends to block on acquiring more than one lock.
|
2012-05-17 13:55:59 -03:00
|
|
|
_blocking_on = {}
|
|
|
|
|
|
|
|
|
2023-01-20 20:00:39 -04:00
|
|
|
class _BlockingOnManager:
|
|
|
|
"""A context manager responsible to updating ``_blocking_on``."""
|
|
|
|
def __init__(self, thread_id, lock):
|
|
|
|
self.thread_id = thread_id
|
|
|
|
self.lock = lock
|
|
|
|
|
|
|
|
def __enter__(self):
|
|
|
|
"""Mark the running thread as waiting for self.lock. via _blocking_on."""
|
|
|
|
# Interactions with _blocking_on are *not* protected by the global
|
|
|
|
# import lock here because each thread only touches the state that it
|
|
|
|
# owns (state keyed on its thread id). The global import lock is
|
|
|
|
# re-entrant (i.e., a single thread may take it more than once) so it
|
|
|
|
# wouldn't help us be correct in the face of re-entrancy either.
|
|
|
|
|
|
|
|
self.blocked_on = _blocking_on.setdefault(self.thread_id, [])
|
|
|
|
self.blocked_on.append(self.lock)
|
|
|
|
|
|
|
|
def __exit__(self, *args, **kwargs):
|
|
|
|
"""Remove self.lock from this thread's _blocking_on list."""
|
|
|
|
self.blocked_on.remove(self.lock)
|
|
|
|
|
|
|
|
|
2012-05-17 13:55:59 -03:00
|
|
|
class _DeadlockError(RuntimeError):
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
2023-01-20 20:00:39 -04:00
|
|
|
|
|
|
|
def _has_deadlocked(target_id, *, seen_ids, candidate_ids, blocking_on):
|
|
|
|
"""Check if 'target_id' is holding the same lock as another thread(s).
|
|
|
|
|
|
|
|
The search within 'blocking_on' starts with the threads listed in
|
|
|
|
'candidate_ids'. 'seen_ids' contains any threads that are considered
|
|
|
|
already traversed in the search.
|
|
|
|
|
|
|
|
Keyword arguments:
|
|
|
|
target_id -- The thread id to try to reach.
|
|
|
|
seen_ids -- A set of threads that have already been visited.
|
|
|
|
candidate_ids -- The thread ids from which to begin.
|
|
|
|
blocking_on -- A dict representing the thread/blocking-on graph. This may
|
|
|
|
be the same object as the global '_blocking_on' but it is
|
|
|
|
a parameter to reduce the impact that global mutable
|
|
|
|
state has on the result of this function.
|
|
|
|
"""
|
|
|
|
if target_id in candidate_ids:
|
|
|
|
# If we have already reached the target_id, we're done - signal that it
|
|
|
|
# is reachable.
|
|
|
|
return True
|
|
|
|
|
|
|
|
# Otherwise, try to reach the target_id from each of the given candidate_ids.
|
|
|
|
for tid in candidate_ids:
|
|
|
|
if not (candidate_blocking_on := blocking_on.get(tid)):
|
|
|
|
# There are no edges out from this node, skip it.
|
|
|
|
continue
|
|
|
|
elif tid in seen_ids:
|
|
|
|
# bpo 38091: the chain of tid's we encounter here eventually leads
|
|
|
|
# to a fixed point or a cycle, but does not reach target_id.
|
|
|
|
# This means we would not actually deadlock. This can happen if
|
|
|
|
# other threads are at the beginning of acquire() below.
|
|
|
|
return False
|
|
|
|
seen_ids.add(tid)
|
|
|
|
|
|
|
|
# Follow the edges out from this thread.
|
|
|
|
edges = [lock.owner for lock in candidate_blocking_on]
|
|
|
|
if _has_deadlocked(target_id, seen_ids=seen_ids, candidate_ids=edges,
|
|
|
|
blocking_on=blocking_on):
|
|
|
|
return True
|
|
|
|
|
|
|
|
return False
|
|
|
|
|
|
|
|
|
2012-05-17 13:55:59 -03:00
|
|
|
class _ModuleLock:
|
|
|
|
"""A recursive lock implementation which is able to detect deadlocks
|
|
|
|
(e.g. thread 1 trying to take locks A then B, and thread 2 trying to
|
|
|
|
take locks B then A).
|
|
|
|
"""
|
|
|
|
|
|
|
|
def __init__(self, name):
|
2023-01-20 20:00:39 -04:00
|
|
|
# Create an RLock for protecting the import process for the
|
|
|
|
# corresponding module. Since it is an RLock, a single thread will be
|
|
|
|
# able to take it more than once. This is necessary to support
|
|
|
|
# re-entrancy in the import system that arises from (at least) signal
|
|
|
|
# handlers and the garbage collector. Consider the case of:
|
|
|
|
#
|
|
|
|
# import foo
|
|
|
|
# -> ...
|
|
|
|
# -> importlib._bootstrap._ModuleLock.acquire
|
|
|
|
# -> ...
|
|
|
|
# -> <garbage collector>
|
|
|
|
# -> __del__
|
|
|
|
# -> import foo
|
|
|
|
# -> ...
|
|
|
|
# -> importlib._bootstrap._ModuleLock.acquire
|
|
|
|
# -> _BlockingOnManager.__enter__
|
|
|
|
#
|
|
|
|
# If a different thread than the running one holds the lock then the
|
|
|
|
# thread will have to block on taking the lock, which is what we want
|
|
|
|
# for thread safety.
|
|
|
|
self.lock = _thread.RLock()
|
2012-05-17 13:55:59 -03:00
|
|
|
self.wakeup = _thread.allocate_lock()
|
2023-01-20 20:00:39 -04:00
|
|
|
|
|
|
|
# The name of the module for which this is a lock.
|
2012-05-17 13:55:59 -03:00
|
|
|
self.name = name
|
2023-01-20 20:00:39 -04:00
|
|
|
|
|
|
|
# Can end up being set to None if this lock is not owned by any thread
|
|
|
|
# or the thread identifier for the owning thread.
|
2012-05-17 13:55:59 -03:00
|
|
|
self.owner = None
|
2023-01-20 20:00:39 -04:00
|
|
|
|
|
|
|
# Represent the number of times the owning thread has acquired this lock
|
|
|
|
# via a list of True. This supports RLock-like ("re-entrant lock")
|
|
|
|
# behavior, necessary in case a single thread is following a circular
|
|
|
|
# import dependency and needs to take the lock for a single module
|
|
|
|
# more than once.
|
|
|
|
#
|
|
|
|
# Counts are represented as a list of True because list.append(True)
|
|
|
|
# and list.pop() are both atomic and thread-safe in CPython and it's hard
|
|
|
|
# to find another primitive with the same properties.
|
|
|
|
self.count = []
|
|
|
|
|
|
|
|
# This is a count of the number of threads that are blocking on
|
|
|
|
# self.wakeup.acquire() awaiting to get their turn holding this module
|
|
|
|
# lock. When the module lock is released, if this is greater than
|
|
|
|
# zero, it is decremented and `self.wakeup` is released one time. The
|
|
|
|
# intent is that this will let one other thread make more progress on
|
|
|
|
# acquiring this module lock. This repeats until all the threads have
|
|
|
|
# gotten a turn.
|
|
|
|
#
|
|
|
|
# This is incremented in self.acquire() when a thread notices it is
|
|
|
|
# going to have to wait for another thread to finish.
|
|
|
|
#
|
|
|
|
# See the comment above count for explanation of the representation.
|
|
|
|
self.waiters = []
|
2012-05-17 13:55:59 -03:00
|
|
|
|
|
|
|
def has_deadlock(self):
|
2023-01-20 20:00:39 -04:00
|
|
|
# To avoid deadlocks for concurrent or re-entrant circular imports,
|
|
|
|
# look at _blocking_on to see if any threads are blocking
|
|
|
|
# on getting the import lock for any module for which the import lock
|
|
|
|
# is held by this thread.
|
|
|
|
return _has_deadlocked(
|
|
|
|
# Try to find this thread.
|
|
|
|
target_id=_thread.get_ident(),
|
|
|
|
seen_ids=set(),
|
|
|
|
# Start from the thread that holds the import lock for this
|
|
|
|
# module.
|
|
|
|
candidate_ids=[self.owner],
|
|
|
|
# Use the global "blocking on" state.
|
|
|
|
blocking_on=_blocking_on,
|
|
|
|
)
|
2012-05-17 13:55:59 -03:00
|
|
|
|
|
|
|
def acquire(self):
|
|
|
|
"""
|
|
|
|
Acquire the module lock. If a potential deadlock is detected,
|
|
|
|
a _DeadlockError is raised.
|
|
|
|
Otherwise, the lock is always acquired and True is returned.
|
|
|
|
"""
|
|
|
|
tid = _thread.get_ident()
|
2023-01-20 20:00:39 -04:00
|
|
|
with _BlockingOnManager(tid, self):
|
2012-05-17 13:55:59 -03:00
|
|
|
while True:
|
2023-01-20 20:00:39 -04:00
|
|
|
# Protect interaction with state on self with a per-module
|
|
|
|
# lock. This makes it safe for more than one thread to try to
|
|
|
|
# acquire the lock for a single module at the same time.
|
2012-05-17 13:55:59 -03:00
|
|
|
with self.lock:
|
2023-01-20 20:00:39 -04:00
|
|
|
if self.count == [] or self.owner == tid:
|
|
|
|
# If the lock for this module is unowned then we can
|
|
|
|
# take the lock immediately and succeed. If the lock
|
|
|
|
# for this module is owned by the running thread then
|
|
|
|
# we can also allow the acquire to succeed. This
|
|
|
|
# supports circular imports (thread T imports module A
|
|
|
|
# which imports module B which imports module A).
|
2012-05-17 13:55:59 -03:00
|
|
|
self.owner = tid
|
2023-01-20 20:00:39 -04:00
|
|
|
self.count.append(True)
|
2012-05-17 13:55:59 -03:00
|
|
|
return True
|
2023-01-20 20:00:39 -04:00
|
|
|
|
|
|
|
# At this point we know the lock is held (because count !=
|
|
|
|
# 0) by another thread (because owner != tid). We'll have
|
|
|
|
# to get in line to take the module lock.
|
|
|
|
|
|
|
|
# But first, check to see if this thread would create a
|
|
|
|
# deadlock by acquiring this module lock. If it would
|
|
|
|
# then just stop with an error.
|
|
|
|
#
|
|
|
|
# It's not clear who is expected to handle this error.
|
|
|
|
# There is one handler in _lock_unlock_module but many
|
|
|
|
# times this method is called when entering the context
|
|
|
|
# manager _ModuleLockManager instead - so _DeadlockError
|
|
|
|
# will just propagate up to application code.
|
|
|
|
#
|
|
|
|
# This seems to be more than just a hypothetical -
|
|
|
|
# https://stackoverflow.com/questions/59509154
|
|
|
|
# https://github.com/encode/django-rest-framework/issues/7078
|
2012-05-17 13:55:59 -03:00
|
|
|
if self.has_deadlock():
|
2023-01-20 20:00:39 -04:00
|
|
|
raise _DeadlockError(f'deadlock detected by {self!r}')
|
|
|
|
|
|
|
|
# Check to see if we're going to be able to acquire the
|
|
|
|
# lock. If we are going to have to wait then increment
|
|
|
|
# the waiters so `self.release` will know to unblock us
|
|
|
|
# later on. We do this part non-blockingly so we don't
|
|
|
|
# get stuck here before we increment waiters. We have
|
|
|
|
# this extra acquire call (in addition to the one below,
|
|
|
|
# outside the self.lock context manager) to make sure
|
|
|
|
# self.wakeup is held when the next acquire is called (so
|
|
|
|
# we block). This is probably needlessly complex and we
|
|
|
|
# should just take self.wakeup in the return codepath
|
|
|
|
# above.
|
2012-05-17 13:55:59 -03:00
|
|
|
if self.wakeup.acquire(False):
|
2023-01-20 20:00:39 -04:00
|
|
|
self.waiters.append(None)
|
|
|
|
|
|
|
|
# Now take the lock in a blocking fashion. This won't
|
|
|
|
# complete until the thread holding this lock
|
|
|
|
# (self.owner) calls self.release.
|
2012-05-17 13:55:59 -03:00
|
|
|
self.wakeup.acquire()
|
2023-01-20 20:00:39 -04:00
|
|
|
|
|
|
|
# Taking the lock has served its purpose (making us wait), so we can
|
|
|
|
# give it up now. We'll take it w/o blocking again on the
|
|
|
|
# next iteration around this 'while' loop.
|
2012-05-17 13:55:59 -03:00
|
|
|
self.wakeup.release()
|
|
|
|
|
|
|
|
def release(self):
|
|
|
|
tid = _thread.get_ident()
|
|
|
|
with self.lock:
|
|
|
|
if self.owner != tid:
|
2013-07-12 12:04:23 -03:00
|
|
|
raise RuntimeError('cannot release un-acquired lock')
|
2023-01-20 20:00:39 -04:00
|
|
|
assert len(self.count) > 0
|
|
|
|
self.count.pop()
|
|
|
|
if not len(self.count):
|
2012-05-17 13:55:59 -03:00
|
|
|
self.owner = None
|
2023-01-20 20:00:39 -04:00
|
|
|
if len(self.waiters) > 0:
|
|
|
|
self.waiters.pop()
|
2012-05-17 13:55:59 -03:00
|
|
|
self.wakeup.release()
|
|
|
|
|
|
|
|
def __repr__(self):
|
2022-10-06 20:43:16 -03:00
|
|
|
return f'_ModuleLock({self.name!r}) at {id(self)}'
|
2012-05-17 13:55:59 -03:00
|
|
|
|
|
|
|
|
|
|
|
class _DummyModuleLock:
|
|
|
|
"""A simple _ModuleLock equivalent for Python builds without
|
|
|
|
multi-threading support."""
|
|
|
|
|
|
|
|
def __init__(self, name):
|
|
|
|
self.name = name
|
|
|
|
self.count = 0
|
|
|
|
|
|
|
|
def acquire(self):
|
|
|
|
self.count += 1
|
|
|
|
return True
|
|
|
|
|
|
|
|
def release(self):
|
|
|
|
if self.count == 0:
|
2013-07-12 12:04:23 -03:00
|
|
|
raise RuntimeError('cannot release un-acquired lock')
|
2012-05-17 13:55:59 -03:00
|
|
|
self.count -= 1
|
|
|
|
|
|
|
|
def __repr__(self):
|
2022-10-06 20:43:16 -03:00
|
|
|
return f'_DummyModuleLock({self.name!r}) at {id(self)}'
|
2012-05-17 13:55:59 -03:00
|
|
|
|
|
|
|
|
2013-11-22 12:05:39 -04:00
|
|
|
class _ModuleLockManager:
|
|
|
|
|
|
|
|
def __init__(self, name):
|
|
|
|
self._name = name
|
|
|
|
self._lock = None
|
|
|
|
|
|
|
|
def __enter__(self):
|
2017-07-10 17:52:32 -03:00
|
|
|
self._lock = _get_module_lock(self._name)
|
2013-11-22 12:05:39 -04:00
|
|
|
self._lock.acquire()
|
|
|
|
|
|
|
|
def __exit__(self, *args, **kwargs):
|
|
|
|
self._lock.release()
|
|
|
|
|
|
|
|
|
2012-05-17 13:55:59 -03:00
|
|
|
# The following two functions are for consumption by Python/import.c.
|
|
|
|
|
|
|
|
def _get_module_lock(name):
|
|
|
|
"""Get or create the module lock for a given module name.
|
|
|
|
|
2017-07-10 17:52:32 -03:00
|
|
|
Acquire/release internally the global import lock to protect
|
|
|
|
_module_locks."""
|
|
|
|
|
|
|
|
_imp.acquire_lock()
|
2012-08-27 19:24:52 -03:00
|
|
|
try:
|
2017-07-10 17:52:32 -03:00
|
|
|
try:
|
|
|
|
lock = _module_locks[name]()
|
|
|
|
except KeyError:
|
|
|
|
lock = None
|
|
|
|
|
|
|
|
if lock is None:
|
|
|
|
if _thread is None:
|
|
|
|
lock = _DummyModuleLock(name)
|
|
|
|
else:
|
|
|
|
lock = _ModuleLock(name)
|
2017-08-09 08:29:12 -03:00
|
|
|
|
|
|
|
def cb(ref, name=name):
|
|
|
|
_imp.acquire_lock()
|
|
|
|
try:
|
|
|
|
# bpo-31070: Check if another thread created a new lock
|
|
|
|
# after the previous lock was destroyed
|
|
|
|
# but before the weakref callback was called.
|
|
|
|
if _module_locks.get(name) is ref:
|
|
|
|
del _module_locks[name]
|
|
|
|
finally:
|
|
|
|
_imp.release_lock()
|
|
|
|
|
2017-07-10 17:52:32 -03:00
|
|
|
_module_locks[name] = _weakref.ref(lock, cb)
|
|
|
|
finally:
|
|
|
|
_imp.release_lock()
|
|
|
|
|
2012-05-17 13:55:59 -03:00
|
|
|
return lock
|
|
|
|
|
2017-07-10 17:52:32 -03:00
|
|
|
|
2012-05-17 13:55:59 -03:00
|
|
|
def _lock_unlock_module(name):
|
2017-07-10 17:52:32 -03:00
|
|
|
"""Acquires then releases the module lock for a given module name.
|
|
|
|
|
2012-05-17 13:55:59 -03:00
|
|
|
This is used to ensure a module is completely initialized, in the
|
|
|
|
event it is being imported by another thread.
|
2017-07-10 17:52:32 -03:00
|
|
|
"""
|
2012-05-17 13:55:59 -03:00
|
|
|
lock = _get_module_lock(name)
|
|
|
|
try:
|
|
|
|
lock.acquire()
|
|
|
|
except _DeadlockError:
|
|
|
|
# Concurrent circular import, we'll accept a partially initialized
|
|
|
|
# module object.
|
|
|
|
pass
|
|
|
|
else:
|
|
|
|
lock.release()
|
|
|
|
|
2012-07-31 08:14:18 -03:00
|
|
|
# Frame stripping magic ###############################################
|
|
|
|
def _call_with_frames_removed(f, *args, **kwds):
|
|
|
|
"""remove_importlib_frames in import.c will always remove sequences
|
|
|
|
of importlib frames that end with a call to this function
|
|
|
|
|
|
|
|
Use it instead of a normal call in places where including the importlib
|
|
|
|
frames introduces unwanted noise into the traceback (e.g. when executing
|
|
|
|
module code)
|
|
|
|
"""
|
|
|
|
return f(*args, **kwds)
|
|
|
|
|
|
|
|
|
2013-04-01 14:10:51 -03:00
|
|
|
def _verbose_message(message, *args, verbosity=1):
|
2012-04-14 15:10:13 -03:00
|
|
|
"""Print the message to stderr if -v/PYTHONVERBOSE is turned on."""
|
2013-04-01 14:10:51 -03:00
|
|
|
if sys.flags.verbose >= verbosity:
|
2012-04-15 16:21:32 -03:00
|
|
|
if not message.startswith(('#', 'import ')):
|
2012-04-14 15:10:13 -03:00
|
|
|
message = '# ' + message
|
|
|
|
print(message.format(*args), file=sys.stderr)
|
|
|
|
|
2012-02-19 20:48:16 -04:00
|
|
|
|
2009-03-14 22:41:33 -03:00
|
|
|
def _requires_builtin(fxn):
|
|
|
|
"""Decorator to verify the named module is built-in."""
|
2012-01-16 12:46:22 -04:00
|
|
|
def _requires_builtin_wrapper(self, fullname):
|
2009-03-14 22:41:33 -03:00
|
|
|
if fullname not in sys.builtin_module_names:
|
2022-10-06 20:43:16 -03:00
|
|
|
raise ImportError(f'{fullname!r} is not a built-in module',
|
2012-04-12 22:09:01 -03:00
|
|
|
name=fullname)
|
2009-03-14 22:41:33 -03:00
|
|
|
return fxn(self, fullname)
|
2012-01-16 12:46:22 -04:00
|
|
|
_wrap(_requires_builtin_wrapper, fxn)
|
|
|
|
return _requires_builtin_wrapper
|
2009-03-14 22:41:33 -03:00
|
|
|
|
|
|
|
|
2009-03-14 23:20:16 -03:00
|
|
|
def _requires_frozen(fxn):
|
|
|
|
"""Decorator to verify the named module is frozen."""
|
2012-01-16 12:46:22 -04:00
|
|
|
def _requires_frozen_wrapper(self, fullname):
|
2012-04-15 17:08:47 -03:00
|
|
|
if not _imp.is_frozen(fullname):
|
2022-10-06 20:43:16 -03:00
|
|
|
raise ImportError(f'{fullname!r} is not a frozen module',
|
2012-04-12 22:09:01 -03:00
|
|
|
name=fullname)
|
2009-03-14 23:20:16 -03:00
|
|
|
return fxn(self, fullname)
|
2012-01-16 12:46:22 -04:00
|
|
|
_wrap(_requires_frozen_wrapper, fxn)
|
|
|
|
return _requires_frozen_wrapper
|
2009-03-14 23:20:16 -03:00
|
|
|
|
|
|
|
|
2014-05-30 17:28:00 -03:00
|
|
|
# Typically used by loader classes as a method replacement.
|
|
|
|
def _load_module_shim(self, fullname):
|
2014-01-06 23:49:04 -04:00
|
|
|
"""Load the specified module into sys.modules and return it.
|
|
|
|
|
2021-03-24 12:26:56 -03:00
|
|
|
This method is deprecated. Use loader.exec_module() instead.
|
2014-01-06 23:49:04 -04:00
|
|
|
|
|
|
|
"""
|
2020-12-04 19:39:21 -04:00
|
|
|
msg = ("the load_module() method is deprecated and slated for removal in "
|
|
|
|
"Python 3.12; use exec_module() instead")
|
|
|
|
_warnings.warn(msg, DeprecationWarning)
|
2014-05-30 17:28:00 -03:00
|
|
|
spec = spec_from_loader(fullname, self)
|
2013-11-22 12:05:39 -04:00
|
|
|
if fullname in sys.modules:
|
|
|
|
module = sys.modules[fullname]
|
2014-05-30 15:55:29 -03:00
|
|
|
_exec(spec, module)
|
2013-11-22 12:05:39 -04:00
|
|
|
return sys.modules[fullname]
|
|
|
|
else:
|
2014-05-30 15:55:29 -03:00
|
|
|
return _load(spec)
|
2013-11-22 12:05:39 -04:00
|
|
|
|
|
|
|
# Module specifications #######################################################
|
|
|
|
|
|
|
|
def _module_repr(module):
|
2021-03-24 12:26:56 -03:00
|
|
|
"""The implementation of ModuleType.__repr__()."""
|
2013-11-22 12:05:39 -04:00
|
|
|
loader = getattr(module, '__loader__', None)
|
2021-03-24 12:26:56 -03:00
|
|
|
if spec := getattr(module, "__spec__", None):
|
|
|
|
return _module_repr_from_spec(spec)
|
|
|
|
# Fall through to a catch-all which always succeeds.
|
2013-11-22 12:05:39 -04:00
|
|
|
try:
|
|
|
|
name = module.__name__
|
|
|
|
except AttributeError:
|
|
|
|
name = '?'
|
|
|
|
try:
|
|
|
|
filename = module.__file__
|
|
|
|
except AttributeError:
|
|
|
|
if loader is None:
|
2022-10-06 20:43:16 -03:00
|
|
|
return f'<module {name!r}>'
|
2013-11-22 12:05:39 -04:00
|
|
|
else:
|
2022-10-06 20:43:16 -03:00
|
|
|
return f'<module {name!r} ({loader!r})>'
|
2013-11-22 12:05:39 -04:00
|
|
|
else:
|
2022-10-06 20:43:16 -03:00
|
|
|
return f'<module {name!r} from {filename!r}>'
|
2013-11-22 12:05:39 -04:00
|
|
|
|
|
|
|
|
|
|
|
class ModuleSpec:
|
|
|
|
"""The specification for a module, used for loading.
|
|
|
|
|
|
|
|
A module's spec is the source for information about the module. For
|
|
|
|
data associated with the module, including source, use the spec's
|
|
|
|
loader.
|
|
|
|
|
|
|
|
`name` is the absolute name of the module. `loader` is the loader
|
|
|
|
to use when loading the module. `parent` is the name of the
|
|
|
|
package the module is in. The parent is derived from the name.
|
|
|
|
|
|
|
|
`is_package` determines if the module is considered a package or
|
|
|
|
not. On modules this is reflected by the `__path__` attribute.
|
|
|
|
|
|
|
|
`origin` is the specific location used by the loader from which to
|
|
|
|
load the module, if that information is available. When filename is
|
|
|
|
set, origin will match.
|
|
|
|
|
|
|
|
`has_location` indicates that a spec's "origin" reflects a location.
|
|
|
|
When this is True, `__file__` attribute of the module is set.
|
|
|
|
|
|
|
|
`cached` is the location of the cached bytecode file, if any. It
|
|
|
|
corresponds to the `__cached__` attribute.
|
|
|
|
|
|
|
|
`submodule_search_locations` is the sequence of path entries to
|
|
|
|
search when importing submodules. If set, is_package should be
|
|
|
|
True--and False otherwise.
|
|
|
|
|
|
|
|
Packages are simply modules that (may) have submodules. If a spec
|
|
|
|
has a non-None value in `submodule_search_locations`, the import
|
|
|
|
system will consider modules loaded from the spec as packages.
|
|
|
|
|
|
|
|
Only finders (see importlib.abc.MetaPathFinder and
|
|
|
|
importlib.abc.PathEntryFinder) should modify ModuleSpec instances.
|
|
|
|
|
|
|
|
"""
|
|
|
|
|
|
|
|
def __init__(self, name, loader, *, origin=None, loader_state=None,
|
|
|
|
is_package=None):
|
|
|
|
self.name = name
|
|
|
|
self.loader = loader
|
|
|
|
self.origin = origin
|
|
|
|
self.loader_state = loader_state
|
|
|
|
self.submodule_search_locations = [] if is_package else None
|
2021-07-24 19:44:46 -03:00
|
|
|
self._uninitialized_submodules = []
|
2013-11-22 12:05:39 -04:00
|
|
|
|
|
|
|
# file-location attributes
|
|
|
|
self._set_fileattr = False
|
|
|
|
self._cached = None
|
|
|
|
|
|
|
|
def __repr__(self):
|
2022-10-06 20:43:16 -03:00
|
|
|
args = [f'name={self.name!r}', f'loader={self.loader!r}']
|
2013-11-22 12:05:39 -04:00
|
|
|
if self.origin is not None:
|
2022-10-06 20:43:16 -03:00
|
|
|
args.append(f'origin={self.origin!r}')
|
2013-11-22 12:05:39 -04:00
|
|
|
if self.submodule_search_locations is not None:
|
2022-10-06 20:43:16 -03:00
|
|
|
args.append(f'submodule_search_locations={self.submodule_search_locations}')
|
|
|
|
return f'{self.__class__.__name__}({", ".join(args)})'
|
2013-11-22 12:05:39 -04:00
|
|
|
|
|
|
|
def __eq__(self, other):
|
|
|
|
smsl = self.submodule_search_locations
|
|
|
|
try:
|
|
|
|
return (self.name == other.name and
|
|
|
|
self.loader == other.loader and
|
|
|
|
self.origin == other.origin and
|
|
|
|
smsl == other.submodule_search_locations and
|
|
|
|
self.cached == other.cached and
|
|
|
|
self.has_location == other.has_location)
|
|
|
|
except AttributeError:
|
2019-08-08 02:42:54 -03:00
|
|
|
return NotImplemented
|
2013-11-22 12:05:39 -04:00
|
|
|
|
|
|
|
@property
|
|
|
|
def cached(self):
|
|
|
|
if self._cached is None:
|
|
|
|
if self.origin is not None and self._set_fileattr:
|
2015-05-16 00:54:59 -03:00
|
|
|
if _bootstrap_external is None:
|
|
|
|
raise NotImplementedError
|
2015-05-02 22:15:18 -03:00
|
|
|
self._cached = _bootstrap_external._get_cached(self.origin)
|
2013-11-22 12:05:39 -04:00
|
|
|
return self._cached
|
|
|
|
|
|
|
|
@cached.setter
|
|
|
|
def cached(self, cached):
|
|
|
|
self._cached = cached
|
|
|
|
|
|
|
|
@property
|
|
|
|
def parent(self):
|
|
|
|
"""The name of the module's parent."""
|
|
|
|
if self.submodule_search_locations is None:
|
|
|
|
return self.name.rpartition('.')[0]
|
|
|
|
else:
|
|
|
|
return self.name
|
|
|
|
|
|
|
|
@property
|
|
|
|
def has_location(self):
|
|
|
|
return self._set_fileattr
|
|
|
|
|
2013-12-11 01:16:41 -04:00
|
|
|
@has_location.setter
|
|
|
|
def has_location(self, value):
|
|
|
|
self._set_fileattr = bool(value)
|
|
|
|
|
2013-11-22 12:05:39 -04:00
|
|
|
|
|
|
|
def spec_from_loader(name, loader, *, origin=None, is_package=None):
|
|
|
|
"""Return a module spec based on various loader methods."""
|
bpo-21736: Set __file__ on frozen stdlib modules. (gh-28656)
Currently frozen modules do not have __file__ set. In their spec, origin is set to "frozen" and they are marked as not having a location. (Similarly, for frozen packages __path__ is set to an empty list.) However, for frozen stdlib modules we are able to extrapolate __file__ as long as we can determine the stdlib directory at runtime. (We now do so since gh-28586.) Having __file__ set is helpful for a number of reasons. Likewise, having a non-empty __path__ means we can import submodules of a frozen package from the filesystem (e.g. we could partially freeze the encodings module).
This change sets __file__ (and adds to __path__) for frozen stdlib modules. It uses sys._stdlibdir (from gh-28586) and the frozen module alias information (from gh-28655). All that work is done in FrozenImporter (in Lib/importlib/_bootstrap.py).
Also, if a frozen module is imported before importlib is bootstrapped (during interpreter initialization) then we fix up that module and its spec during the importlib bootstrapping step (i.e. imporlib._bootstrap._setup()) to match what gets set by FrozenImporter, including setting the file info (if the stdlib dir is known). To facilitate this, modules imported using PyImport_ImportFrozenModule() have __origname__ set using the frozen module alias info. __origname__ is popped off during importlib bootstrap.
(To be clear, even with this change the new code to set __file__ during fixups in imporlib._bootstrap._setup() doesn't actually get triggered yet. This is because sys._stdlibdir hasn't been set yet in interpreter initialization at the point importlib is bootstrapped. However, we do fix up such modules at that point to otherwise match the result of importing through FrozenImporter, just not the __file__ and __path__ parts. Doing so will require changes in the order in which things happen during interpreter initialization. That can be addressed separately. Once it is, the file-related fixup code from this PR will kick in.)
Here are things this change does not do:
* set __file__ for non-stdlib modules (no way of knowing the parent dir)
* set __file__ if the stdlib dir is not known (nor assume the expense of finding it)
* relatedly, set __file__ if the stdlib is in a zip file
* verify that the filename set to __file__ actually exists (too expensive)
* update __path__ for frozen packages that alias a non-package (since there is no package dir)
Other things this change skips, but we may do later:
* set __file__ on modules imported using PyImport_ImportFrozenModule()
* set co_filename when we unmarshal the frozen code object while importing the module (e.g. in FrozenImporter.exec_module()) -- this would allow tracebacks to show source lines
* implement FrozenImporter.get_filename() and FrozenImporter.get_source()
https://bugs.python.org/issue21736
2021-10-14 18:32:18 -03:00
|
|
|
if origin is None:
|
|
|
|
origin = getattr(loader, '_ORIGIN', None)
|
|
|
|
|
|
|
|
if not origin and hasattr(loader, 'get_filename'):
|
2015-05-16 00:54:59 -03:00
|
|
|
if _bootstrap_external is None:
|
|
|
|
raise NotImplementedError
|
|
|
|
spec_from_file_location = _bootstrap_external.spec_from_file_location
|
|
|
|
|
2013-11-22 12:05:39 -04:00
|
|
|
if is_package is None:
|
|
|
|
return spec_from_file_location(name, loader=loader)
|
|
|
|
search = [] if is_package else None
|
|
|
|
return spec_from_file_location(name, loader=loader,
|
|
|
|
submodule_search_locations=search)
|
|
|
|
|
|
|
|
if is_package is None:
|
|
|
|
if hasattr(loader, 'is_package'):
|
|
|
|
try:
|
|
|
|
is_package = loader.is_package(name)
|
|
|
|
except ImportError:
|
|
|
|
is_package = None # aka, undefined
|
|
|
|
else:
|
|
|
|
# the default
|
|
|
|
is_package = False
|
|
|
|
|
|
|
|
return ModuleSpec(name, loader, origin=origin, is_package=is_package)
|
|
|
|
|
|
|
|
|
|
|
|
def _spec_from_module(module, loader=None, origin=None):
|
|
|
|
# This function is meant for use in _setup().
|
|
|
|
try:
|
|
|
|
spec = module.__spec__
|
|
|
|
except AttributeError:
|
|
|
|
pass
|
|
|
|
else:
|
|
|
|
if spec is not None:
|
|
|
|
return spec
|
|
|
|
|
|
|
|
name = module.__name__
|
|
|
|
if loader is None:
|
|
|
|
try:
|
|
|
|
loader = module.__loader__
|
|
|
|
except AttributeError:
|
|
|
|
# loader will stay None.
|
|
|
|
pass
|
|
|
|
try:
|
|
|
|
location = module.__file__
|
|
|
|
except AttributeError:
|
|
|
|
location = None
|
|
|
|
if origin is None:
|
bpo-21736: Set __file__ on frozen stdlib modules. (gh-28656)
Currently frozen modules do not have __file__ set. In their spec, origin is set to "frozen" and they are marked as not having a location. (Similarly, for frozen packages __path__ is set to an empty list.) However, for frozen stdlib modules we are able to extrapolate __file__ as long as we can determine the stdlib directory at runtime. (We now do so since gh-28586.) Having __file__ set is helpful for a number of reasons. Likewise, having a non-empty __path__ means we can import submodules of a frozen package from the filesystem (e.g. we could partially freeze the encodings module).
This change sets __file__ (and adds to __path__) for frozen stdlib modules. It uses sys._stdlibdir (from gh-28586) and the frozen module alias information (from gh-28655). All that work is done in FrozenImporter (in Lib/importlib/_bootstrap.py).
Also, if a frozen module is imported before importlib is bootstrapped (during interpreter initialization) then we fix up that module and its spec during the importlib bootstrapping step (i.e. imporlib._bootstrap._setup()) to match what gets set by FrozenImporter, including setting the file info (if the stdlib dir is known). To facilitate this, modules imported using PyImport_ImportFrozenModule() have __origname__ set using the frozen module alias info. __origname__ is popped off during importlib bootstrap.
(To be clear, even with this change the new code to set __file__ during fixups in imporlib._bootstrap._setup() doesn't actually get triggered yet. This is because sys._stdlibdir hasn't been set yet in interpreter initialization at the point importlib is bootstrapped. However, we do fix up such modules at that point to otherwise match the result of importing through FrozenImporter, just not the __file__ and __path__ parts. Doing so will require changes in the order in which things happen during interpreter initialization. That can be addressed separately. Once it is, the file-related fixup code from this PR will kick in.)
Here are things this change does not do:
* set __file__ for non-stdlib modules (no way of knowing the parent dir)
* set __file__ if the stdlib dir is not known (nor assume the expense of finding it)
* relatedly, set __file__ if the stdlib is in a zip file
* verify that the filename set to __file__ actually exists (too expensive)
* update __path__ for frozen packages that alias a non-package (since there is no package dir)
Other things this change skips, but we may do later:
* set __file__ on modules imported using PyImport_ImportFrozenModule()
* set co_filename when we unmarshal the frozen code object while importing the module (e.g. in FrozenImporter.exec_module()) -- this would allow tracebacks to show source lines
* implement FrozenImporter.get_filename() and FrozenImporter.get_source()
https://bugs.python.org/issue21736
2021-10-14 18:32:18 -03:00
|
|
|
if loader is not None:
|
|
|
|
origin = getattr(loader, '_ORIGIN', None)
|
|
|
|
if not origin and location is not None:
|
2013-11-22 12:05:39 -04:00
|
|
|
origin = location
|
|
|
|
try:
|
|
|
|
cached = module.__cached__
|
|
|
|
except AttributeError:
|
|
|
|
cached = None
|
|
|
|
try:
|
|
|
|
submodule_search_locations = list(module.__path__)
|
|
|
|
except AttributeError:
|
|
|
|
submodule_search_locations = None
|
|
|
|
|
|
|
|
spec = ModuleSpec(name, loader, origin=origin)
|
bpo-21736: Set __file__ on frozen stdlib modules. (gh-28656)
Currently frozen modules do not have __file__ set. In their spec, origin is set to "frozen" and they are marked as not having a location. (Similarly, for frozen packages __path__ is set to an empty list.) However, for frozen stdlib modules we are able to extrapolate __file__ as long as we can determine the stdlib directory at runtime. (We now do so since gh-28586.) Having __file__ set is helpful for a number of reasons. Likewise, having a non-empty __path__ means we can import submodules of a frozen package from the filesystem (e.g. we could partially freeze the encodings module).
This change sets __file__ (and adds to __path__) for frozen stdlib modules. It uses sys._stdlibdir (from gh-28586) and the frozen module alias information (from gh-28655). All that work is done in FrozenImporter (in Lib/importlib/_bootstrap.py).
Also, if a frozen module is imported before importlib is bootstrapped (during interpreter initialization) then we fix up that module and its spec during the importlib bootstrapping step (i.e. imporlib._bootstrap._setup()) to match what gets set by FrozenImporter, including setting the file info (if the stdlib dir is known). To facilitate this, modules imported using PyImport_ImportFrozenModule() have __origname__ set using the frozen module alias info. __origname__ is popped off during importlib bootstrap.
(To be clear, even with this change the new code to set __file__ during fixups in imporlib._bootstrap._setup() doesn't actually get triggered yet. This is because sys._stdlibdir hasn't been set yet in interpreter initialization at the point importlib is bootstrapped. However, we do fix up such modules at that point to otherwise match the result of importing through FrozenImporter, just not the __file__ and __path__ parts. Doing so will require changes in the order in which things happen during interpreter initialization. That can be addressed separately. Once it is, the file-related fixup code from this PR will kick in.)
Here are things this change does not do:
* set __file__ for non-stdlib modules (no way of knowing the parent dir)
* set __file__ if the stdlib dir is not known (nor assume the expense of finding it)
* relatedly, set __file__ if the stdlib is in a zip file
* verify that the filename set to __file__ actually exists (too expensive)
* update __path__ for frozen packages that alias a non-package (since there is no package dir)
Other things this change skips, but we may do later:
* set __file__ on modules imported using PyImport_ImportFrozenModule()
* set co_filename when we unmarshal the frozen code object while importing the module (e.g. in FrozenImporter.exec_module()) -- this would allow tracebacks to show source lines
* implement FrozenImporter.get_filename() and FrozenImporter.get_source()
https://bugs.python.org/issue21736
2021-10-14 18:32:18 -03:00
|
|
|
spec._set_fileattr = False if location is None else (origin == location)
|
2013-11-22 12:05:39 -04:00
|
|
|
spec.cached = cached
|
|
|
|
spec.submodule_search_locations = submodule_search_locations
|
|
|
|
return spec
|
|
|
|
|
|
|
|
|
2014-05-30 15:55:29 -03:00
|
|
|
def _init_module_attrs(spec, module, *, override=False):
|
|
|
|
# The passed-in module may be not support attribute assignment,
|
|
|
|
# in which case we simply don't set the attributes.
|
|
|
|
# __name__
|
|
|
|
if (override or getattr(module, '__name__', None) is None):
|
|
|
|
try:
|
|
|
|
module.__name__ = spec.name
|
|
|
|
except AttributeError:
|
|
|
|
pass
|
|
|
|
# __loader__
|
|
|
|
if override or getattr(module, '__loader__', None) is None:
|
|
|
|
loader = spec.loader
|
|
|
|
if loader is None:
|
|
|
|
# A backward compatibility hack.
|
|
|
|
if spec.submodule_search_locations is not None:
|
2015-05-16 00:54:59 -03:00
|
|
|
if _bootstrap_external is None:
|
|
|
|
raise NotImplementedError
|
2021-10-20 18:05:29 -03:00
|
|
|
NamespaceLoader = _bootstrap_external.NamespaceLoader
|
2015-05-16 00:54:59 -03:00
|
|
|
|
2021-10-20 18:05:29 -03:00
|
|
|
loader = NamespaceLoader.__new__(NamespaceLoader)
|
2014-05-30 15:55:29 -03:00
|
|
|
loader._path = spec.submodule_search_locations
|
2018-02-02 16:15:58 -04:00
|
|
|
spec.loader = loader
|
|
|
|
# While the docs say that module.__file__ is not set for
|
|
|
|
# built-in modules, and the code below will avoid setting it if
|
|
|
|
# spec.has_location is false, this is incorrect for namespace
|
|
|
|
# packages. Namespace packages have no location, but their
|
|
|
|
# __spec__.origin is None, and thus their module.__file__
|
|
|
|
# should also be None for consistency. While a bit of a hack,
|
|
|
|
# this is the best place to ensure this consistency.
|
|
|
|
#
|
|
|
|
# See # https://docs.python.org/3/library/importlib.html#importlib.abc.Loader.load_module
|
|
|
|
# and bpo-32305
|
|
|
|
module.__file__ = None
|
2014-05-30 15:55:29 -03:00
|
|
|
try:
|
|
|
|
module.__loader__ = loader
|
|
|
|
except AttributeError:
|
|
|
|
pass
|
|
|
|
# __package__
|
|
|
|
if override or getattr(module, '__package__', None) is None:
|
|
|
|
try:
|
|
|
|
module.__package__ = spec.parent
|
|
|
|
except AttributeError:
|
|
|
|
pass
|
|
|
|
# __spec__
|
|
|
|
try:
|
|
|
|
module.__spec__ = spec
|
|
|
|
except AttributeError:
|
|
|
|
pass
|
|
|
|
# __path__
|
|
|
|
if override or getattr(module, '__path__', None) is None:
|
|
|
|
if spec.submodule_search_locations is not None:
|
bpo-21736: Set __file__ on frozen stdlib modules. (gh-28656)
Currently frozen modules do not have __file__ set. In their spec, origin is set to "frozen" and they are marked as not having a location. (Similarly, for frozen packages __path__ is set to an empty list.) However, for frozen stdlib modules we are able to extrapolate __file__ as long as we can determine the stdlib directory at runtime. (We now do so since gh-28586.) Having __file__ set is helpful for a number of reasons. Likewise, having a non-empty __path__ means we can import submodules of a frozen package from the filesystem (e.g. we could partially freeze the encodings module).
This change sets __file__ (and adds to __path__) for frozen stdlib modules. It uses sys._stdlibdir (from gh-28586) and the frozen module alias information (from gh-28655). All that work is done in FrozenImporter (in Lib/importlib/_bootstrap.py).
Also, if a frozen module is imported before importlib is bootstrapped (during interpreter initialization) then we fix up that module and its spec during the importlib bootstrapping step (i.e. imporlib._bootstrap._setup()) to match what gets set by FrozenImporter, including setting the file info (if the stdlib dir is known). To facilitate this, modules imported using PyImport_ImportFrozenModule() have __origname__ set using the frozen module alias info. __origname__ is popped off during importlib bootstrap.
(To be clear, even with this change the new code to set __file__ during fixups in imporlib._bootstrap._setup() doesn't actually get triggered yet. This is because sys._stdlibdir hasn't been set yet in interpreter initialization at the point importlib is bootstrapped. However, we do fix up such modules at that point to otherwise match the result of importing through FrozenImporter, just not the __file__ and __path__ parts. Doing so will require changes in the order in which things happen during interpreter initialization. That can be addressed separately. Once it is, the file-related fixup code from this PR will kick in.)
Here are things this change does not do:
* set __file__ for non-stdlib modules (no way of knowing the parent dir)
* set __file__ if the stdlib dir is not known (nor assume the expense of finding it)
* relatedly, set __file__ if the stdlib is in a zip file
* verify that the filename set to __file__ actually exists (too expensive)
* update __path__ for frozen packages that alias a non-package (since there is no package dir)
Other things this change skips, but we may do later:
* set __file__ on modules imported using PyImport_ImportFrozenModule()
* set co_filename when we unmarshal the frozen code object while importing the module (e.g. in FrozenImporter.exec_module()) -- this would allow tracebacks to show source lines
* implement FrozenImporter.get_filename() and FrozenImporter.get_source()
https://bugs.python.org/issue21736
2021-10-14 18:32:18 -03:00
|
|
|
# XXX We should extend __path__ if it's already a list.
|
2013-11-22 12:05:39 -04:00
|
|
|
try:
|
2014-05-30 15:55:29 -03:00
|
|
|
module.__path__ = spec.submodule_search_locations
|
2013-11-22 12:05:39 -04:00
|
|
|
except AttributeError:
|
|
|
|
pass
|
2014-05-30 15:55:29 -03:00
|
|
|
# __file__/__cached__
|
|
|
|
if spec.has_location:
|
|
|
|
if override or getattr(module, '__file__', None) is None:
|
2013-11-22 12:05:39 -04:00
|
|
|
try:
|
2014-05-30 15:55:29 -03:00
|
|
|
module.__file__ = spec.origin
|
2013-11-22 12:05:39 -04:00
|
|
|
except AttributeError:
|
|
|
|
pass
|
|
|
|
|
2014-05-30 15:55:29 -03:00
|
|
|
if override or getattr(module, '__cached__', None) is None:
|
|
|
|
if spec.cached is not None:
|
2013-11-22 12:05:39 -04:00
|
|
|
try:
|
2014-05-30 15:55:29 -03:00
|
|
|
module.__cached__ = spec.cached
|
2013-11-22 12:05:39 -04:00
|
|
|
except AttributeError:
|
|
|
|
pass
|
2014-05-30 15:55:29 -03:00
|
|
|
return module
|
2013-11-22 12:05:39 -04:00
|
|
|
|
|
|
|
|
2014-05-30 15:55:29 -03:00
|
|
|
def module_from_spec(spec):
|
|
|
|
"""Create a module based on the provided spec."""
|
|
|
|
# Typically loaders will not implement create_module().
|
|
|
|
module = None
|
|
|
|
if hasattr(spec.loader, 'create_module'):
|
|
|
|
# If create_module() returns `None` then it means default
|
|
|
|
# module creation should be used.
|
|
|
|
module = spec.loader.create_module(spec)
|
2015-01-09 12:39:21 -04:00
|
|
|
elif hasattr(spec.loader, 'exec_module'):
|
2016-09-08 15:12:31 -03:00
|
|
|
raise ImportError('loaders that define exec_module() '
|
|
|
|
'must also define create_module()')
|
2014-05-30 15:55:29 -03:00
|
|
|
if module is None:
|
|
|
|
module = _new_module(spec.name)
|
|
|
|
_init_module_attrs(spec, module)
|
|
|
|
return module
|
2013-11-22 12:05:39 -04:00
|
|
|
|
|
|
|
|
2014-05-30 15:55:29 -03:00
|
|
|
def _module_repr_from_spec(spec):
|
|
|
|
"""Return the repr to use for the module."""
|
|
|
|
name = '?' if spec.name is None else spec.name
|
|
|
|
if spec.origin is None:
|
|
|
|
if spec.loader is None:
|
2022-10-06 20:43:16 -03:00
|
|
|
return f'<module {name!r}>'
|
2013-11-22 12:05:39 -04:00
|
|
|
else:
|
2022-11-06 18:13:40 -04:00
|
|
|
return f'<module {name!r} (namespace) from {list(spec.loader._path)}>'
|
2014-05-30 15:55:29 -03:00
|
|
|
else:
|
|
|
|
if spec.has_location:
|
2022-10-06 20:43:16 -03:00
|
|
|
return f'<module {name!r} from {spec.origin!r}>'
|
2014-05-30 15:55:29 -03:00
|
|
|
else:
|
2022-10-06 20:43:16 -03:00
|
|
|
return f'<module {spec.name!r} ({spec.origin})>'
|
2013-11-22 12:05:39 -04:00
|
|
|
|
|
|
|
|
2014-05-30 15:55:29 -03:00
|
|
|
# Used by importlib.reload() and _load_module_shim().
|
|
|
|
def _exec(spec, module):
|
2016-07-08 15:09:35 -03:00
|
|
|
"""Execute the spec's specified module in an existing module's namespace."""
|
2014-05-30 15:55:29 -03:00
|
|
|
name = spec.name
|
|
|
|
with _ModuleLockManager(name):
|
|
|
|
if sys.modules.get(name) is not module:
|
2022-10-06 20:43:16 -03:00
|
|
|
msg = f'module {name!r} not in sys.modules'
|
2014-05-30 15:55:29 -03:00
|
|
|
raise ImportError(msg, name=name)
|
2018-10-29 14:30:16 -03:00
|
|
|
try:
|
|
|
|
if spec.loader is None:
|
|
|
|
if spec.submodule_search_locations is None:
|
|
|
|
raise ImportError('missing loader', name=spec.name)
|
|
|
|
# Namespace package.
|
|
|
|
_init_module_attrs(spec, module, override=True)
|
|
|
|
else:
|
|
|
|
_init_module_attrs(spec, module, override=True)
|
|
|
|
if not hasattr(spec.loader, 'exec_module'):
|
2020-12-04 19:39:21 -04:00
|
|
|
msg = (f"{_object_name(spec.loader)}.exec_module() not found; "
|
|
|
|
"falling back to load_module()")
|
|
|
|
_warnings.warn(msg, ImportWarning)
|
2018-10-29 14:30:16 -03:00
|
|
|
spec.loader.load_module(name)
|
|
|
|
else:
|
|
|
|
spec.loader.exec_module(module)
|
|
|
|
finally:
|
|
|
|
# Update the order of insertion into sys.modules for module
|
|
|
|
# clean-up at shutdown.
|
|
|
|
module = sys.modules.pop(spec.name)
|
|
|
|
sys.modules[spec.name] = module
|
|
|
|
return module
|
2014-05-30 15:55:29 -03:00
|
|
|
|
|
|
|
|
|
|
|
def _load_backward_compatible(spec):
|
2020-12-04 19:39:21 -04:00
|
|
|
# It is assumed that all callers have been warned about using load_module()
|
|
|
|
# appropriately before calling this function.
|
2018-10-29 14:30:16 -03:00
|
|
|
try:
|
|
|
|
spec.loader.load_module(spec.name)
|
|
|
|
except:
|
|
|
|
if spec.name in sys.modules:
|
|
|
|
module = sys.modules.pop(spec.name)
|
|
|
|
sys.modules[spec.name] = module
|
|
|
|
raise
|
2014-05-30 15:55:29 -03:00
|
|
|
# The module must be in sys.modules at this point!
|
2018-10-29 14:30:16 -03:00
|
|
|
# Move it to the end of sys.modules.
|
|
|
|
module = sys.modules.pop(spec.name)
|
|
|
|
sys.modules[spec.name] = module
|
2014-05-30 15:55:29 -03:00
|
|
|
if getattr(module, '__loader__', None) is None:
|
|
|
|
try:
|
|
|
|
module.__loader__ = spec.loader
|
|
|
|
except AttributeError:
|
|
|
|
pass
|
|
|
|
if getattr(module, '__package__', None) is None:
|
|
|
|
try:
|
|
|
|
# Since module.__path__ may not line up with
|
|
|
|
# spec.submodule_search_paths, we can't necessarily rely
|
|
|
|
# on spec.parent here.
|
|
|
|
module.__package__ = module.__name__
|
|
|
|
if not hasattr(module, '__path__'):
|
|
|
|
module.__package__ = spec.name.rpartition('.')[0]
|
|
|
|
except AttributeError:
|
|
|
|
pass
|
|
|
|
if getattr(module, '__spec__', None) is None:
|
|
|
|
try:
|
|
|
|
module.__spec__ = spec
|
|
|
|
except AttributeError:
|
|
|
|
pass
|
|
|
|
return module
|
2013-11-22 12:05:39 -04:00
|
|
|
|
2014-05-30 15:55:29 -03:00
|
|
|
def _load_unlocked(spec):
|
|
|
|
# A helper for direct use by the import system.
|
|
|
|
if spec.loader is not None:
|
2018-10-29 14:30:16 -03:00
|
|
|
# Not a namespace package.
|
2014-05-30 15:55:29 -03:00
|
|
|
if not hasattr(spec.loader, 'exec_module'):
|
2020-12-04 19:39:21 -04:00
|
|
|
msg = (f"{_object_name(spec.loader)}.exec_module() not found; "
|
|
|
|
"falling back to load_module()")
|
|
|
|
_warnings.warn(msg, ImportWarning)
|
2014-05-30 15:55:29 -03:00
|
|
|
return _load_backward_compatible(spec)
|
|
|
|
|
|
|
|
module = module_from_spec(spec)
|
2013-11-22 12:05:39 -04:00
|
|
|
|
2018-10-29 14:30:16 -03:00
|
|
|
# This must be done before putting the module in sys.modules
|
|
|
|
# (otherwise an optimization shortcut in import.c becomes
|
|
|
|
# wrong).
|
|
|
|
spec._initializing = True
|
|
|
|
try:
|
|
|
|
sys.modules[spec.name] = module
|
|
|
|
try:
|
|
|
|
if spec.loader is None:
|
|
|
|
if spec.submodule_search_locations is None:
|
|
|
|
raise ImportError('missing loader', name=spec.name)
|
|
|
|
# A namespace package so do nothing.
|
|
|
|
else:
|
|
|
|
spec.loader.exec_module(module)
|
|
|
|
except:
|
|
|
|
try:
|
|
|
|
del sys.modules[spec.name]
|
|
|
|
except KeyError:
|
|
|
|
pass
|
|
|
|
raise
|
|
|
|
# Move the module to the end of sys.modules.
|
|
|
|
# We don't ensure that the import-related module attributes get
|
|
|
|
# set in the sys.modules replacement case. Such modules are on
|
|
|
|
# their own.
|
|
|
|
module = sys.modules.pop(spec.name)
|
|
|
|
sys.modules[spec.name] = module
|
|
|
|
_verbose_message('import {!r} # {!r}', spec.name, spec.loader)
|
|
|
|
finally:
|
|
|
|
spec._initializing = False
|
|
|
|
|
|
|
|
return module
|
2013-11-22 12:05:39 -04:00
|
|
|
|
2014-05-30 15:55:29 -03:00
|
|
|
# A method used during testing of _load_unlocked() and by
|
|
|
|
# _load_module_shim().
|
|
|
|
def _load(spec):
|
|
|
|
"""Return a new module object, loaded by the spec's loader.
|
2013-11-22 12:05:39 -04:00
|
|
|
|
2014-05-30 15:55:29 -03:00
|
|
|
The module is not added to its parent.
|
2013-11-22 12:05:39 -04:00
|
|
|
|
2014-05-30 15:55:29 -03:00
|
|
|
If a module is already in sys.modules, that existing module gets
|
|
|
|
clobbered.
|
2013-11-22 12:05:39 -04:00
|
|
|
|
2014-05-30 15:55:29 -03:00
|
|
|
"""
|
|
|
|
with _ModuleLockManager(spec.name):
|
|
|
|
return _load_unlocked(spec)
|
2013-11-22 12:05:39 -04:00
|
|
|
|
|
|
|
|
2009-03-12 19:37:06 -03:00
|
|
|
# Loaders #####################################################################
|
2009-03-12 19:28:55 -03:00
|
|
|
|
2009-01-22 18:43:07 -04:00
|
|
|
class BuiltinImporter:
|
2009-01-17 20:24:28 -04:00
|
|
|
|
2009-03-14 21:53:05 -03:00
|
|
|
"""Meta path import for built-in modules.
|
2009-01-17 20:24:28 -04:00
|
|
|
|
2009-03-14 21:53:05 -03:00
|
|
|
All methods are either class or static methods to avoid the need to
|
|
|
|
instantiate the class.
|
2009-01-17 20:24:28 -04:00
|
|
|
|
|
|
|
"""
|
|
|
|
|
2019-09-11 13:00:02 -03:00
|
|
|
_ORIGIN = "built-in"
|
|
|
|
|
2013-11-22 12:05:39 -04:00
|
|
|
@classmethod
|
|
|
|
def find_spec(cls, fullname, path=None, target=None):
|
|
|
|
if path is not None:
|
|
|
|
return None
|
|
|
|
if _imp.is_builtin(fullname):
|
2019-09-11 13:00:02 -03:00
|
|
|
return spec_from_loader(fullname, cls, origin=cls._ORIGIN)
|
2013-11-22 12:05:39 -04:00
|
|
|
else:
|
|
|
|
return None
|
|
|
|
|
2009-01-22 18:43:07 -04:00
|
|
|
@classmethod
|
|
|
|
def find_module(cls, fullname, path=None):
|
2009-03-14 21:53:05 -03:00
|
|
|
"""Find the built-in module.
|
2009-01-17 20:24:28 -04:00
|
|
|
|
|
|
|
If 'path' is ever specified then the search is considered a failure.
|
|
|
|
|
2014-01-06 23:49:04 -04:00
|
|
|
This method is deprecated. Use find_spec() instead.
|
|
|
|
|
2009-01-17 20:24:28 -04:00
|
|
|
"""
|
2021-04-06 12:56:57 -03:00
|
|
|
_warnings.warn("BuiltinImporter.find_module() is deprecated and "
|
|
|
|
"slated for removal in Python 3.12; use find_spec() instead",
|
|
|
|
DeprecationWarning)
|
2013-11-22 12:05:39 -04:00
|
|
|
spec = cls.find_spec(fullname, path)
|
|
|
|
return spec.loader if spec is not None else None
|
|
|
|
|
2020-11-20 09:44:02 -04:00
|
|
|
@staticmethod
|
|
|
|
def create_module(spec):
|
2015-05-23 09:24:10 -03:00
|
|
|
"""Create a built-in module"""
|
|
|
|
if spec.name not in sys.builtin_module_names:
|
2022-10-06 20:43:16 -03:00
|
|
|
raise ImportError(f'{spec.name!r} is not a built-in module',
|
2015-05-23 09:24:10 -03:00
|
|
|
name=spec.name)
|
|
|
|
return _call_with_frames_removed(_imp.create_builtin, spec)
|
|
|
|
|
2020-11-20 09:44:02 -04:00
|
|
|
@staticmethod
|
|
|
|
def exec_module(module):
|
2015-05-23 09:24:10 -03:00
|
|
|
"""Exec a built-in module"""
|
2015-08-24 23:53:56 -03:00
|
|
|
_call_with_frames_removed(_imp.exec_builtin, module)
|
2009-01-17 20:24:28 -04:00
|
|
|
|
2009-03-14 22:41:33 -03:00
|
|
|
@classmethod
|
|
|
|
@_requires_builtin
|
|
|
|
def get_code(cls, fullname):
|
|
|
|
"""Return None as built-in modules do not have code objects."""
|
|
|
|
return None
|
|
|
|
|
|
|
|
@classmethod
|
|
|
|
@_requires_builtin
|
|
|
|
def get_source(cls, fullname):
|
|
|
|
"""Return None as built-in modules do not have source code."""
|
|
|
|
return None
|
|
|
|
|
|
|
|
@classmethod
|
|
|
|
@_requires_builtin
|
|
|
|
def is_package(cls, fullname):
|
2012-07-07 08:16:44 -03:00
|
|
|
"""Return False as built-in modules are never packages."""
|
2009-03-14 22:41:33 -03:00
|
|
|
return False
|
|
|
|
|
2015-05-23 09:24:10 -03:00
|
|
|
load_module = classmethod(_load_module_shim)
|
|
|
|
|
2009-01-17 20:24:28 -04:00
|
|
|
|
2009-01-22 18:43:07 -04:00
|
|
|
class FrozenImporter:
|
2009-01-17 20:24:28 -04:00
|
|
|
|
2009-03-14 21:53:05 -03:00
|
|
|
"""Meta path import for frozen modules.
|
2009-01-17 20:24:28 -04:00
|
|
|
|
2009-03-14 21:53:05 -03:00
|
|
|
All methods are either class or static methods to avoid the need to
|
|
|
|
instantiate the class.
|
2009-01-17 20:24:28 -04:00
|
|
|
|
2009-01-22 18:43:07 -04:00
|
|
|
"""
|
2009-01-17 20:24:28 -04:00
|
|
|
|
2019-02-04 20:56:26 -04:00
|
|
|
_ORIGIN = "frozen"
|
|
|
|
|
2021-10-05 14:26:37 -03:00
|
|
|
@classmethod
|
bpo-21736: Set __file__ on frozen stdlib modules. (gh-28656)
Currently frozen modules do not have __file__ set. In their spec, origin is set to "frozen" and they are marked as not having a location. (Similarly, for frozen packages __path__ is set to an empty list.) However, for frozen stdlib modules we are able to extrapolate __file__ as long as we can determine the stdlib directory at runtime. (We now do so since gh-28586.) Having __file__ set is helpful for a number of reasons. Likewise, having a non-empty __path__ means we can import submodules of a frozen package from the filesystem (e.g. we could partially freeze the encodings module).
This change sets __file__ (and adds to __path__) for frozen stdlib modules. It uses sys._stdlibdir (from gh-28586) and the frozen module alias information (from gh-28655). All that work is done in FrozenImporter (in Lib/importlib/_bootstrap.py).
Also, if a frozen module is imported before importlib is bootstrapped (during interpreter initialization) then we fix up that module and its spec during the importlib bootstrapping step (i.e. imporlib._bootstrap._setup()) to match what gets set by FrozenImporter, including setting the file info (if the stdlib dir is known). To facilitate this, modules imported using PyImport_ImportFrozenModule() have __origname__ set using the frozen module alias info. __origname__ is popped off during importlib bootstrap.
(To be clear, even with this change the new code to set __file__ during fixups in imporlib._bootstrap._setup() doesn't actually get triggered yet. This is because sys._stdlibdir hasn't been set yet in interpreter initialization at the point importlib is bootstrapped. However, we do fix up such modules at that point to otherwise match the result of importing through FrozenImporter, just not the __file__ and __path__ parts. Doing so will require changes in the order in which things happen during interpreter initialization. That can be addressed separately. Once it is, the file-related fixup code from this PR will kick in.)
Here are things this change does not do:
* set __file__ for non-stdlib modules (no way of knowing the parent dir)
* set __file__ if the stdlib dir is not known (nor assume the expense of finding it)
* relatedly, set __file__ if the stdlib is in a zip file
* verify that the filename set to __file__ actually exists (too expensive)
* update __path__ for frozen packages that alias a non-package (since there is no package dir)
Other things this change skips, but we may do later:
* set __file__ on modules imported using PyImport_ImportFrozenModule()
* set co_filename when we unmarshal the frozen code object while importing the module (e.g. in FrozenImporter.exec_module()) -- this would allow tracebacks to show source lines
* implement FrozenImporter.get_filename() and FrozenImporter.get_source()
https://bugs.python.org/issue21736
2021-10-14 18:32:18 -03:00
|
|
|
def _fix_up_module(cls, module):
|
2021-10-05 14:26:37 -03:00
|
|
|
spec = module.__spec__
|
bpo-21736: Set __file__ on frozen stdlib modules. (gh-28656)
Currently frozen modules do not have __file__ set. In their spec, origin is set to "frozen" and they are marked as not having a location. (Similarly, for frozen packages __path__ is set to an empty list.) However, for frozen stdlib modules we are able to extrapolate __file__ as long as we can determine the stdlib directory at runtime. (We now do so since gh-28586.) Having __file__ set is helpful for a number of reasons. Likewise, having a non-empty __path__ means we can import submodules of a frozen package from the filesystem (e.g. we could partially freeze the encodings module).
This change sets __file__ (and adds to __path__) for frozen stdlib modules. It uses sys._stdlibdir (from gh-28586) and the frozen module alias information (from gh-28655). All that work is done in FrozenImporter (in Lib/importlib/_bootstrap.py).
Also, if a frozen module is imported before importlib is bootstrapped (during interpreter initialization) then we fix up that module and its spec during the importlib bootstrapping step (i.e. imporlib._bootstrap._setup()) to match what gets set by FrozenImporter, including setting the file info (if the stdlib dir is known). To facilitate this, modules imported using PyImport_ImportFrozenModule() have __origname__ set using the frozen module alias info. __origname__ is popped off during importlib bootstrap.
(To be clear, even with this change the new code to set __file__ during fixups in imporlib._bootstrap._setup() doesn't actually get triggered yet. This is because sys._stdlibdir hasn't been set yet in interpreter initialization at the point importlib is bootstrapped. However, we do fix up such modules at that point to otherwise match the result of importing through FrozenImporter, just not the __file__ and __path__ parts. Doing so will require changes in the order in which things happen during interpreter initialization. That can be addressed separately. Once it is, the file-related fixup code from this PR will kick in.)
Here are things this change does not do:
* set __file__ for non-stdlib modules (no way of knowing the parent dir)
* set __file__ if the stdlib dir is not known (nor assume the expense of finding it)
* relatedly, set __file__ if the stdlib is in a zip file
* verify that the filename set to __file__ actually exists (too expensive)
* update __path__ for frozen packages that alias a non-package (since there is no package dir)
Other things this change skips, but we may do later:
* set __file__ on modules imported using PyImport_ImportFrozenModule()
* set co_filename when we unmarshal the frozen code object while importing the module (e.g. in FrozenImporter.exec_module()) -- this would allow tracebacks to show source lines
* implement FrozenImporter.get_filename() and FrozenImporter.get_source()
https://bugs.python.org/issue21736
2021-10-14 18:32:18 -03:00
|
|
|
state = spec.loader_state
|
|
|
|
if state is None:
|
|
|
|
# The module is missing FrozenImporter-specific values.
|
2021-10-05 14:26:37 -03:00
|
|
|
|
bpo-21736: Set __file__ on frozen stdlib modules. (gh-28656)
Currently frozen modules do not have __file__ set. In their spec, origin is set to "frozen" and they are marked as not having a location. (Similarly, for frozen packages __path__ is set to an empty list.) However, for frozen stdlib modules we are able to extrapolate __file__ as long as we can determine the stdlib directory at runtime. (We now do so since gh-28586.) Having __file__ set is helpful for a number of reasons. Likewise, having a non-empty __path__ means we can import submodules of a frozen package from the filesystem (e.g. we could partially freeze the encodings module).
This change sets __file__ (and adds to __path__) for frozen stdlib modules. It uses sys._stdlibdir (from gh-28586) and the frozen module alias information (from gh-28655). All that work is done in FrozenImporter (in Lib/importlib/_bootstrap.py).
Also, if a frozen module is imported before importlib is bootstrapped (during interpreter initialization) then we fix up that module and its spec during the importlib bootstrapping step (i.e. imporlib._bootstrap._setup()) to match what gets set by FrozenImporter, including setting the file info (if the stdlib dir is known). To facilitate this, modules imported using PyImport_ImportFrozenModule() have __origname__ set using the frozen module alias info. __origname__ is popped off during importlib bootstrap.
(To be clear, even with this change the new code to set __file__ during fixups in imporlib._bootstrap._setup() doesn't actually get triggered yet. This is because sys._stdlibdir hasn't been set yet in interpreter initialization at the point importlib is bootstrapped. However, we do fix up such modules at that point to otherwise match the result of importing through FrozenImporter, just not the __file__ and __path__ parts. Doing so will require changes in the order in which things happen during interpreter initialization. That can be addressed separately. Once it is, the file-related fixup code from this PR will kick in.)
Here are things this change does not do:
* set __file__ for non-stdlib modules (no way of knowing the parent dir)
* set __file__ if the stdlib dir is not known (nor assume the expense of finding it)
* relatedly, set __file__ if the stdlib is in a zip file
* verify that the filename set to __file__ actually exists (too expensive)
* update __path__ for frozen packages that alias a non-package (since there is no package dir)
Other things this change skips, but we may do later:
* set __file__ on modules imported using PyImport_ImportFrozenModule()
* set co_filename when we unmarshal the frozen code object while importing the module (e.g. in FrozenImporter.exec_module()) -- this would allow tracebacks to show source lines
* implement FrozenImporter.get_filename() and FrozenImporter.get_source()
https://bugs.python.org/issue21736
2021-10-14 18:32:18 -03:00
|
|
|
# Fix up the spec attrs.
|
2021-10-05 14:26:37 -03:00
|
|
|
origname = vars(module).pop('__origname__', None)
|
|
|
|
assert origname, 'see PyImport_ImportFrozenModuleObject()'
|
bpo-21736: Set __file__ on frozen stdlib modules. (gh-28656)
Currently frozen modules do not have __file__ set. In their spec, origin is set to "frozen" and they are marked as not having a location. (Similarly, for frozen packages __path__ is set to an empty list.) However, for frozen stdlib modules we are able to extrapolate __file__ as long as we can determine the stdlib directory at runtime. (We now do so since gh-28586.) Having __file__ set is helpful for a number of reasons. Likewise, having a non-empty __path__ means we can import submodules of a frozen package from the filesystem (e.g. we could partially freeze the encodings module).
This change sets __file__ (and adds to __path__) for frozen stdlib modules. It uses sys._stdlibdir (from gh-28586) and the frozen module alias information (from gh-28655). All that work is done in FrozenImporter (in Lib/importlib/_bootstrap.py).
Also, if a frozen module is imported before importlib is bootstrapped (during interpreter initialization) then we fix up that module and its spec during the importlib bootstrapping step (i.e. imporlib._bootstrap._setup()) to match what gets set by FrozenImporter, including setting the file info (if the stdlib dir is known). To facilitate this, modules imported using PyImport_ImportFrozenModule() have __origname__ set using the frozen module alias info. __origname__ is popped off during importlib bootstrap.
(To be clear, even with this change the new code to set __file__ during fixups in imporlib._bootstrap._setup() doesn't actually get triggered yet. This is because sys._stdlibdir hasn't been set yet in interpreter initialization at the point importlib is bootstrapped. However, we do fix up such modules at that point to otherwise match the result of importing through FrozenImporter, just not the __file__ and __path__ parts. Doing so will require changes in the order in which things happen during interpreter initialization. That can be addressed separately. Once it is, the file-related fixup code from this PR will kick in.)
Here are things this change does not do:
* set __file__ for non-stdlib modules (no way of knowing the parent dir)
* set __file__ if the stdlib dir is not known (nor assume the expense of finding it)
* relatedly, set __file__ if the stdlib is in a zip file
* verify that the filename set to __file__ actually exists (too expensive)
* update __path__ for frozen packages that alias a non-package (since there is no package dir)
Other things this change skips, but we may do later:
* set __file__ on modules imported using PyImport_ImportFrozenModule()
* set co_filename when we unmarshal the frozen code object while importing the module (e.g. in FrozenImporter.exec_module()) -- this would allow tracebacks to show source lines
* implement FrozenImporter.get_filename() and FrozenImporter.get_source()
https://bugs.python.org/issue21736
2021-10-14 18:32:18 -03:00
|
|
|
ispkg = hasattr(module, '__path__')
|
|
|
|
assert _imp.is_frozen_package(module.__name__) == ispkg, ispkg
|
|
|
|
filename, pkgdir = cls._resolve_filename(origname, spec.name, ispkg)
|
|
|
|
spec.loader_state = type(sys.implementation)(
|
|
|
|
filename=filename,
|
|
|
|
origname=origname,
|
|
|
|
)
|
|
|
|
__path__ = spec.submodule_search_locations
|
|
|
|
if ispkg:
|
|
|
|
assert __path__ == [], __path__
|
|
|
|
if pkgdir:
|
|
|
|
spec.submodule_search_locations.insert(0, pkgdir)
|
|
|
|
else:
|
|
|
|
assert __path__ is None, __path__
|
|
|
|
|
|
|
|
# Fix up the module attrs (the bare minimum).
|
|
|
|
assert not hasattr(module, '__file__'), module.__file__
|
|
|
|
if filename:
|
|
|
|
try:
|
|
|
|
module.__file__ = filename
|
|
|
|
except AttributeError:
|
|
|
|
pass
|
|
|
|
if ispkg:
|
|
|
|
if module.__path__ != __path__:
|
|
|
|
assert module.__path__ == [], module.__path__
|
|
|
|
module.__path__.extend(__path__)
|
|
|
|
else:
|
|
|
|
# These checks ensure that _fix_up_module() is only called
|
|
|
|
# in the right places.
|
|
|
|
__path__ = spec.submodule_search_locations
|
|
|
|
ispkg = __path__ is not None
|
|
|
|
# Check the loader state.
|
|
|
|
assert sorted(vars(state)) == ['filename', 'origname'], state
|
|
|
|
if state.origname:
|
|
|
|
# The only frozen modules with "origname" set are stdlib modules.
|
|
|
|
(__file__, pkgdir,
|
|
|
|
) = cls._resolve_filename(state.origname, spec.name, ispkg)
|
|
|
|
assert state.filename == __file__, (state.filename, __file__)
|
|
|
|
if pkgdir:
|
|
|
|
assert __path__ == [pkgdir], (__path__, pkgdir)
|
|
|
|
else:
|
|
|
|
assert __path__ == ([] if ispkg else None), __path__
|
|
|
|
else:
|
|
|
|
__file__ = None
|
|
|
|
assert state.filename is None, state.filename
|
|
|
|
assert __path__ == ([] if ispkg else None), __path__
|
|
|
|
# Check the file attrs.
|
|
|
|
if __file__:
|
|
|
|
assert hasattr(module, '__file__')
|
|
|
|
assert module.__file__ == __file__, (module.__file__, __file__)
|
|
|
|
else:
|
|
|
|
assert not hasattr(module, '__file__'), module.__file__
|
|
|
|
if ispkg:
|
|
|
|
assert hasattr(module, '__path__')
|
|
|
|
assert module.__path__ == __path__, (module.__path__, __path__)
|
|
|
|
else:
|
|
|
|
assert not hasattr(module, '__path__'), module.__path__
|
|
|
|
assert not spec.has_location
|
|
|
|
|
|
|
|
@classmethod
|
|
|
|
def _resolve_filename(cls, fullname, alias=None, ispkg=False):
|
|
|
|
if not fullname or not getattr(sys, '_stdlib_dir', None):
|
|
|
|
return None, None
|
|
|
|
try:
|
|
|
|
sep = cls._SEP
|
|
|
|
except AttributeError:
|
|
|
|
sep = cls._SEP = '\\' if sys.platform == 'win32' else '/'
|
|
|
|
|
|
|
|
if fullname != alias:
|
|
|
|
if fullname.startswith('<'):
|
|
|
|
fullname = fullname[1:]
|
|
|
|
if not ispkg:
|
|
|
|
fullname = f'{fullname}.__init__'
|
|
|
|
else:
|
|
|
|
ispkg = False
|
|
|
|
relfile = fullname.replace('.', sep)
|
|
|
|
if ispkg:
|
|
|
|
pkgdir = f'{sys._stdlib_dir}{sep}{relfile}'
|
|
|
|
filename = f'{pkgdir}{sep}__init__.py'
|
|
|
|
else:
|
|
|
|
pkgdir = None
|
|
|
|
filename = f'{sys._stdlib_dir}{sep}{relfile}.py'
|
|
|
|
return filename, pkgdir
|
2021-10-05 14:26:37 -03:00
|
|
|
|
2013-11-22 12:05:39 -04:00
|
|
|
@classmethod
|
|
|
|
def find_spec(cls, fullname, path=None, target=None):
|
2021-10-05 13:01:27 -03:00
|
|
|
info = _call_with_frames_removed(_imp.find_frozen, fullname)
|
|
|
|
if info is None:
|
2013-11-22 12:05:39 -04:00
|
|
|
return None
|
bpo-21736: Set __file__ on frozen stdlib modules. (gh-28656)
Currently frozen modules do not have __file__ set. In their spec, origin is set to "frozen" and they are marked as not having a location. (Similarly, for frozen packages __path__ is set to an empty list.) However, for frozen stdlib modules we are able to extrapolate __file__ as long as we can determine the stdlib directory at runtime. (We now do so since gh-28586.) Having __file__ set is helpful for a number of reasons. Likewise, having a non-empty __path__ means we can import submodules of a frozen package from the filesystem (e.g. we could partially freeze the encodings module).
This change sets __file__ (and adds to __path__) for frozen stdlib modules. It uses sys._stdlibdir (from gh-28586) and the frozen module alias information (from gh-28655). All that work is done in FrozenImporter (in Lib/importlib/_bootstrap.py).
Also, if a frozen module is imported before importlib is bootstrapped (during interpreter initialization) then we fix up that module and its spec during the importlib bootstrapping step (i.e. imporlib._bootstrap._setup()) to match what gets set by FrozenImporter, including setting the file info (if the stdlib dir is known). To facilitate this, modules imported using PyImport_ImportFrozenModule() have __origname__ set using the frozen module alias info. __origname__ is popped off during importlib bootstrap.
(To be clear, even with this change the new code to set __file__ during fixups in imporlib._bootstrap._setup() doesn't actually get triggered yet. This is because sys._stdlibdir hasn't been set yet in interpreter initialization at the point importlib is bootstrapped. However, we do fix up such modules at that point to otherwise match the result of importing through FrozenImporter, just not the __file__ and __path__ parts. Doing so will require changes in the order in which things happen during interpreter initialization. That can be addressed separately. Once it is, the file-related fixup code from this PR will kick in.)
Here are things this change does not do:
* set __file__ for non-stdlib modules (no way of knowing the parent dir)
* set __file__ if the stdlib dir is not known (nor assume the expense of finding it)
* relatedly, set __file__ if the stdlib is in a zip file
* verify that the filename set to __file__ actually exists (too expensive)
* update __path__ for frozen packages that alias a non-package (since there is no package dir)
Other things this change skips, but we may do later:
* set __file__ on modules imported using PyImport_ImportFrozenModule()
* set co_filename when we unmarshal the frozen code object while importing the module (e.g. in FrozenImporter.exec_module()) -- this would allow tracebacks to show source lines
* implement FrozenImporter.get_filename() and FrozenImporter.get_source()
https://bugs.python.org/issue21736
2021-10-14 18:32:18 -03:00
|
|
|
# We get the marshaled data in exec_module() (the loader
|
|
|
|
# part of the importer), instead of here (the finder part).
|
|
|
|
# The loader is the usual place to get the data that will
|
|
|
|
# be loaded into the module. (For example, see _LoaderBasics
|
|
|
|
# in _bootstra_external.py.) Most importantly, this importer
|
|
|
|
# is simpler if we wait to get the data.
|
|
|
|
# However, getting as much data in the finder as possible
|
|
|
|
# to later load the module is okay, and sometimes important.
|
|
|
|
# (That's why ModuleSpec.loader_state exists.) This is
|
|
|
|
# especially true if it avoids throwing away expensive data
|
|
|
|
# the loader would otherwise duplicate later and can be done
|
|
|
|
# efficiently. In this case it isn't worth it.
|
|
|
|
_, ispkg, origname = info
|
2021-10-05 13:01:27 -03:00
|
|
|
spec = spec_from_loader(fullname, cls,
|
|
|
|
origin=cls._ORIGIN,
|
|
|
|
is_package=ispkg)
|
bpo-21736: Set __file__ on frozen stdlib modules. (gh-28656)
Currently frozen modules do not have __file__ set. In their spec, origin is set to "frozen" and they are marked as not having a location. (Similarly, for frozen packages __path__ is set to an empty list.) However, for frozen stdlib modules we are able to extrapolate __file__ as long as we can determine the stdlib directory at runtime. (We now do so since gh-28586.) Having __file__ set is helpful for a number of reasons. Likewise, having a non-empty __path__ means we can import submodules of a frozen package from the filesystem (e.g. we could partially freeze the encodings module).
This change sets __file__ (and adds to __path__) for frozen stdlib modules. It uses sys._stdlibdir (from gh-28586) and the frozen module alias information (from gh-28655). All that work is done in FrozenImporter (in Lib/importlib/_bootstrap.py).
Also, if a frozen module is imported before importlib is bootstrapped (during interpreter initialization) then we fix up that module and its spec during the importlib bootstrapping step (i.e. imporlib._bootstrap._setup()) to match what gets set by FrozenImporter, including setting the file info (if the stdlib dir is known). To facilitate this, modules imported using PyImport_ImportFrozenModule() have __origname__ set using the frozen module alias info. __origname__ is popped off during importlib bootstrap.
(To be clear, even with this change the new code to set __file__ during fixups in imporlib._bootstrap._setup() doesn't actually get triggered yet. This is because sys._stdlibdir hasn't been set yet in interpreter initialization at the point importlib is bootstrapped. However, we do fix up such modules at that point to otherwise match the result of importing through FrozenImporter, just not the __file__ and __path__ parts. Doing so will require changes in the order in which things happen during interpreter initialization. That can be addressed separately. Once it is, the file-related fixup code from this PR will kick in.)
Here are things this change does not do:
* set __file__ for non-stdlib modules (no way of knowing the parent dir)
* set __file__ if the stdlib dir is not known (nor assume the expense of finding it)
* relatedly, set __file__ if the stdlib is in a zip file
* verify that the filename set to __file__ actually exists (too expensive)
* update __path__ for frozen packages that alias a non-package (since there is no package dir)
Other things this change skips, but we may do later:
* set __file__ on modules imported using PyImport_ImportFrozenModule()
* set co_filename when we unmarshal the frozen code object while importing the module (e.g. in FrozenImporter.exec_module()) -- this would allow tracebacks to show source lines
* implement FrozenImporter.get_filename() and FrozenImporter.get_source()
https://bugs.python.org/issue21736
2021-10-14 18:32:18 -03:00
|
|
|
filename, pkgdir = cls._resolve_filename(origname, fullname, ispkg)
|
2021-10-05 14:26:37 -03:00
|
|
|
spec.loader_state = type(sys.implementation)(
|
bpo-21736: Set __file__ on frozen stdlib modules. (gh-28656)
Currently frozen modules do not have __file__ set. In their spec, origin is set to "frozen" and they are marked as not having a location. (Similarly, for frozen packages __path__ is set to an empty list.) However, for frozen stdlib modules we are able to extrapolate __file__ as long as we can determine the stdlib directory at runtime. (We now do so since gh-28586.) Having __file__ set is helpful for a number of reasons. Likewise, having a non-empty __path__ means we can import submodules of a frozen package from the filesystem (e.g. we could partially freeze the encodings module).
This change sets __file__ (and adds to __path__) for frozen stdlib modules. It uses sys._stdlibdir (from gh-28586) and the frozen module alias information (from gh-28655). All that work is done in FrozenImporter (in Lib/importlib/_bootstrap.py).
Also, if a frozen module is imported before importlib is bootstrapped (during interpreter initialization) then we fix up that module and its spec during the importlib bootstrapping step (i.e. imporlib._bootstrap._setup()) to match what gets set by FrozenImporter, including setting the file info (if the stdlib dir is known). To facilitate this, modules imported using PyImport_ImportFrozenModule() have __origname__ set using the frozen module alias info. __origname__ is popped off during importlib bootstrap.
(To be clear, even with this change the new code to set __file__ during fixups in imporlib._bootstrap._setup() doesn't actually get triggered yet. This is because sys._stdlibdir hasn't been set yet in interpreter initialization at the point importlib is bootstrapped. However, we do fix up such modules at that point to otherwise match the result of importing through FrozenImporter, just not the __file__ and __path__ parts. Doing so will require changes in the order in which things happen during interpreter initialization. That can be addressed separately. Once it is, the file-related fixup code from this PR will kick in.)
Here are things this change does not do:
* set __file__ for non-stdlib modules (no way of knowing the parent dir)
* set __file__ if the stdlib dir is not known (nor assume the expense of finding it)
* relatedly, set __file__ if the stdlib is in a zip file
* verify that the filename set to __file__ actually exists (too expensive)
* update __path__ for frozen packages that alias a non-package (since there is no package dir)
Other things this change skips, but we may do later:
* set __file__ on modules imported using PyImport_ImportFrozenModule()
* set co_filename when we unmarshal the frozen code object while importing the module (e.g. in FrozenImporter.exec_module()) -- this would allow tracebacks to show source lines
* implement FrozenImporter.get_filename() and FrozenImporter.get_source()
https://bugs.python.org/issue21736
2021-10-14 18:32:18 -03:00
|
|
|
filename=filename,
|
2021-10-05 14:26:37 -03:00
|
|
|
origname=origname,
|
|
|
|
)
|
bpo-21736: Set __file__ on frozen stdlib modules. (gh-28656)
Currently frozen modules do not have __file__ set. In their spec, origin is set to "frozen" and they are marked as not having a location. (Similarly, for frozen packages __path__ is set to an empty list.) However, for frozen stdlib modules we are able to extrapolate __file__ as long as we can determine the stdlib directory at runtime. (We now do so since gh-28586.) Having __file__ set is helpful for a number of reasons. Likewise, having a non-empty __path__ means we can import submodules of a frozen package from the filesystem (e.g. we could partially freeze the encodings module).
This change sets __file__ (and adds to __path__) for frozen stdlib modules. It uses sys._stdlibdir (from gh-28586) and the frozen module alias information (from gh-28655). All that work is done in FrozenImporter (in Lib/importlib/_bootstrap.py).
Also, if a frozen module is imported before importlib is bootstrapped (during interpreter initialization) then we fix up that module and its spec during the importlib bootstrapping step (i.e. imporlib._bootstrap._setup()) to match what gets set by FrozenImporter, including setting the file info (if the stdlib dir is known). To facilitate this, modules imported using PyImport_ImportFrozenModule() have __origname__ set using the frozen module alias info. __origname__ is popped off during importlib bootstrap.
(To be clear, even with this change the new code to set __file__ during fixups in imporlib._bootstrap._setup() doesn't actually get triggered yet. This is because sys._stdlibdir hasn't been set yet in interpreter initialization at the point importlib is bootstrapped. However, we do fix up such modules at that point to otherwise match the result of importing through FrozenImporter, just not the __file__ and __path__ parts. Doing so will require changes in the order in which things happen during interpreter initialization. That can be addressed separately. Once it is, the file-related fixup code from this PR will kick in.)
Here are things this change does not do:
* set __file__ for non-stdlib modules (no way of knowing the parent dir)
* set __file__ if the stdlib dir is not known (nor assume the expense of finding it)
* relatedly, set __file__ if the stdlib is in a zip file
* verify that the filename set to __file__ actually exists (too expensive)
* update __path__ for frozen packages that alias a non-package (since there is no package dir)
Other things this change skips, but we may do later:
* set __file__ on modules imported using PyImport_ImportFrozenModule()
* set co_filename when we unmarshal the frozen code object while importing the module (e.g. in FrozenImporter.exec_module()) -- this would allow tracebacks to show source lines
* implement FrozenImporter.get_filename() and FrozenImporter.get_source()
https://bugs.python.org/issue21736
2021-10-14 18:32:18 -03:00
|
|
|
if pkgdir:
|
|
|
|
spec.submodule_search_locations.insert(0, pkgdir)
|
2021-10-05 13:01:27 -03:00
|
|
|
return spec
|
2013-11-22 12:05:39 -04:00
|
|
|
|
2009-01-22 18:43:07 -04:00
|
|
|
@classmethod
|
|
|
|
def find_module(cls, fullname, path=None):
|
2014-01-06 23:49:04 -04:00
|
|
|
"""Find a frozen module.
|
|
|
|
|
|
|
|
This method is deprecated. Use find_spec() instead.
|
|
|
|
|
|
|
|
"""
|
2021-04-06 12:56:57 -03:00
|
|
|
_warnings.warn("FrozenImporter.find_module() is deprecated and "
|
|
|
|
"slated for removal in Python 3.12; use find_spec() instead",
|
|
|
|
DeprecationWarning)
|
2012-04-15 17:08:47 -03:00
|
|
|
return cls if _imp.is_frozen(fullname) else None
|
2009-01-17 20:24:28 -04:00
|
|
|
|
2020-11-20 09:44:02 -04:00
|
|
|
@staticmethod
|
|
|
|
def create_module(spec):
|
bpo-21736: Set __file__ on frozen stdlib modules. (gh-28656)
Currently frozen modules do not have __file__ set. In their spec, origin is set to "frozen" and they are marked as not having a location. (Similarly, for frozen packages __path__ is set to an empty list.) However, for frozen stdlib modules we are able to extrapolate __file__ as long as we can determine the stdlib directory at runtime. (We now do so since gh-28586.) Having __file__ set is helpful for a number of reasons. Likewise, having a non-empty __path__ means we can import submodules of a frozen package from the filesystem (e.g. we could partially freeze the encodings module).
This change sets __file__ (and adds to __path__) for frozen stdlib modules. It uses sys._stdlibdir (from gh-28586) and the frozen module alias information (from gh-28655). All that work is done in FrozenImporter (in Lib/importlib/_bootstrap.py).
Also, if a frozen module is imported before importlib is bootstrapped (during interpreter initialization) then we fix up that module and its spec during the importlib bootstrapping step (i.e. imporlib._bootstrap._setup()) to match what gets set by FrozenImporter, including setting the file info (if the stdlib dir is known). To facilitate this, modules imported using PyImport_ImportFrozenModule() have __origname__ set using the frozen module alias info. __origname__ is popped off during importlib bootstrap.
(To be clear, even with this change the new code to set __file__ during fixups in imporlib._bootstrap._setup() doesn't actually get triggered yet. This is because sys._stdlibdir hasn't been set yet in interpreter initialization at the point importlib is bootstrapped. However, we do fix up such modules at that point to otherwise match the result of importing through FrozenImporter, just not the __file__ and __path__ parts. Doing so will require changes in the order in which things happen during interpreter initialization. That can be addressed separately. Once it is, the file-related fixup code from this PR will kick in.)
Here are things this change does not do:
* set __file__ for non-stdlib modules (no way of knowing the parent dir)
* set __file__ if the stdlib dir is not known (nor assume the expense of finding it)
* relatedly, set __file__ if the stdlib is in a zip file
* verify that the filename set to __file__ actually exists (too expensive)
* update __path__ for frozen packages that alias a non-package (since there is no package dir)
Other things this change skips, but we may do later:
* set __file__ on modules imported using PyImport_ImportFrozenModule()
* set co_filename when we unmarshal the frozen code object while importing the module (e.g. in FrozenImporter.exec_module()) -- this would allow tracebacks to show source lines
* implement FrozenImporter.get_filename() and FrozenImporter.get_source()
https://bugs.python.org/issue21736
2021-10-14 18:32:18 -03:00
|
|
|
"""Set __file__, if able."""
|
|
|
|
module = _new_module(spec.name)
|
|
|
|
try:
|
|
|
|
filename = spec.loader_state.filename
|
|
|
|
except AttributeError:
|
|
|
|
pass
|
|
|
|
else:
|
|
|
|
if filename:
|
|
|
|
module.__file__ = filename
|
|
|
|
return module
|
2015-01-09 12:39:21 -04:00
|
|
|
|
2013-11-22 12:05:39 -04:00
|
|
|
@staticmethod
|
|
|
|
def exec_module(module):
|
2021-10-05 13:01:27 -03:00
|
|
|
spec = module.__spec__
|
|
|
|
name = spec.name
|
bpo-21736: Set __file__ on frozen stdlib modules. (gh-28656)
Currently frozen modules do not have __file__ set. In their spec, origin is set to "frozen" and they are marked as not having a location. (Similarly, for frozen packages __path__ is set to an empty list.) However, for frozen stdlib modules we are able to extrapolate __file__ as long as we can determine the stdlib directory at runtime. (We now do so since gh-28586.) Having __file__ set is helpful for a number of reasons. Likewise, having a non-empty __path__ means we can import submodules of a frozen package from the filesystem (e.g. we could partially freeze the encodings module).
This change sets __file__ (and adds to __path__) for frozen stdlib modules. It uses sys._stdlibdir (from gh-28586) and the frozen module alias information (from gh-28655). All that work is done in FrozenImporter (in Lib/importlib/_bootstrap.py).
Also, if a frozen module is imported before importlib is bootstrapped (during interpreter initialization) then we fix up that module and its spec during the importlib bootstrapping step (i.e. imporlib._bootstrap._setup()) to match what gets set by FrozenImporter, including setting the file info (if the stdlib dir is known). To facilitate this, modules imported using PyImport_ImportFrozenModule() have __origname__ set using the frozen module alias info. __origname__ is popped off during importlib bootstrap.
(To be clear, even with this change the new code to set __file__ during fixups in imporlib._bootstrap._setup() doesn't actually get triggered yet. This is because sys._stdlibdir hasn't been set yet in interpreter initialization at the point importlib is bootstrapped. However, we do fix up such modules at that point to otherwise match the result of importing through FrozenImporter, just not the __file__ and __path__ parts. Doing so will require changes in the order in which things happen during interpreter initialization. That can be addressed separately. Once it is, the file-related fixup code from this PR will kick in.)
Here are things this change does not do:
* set __file__ for non-stdlib modules (no way of knowing the parent dir)
* set __file__ if the stdlib dir is not known (nor assume the expense of finding it)
* relatedly, set __file__ if the stdlib is in a zip file
* verify that the filename set to __file__ actually exists (too expensive)
* update __path__ for frozen packages that alias a non-package (since there is no package dir)
Other things this change skips, but we may do later:
* set __file__ on modules imported using PyImport_ImportFrozenModule()
* set co_filename when we unmarshal the frozen code object while importing the module (e.g. in FrozenImporter.exec_module()) -- this would allow tracebacks to show source lines
* implement FrozenImporter.get_filename() and FrozenImporter.get_source()
https://bugs.python.org/issue21736
2021-10-14 18:32:18 -03:00
|
|
|
code = _call_with_frames_removed(_imp.get_frozen_object, name)
|
2013-11-22 12:05:39 -04:00
|
|
|
exec(code, module.__dict__)
|
|
|
|
|
2009-01-22 18:43:07 -04:00
|
|
|
@classmethod
|
|
|
|
def load_module(cls, fullname):
|
2014-01-06 23:49:04 -04:00
|
|
|
"""Load a frozen module.
|
|
|
|
|
|
|
|
This method is deprecated. Use exec_module() instead.
|
|
|
|
|
|
|
|
"""
|
2020-12-04 19:39:21 -04:00
|
|
|
# Warning about deprecation implemented in _load_module_shim().
|
bpo-21736: Set __file__ on frozen stdlib modules. (gh-28656)
Currently frozen modules do not have __file__ set. In their spec, origin is set to "frozen" and they are marked as not having a location. (Similarly, for frozen packages __path__ is set to an empty list.) However, for frozen stdlib modules we are able to extrapolate __file__ as long as we can determine the stdlib directory at runtime. (We now do so since gh-28586.) Having __file__ set is helpful for a number of reasons. Likewise, having a non-empty __path__ means we can import submodules of a frozen package from the filesystem (e.g. we could partially freeze the encodings module).
This change sets __file__ (and adds to __path__) for frozen stdlib modules. It uses sys._stdlibdir (from gh-28586) and the frozen module alias information (from gh-28655). All that work is done in FrozenImporter (in Lib/importlib/_bootstrap.py).
Also, if a frozen module is imported before importlib is bootstrapped (during interpreter initialization) then we fix up that module and its spec during the importlib bootstrapping step (i.e. imporlib._bootstrap._setup()) to match what gets set by FrozenImporter, including setting the file info (if the stdlib dir is known). To facilitate this, modules imported using PyImport_ImportFrozenModule() have __origname__ set using the frozen module alias info. __origname__ is popped off during importlib bootstrap.
(To be clear, even with this change the new code to set __file__ during fixups in imporlib._bootstrap._setup() doesn't actually get triggered yet. This is because sys._stdlibdir hasn't been set yet in interpreter initialization at the point importlib is bootstrapped. However, we do fix up such modules at that point to otherwise match the result of importing through FrozenImporter, just not the __file__ and __path__ parts. Doing so will require changes in the order in which things happen during interpreter initialization. That can be addressed separately. Once it is, the file-related fixup code from this PR will kick in.)
Here are things this change does not do:
* set __file__ for non-stdlib modules (no way of knowing the parent dir)
* set __file__ if the stdlib dir is not known (nor assume the expense of finding it)
* relatedly, set __file__ if the stdlib is in a zip file
* verify that the filename set to __file__ actually exists (too expensive)
* update __path__ for frozen packages that alias a non-package (since there is no package dir)
Other things this change skips, but we may do later:
* set __file__ on modules imported using PyImport_ImportFrozenModule()
* set co_filename when we unmarshal the frozen code object while importing the module (e.g. in FrozenImporter.exec_module()) -- this would allow tracebacks to show source lines
* implement FrozenImporter.get_filename() and FrozenImporter.get_source()
https://bugs.python.org/issue21736
2021-10-14 18:32:18 -03:00
|
|
|
module = _load_module_shim(cls, fullname)
|
|
|
|
info = _imp.find_frozen(fullname)
|
|
|
|
assert info is not None
|
|
|
|
_, ispkg, origname = info
|
|
|
|
module.__origname__ = origname
|
|
|
|
vars(module).pop('__file__', None)
|
|
|
|
if ispkg:
|
|
|
|
module.__path__ = []
|
|
|
|
cls._fix_up_module(module)
|
|
|
|
return module
|
2009-01-17 20:24:28 -04:00
|
|
|
|
2009-03-14 23:20:16 -03:00
|
|
|
@classmethod
|
|
|
|
@_requires_frozen
|
|
|
|
def get_code(cls, fullname):
|
|
|
|
"""Return the code object for the frozen module."""
|
2012-04-15 17:08:47 -03:00
|
|
|
return _imp.get_frozen_object(fullname)
|
2009-03-14 23:20:16 -03:00
|
|
|
|
|
|
|
@classmethod
|
|
|
|
@_requires_frozen
|
|
|
|
def get_source(cls, fullname):
|
|
|
|
"""Return None as frozen modules do not have source code."""
|
|
|
|
return None
|
|
|
|
|
|
|
|
@classmethod
|
|
|
|
@_requires_frozen
|
|
|
|
def is_package(cls, fullname):
|
2012-08-10 20:21:35 -03:00
|
|
|
"""Return True if the frozen module is a package."""
|
2012-04-15 17:08:47 -03:00
|
|
|
return _imp.is_frozen_package(fullname)
|
2009-03-14 23:20:16 -03:00
|
|
|
|
2009-01-17 20:24:28 -04:00
|
|
|
|
2009-03-12 19:37:06 -03:00
|
|
|
# Import itself ###############################################################
|
2009-02-02 15:19:36 -04:00
|
|
|
|
2009-03-12 19:37:06 -03:00
|
|
|
class _ImportLockContext:
|
|
|
|
|
|
|
|
"""Context manager for the import lock."""
|
|
|
|
|
|
|
|
def __enter__(self):
|
|
|
|
"""Acquire the import lock."""
|
2012-04-15 17:08:47 -03:00
|
|
|
_imp.acquire_lock()
|
2009-03-12 19:37:06 -03:00
|
|
|
|
|
|
|
def __exit__(self, exc_type, exc_value, exc_traceback):
|
|
|
|
"""Release the import lock regardless of any raised exceptions."""
|
2012-04-15 17:08:47 -03:00
|
|
|
_imp.release_lock()
|
2009-03-12 19:37:06 -03:00
|
|
|
|
2009-01-17 20:24:28 -04:00
|
|
|
|
2012-02-16 14:43:41 -04:00
|
|
|
def _resolve_name(name, package, level):
|
|
|
|
"""Resolve a relative module name to an absolute one."""
|
2012-02-25 01:48:17 -04:00
|
|
|
bits = package.rsplit('.', level - 1)
|
2012-02-24 12:20:54 -04:00
|
|
|
if len(bits) < level:
|
2019-08-03 02:46:02 -03:00
|
|
|
raise ImportError('attempted relative import beyond top-level package')
|
2012-02-24 12:20:54 -04:00
|
|
|
base = bits[0]
|
2022-10-06 20:43:16 -03:00
|
|
|
return f'{base}.{name}' if name else base
|
2012-02-16 14:43:41 -04:00
|
|
|
|
|
|
|
|
2014-01-06 23:49:04 -04:00
|
|
|
def _find_spec_legacy(finder, name, path):
|
2021-03-30 12:43:03 -03:00
|
|
|
msg = (f"{_object_name(finder)}.find_spec() not found; "
|
|
|
|
"falling back to find_module()")
|
|
|
|
_warnings.warn(msg, ImportWarning)
|
2014-01-06 23:49:04 -04:00
|
|
|
loader = finder.find_module(name, path)
|
|
|
|
if loader is None:
|
|
|
|
return None
|
|
|
|
return spec_from_loader(name, loader)
|
|
|
|
|
|
|
|
|
2013-11-22 12:05:39 -04:00
|
|
|
def _find_spec(name, path, target=None):
|
2016-07-08 15:09:35 -03:00
|
|
|
"""Find a module's spec."""
|
2016-03-24 20:40:59 -03:00
|
|
|
meta_path = sys.meta_path
|
|
|
|
if meta_path is None:
|
|
|
|
# PyImport_Cleanup() is running or has been called.
|
|
|
|
raise ImportError("sys.meta_path is None, Python is likely "
|
|
|
|
"shutting down")
|
|
|
|
|
|
|
|
if not meta_path:
|
2012-04-27 15:01:58 -03:00
|
|
|
_warnings.warn('sys.meta_path is empty', ImportWarning)
|
2016-03-24 20:40:59 -03:00
|
|
|
|
2013-11-22 12:05:39 -04:00
|
|
|
# We check sys.modules here for the reload case. While a passed-in
|
|
|
|
# target will usually indicate a reload there is no guarantee, whereas
|
|
|
|
# sys.modules provides one.
|
2013-11-01 01:22:15 -03:00
|
|
|
is_reload = name in sys.modules
|
2016-03-24 20:40:59 -03:00
|
|
|
for finder in meta_path:
|
2012-05-17 13:55:59 -03:00
|
|
|
with _ImportLockContext():
|
2013-11-22 12:05:39 -04:00
|
|
|
try:
|
|
|
|
find_spec = finder.find_spec
|
|
|
|
except AttributeError:
|
2014-01-06 23:49:04 -04:00
|
|
|
spec = _find_spec_legacy(finder, name, path)
|
|
|
|
if spec is None:
|
2013-11-22 12:05:39 -04:00
|
|
|
continue
|
2012-02-16 14:43:41 -04:00
|
|
|
else:
|
2013-11-22 12:05:39 -04:00
|
|
|
spec = find_spec(name, path, target)
|
|
|
|
if spec is not None:
|
|
|
|
# The parent import may have already imported this module.
|
|
|
|
if not is_reload and name in sys.modules:
|
|
|
|
module = sys.modules[name]
|
2013-11-01 01:22:15 -03:00
|
|
|
try:
|
2013-11-22 12:05:39 -04:00
|
|
|
__spec__ = module.__spec__
|
2013-11-01 01:22:15 -03:00
|
|
|
except AttributeError:
|
2013-11-22 12:05:39 -04:00
|
|
|
# We use the found spec since that is the one that
|
|
|
|
# we would have used if the parent module hadn't
|
|
|
|
# beaten us to the punch.
|
|
|
|
return spec
|
|
|
|
else:
|
|
|
|
if __spec__ is None:
|
|
|
|
return spec
|
|
|
|
else:
|
|
|
|
return __spec__
|
|
|
|
else:
|
|
|
|
return spec
|
2012-02-16 14:43:41 -04:00
|
|
|
else:
|
|
|
|
return None
|
2009-02-15 01:48:13 -04:00
|
|
|
|
2010-11-17 23:03:04 -04:00
|
|
|
|
2012-02-16 14:43:41 -04:00
|
|
|
def _sanity_check(name, package, level):
|
|
|
|
"""Verify arguments are "sane"."""
|
2012-02-23 19:18:48 -04:00
|
|
|
if not isinstance(name, str):
|
2022-10-06 20:43:16 -03:00
|
|
|
raise TypeError(f'module name must be str, not {type(name)}')
|
2012-02-22 19:33:05 -04:00
|
|
|
if level < 0:
|
|
|
|
raise ValueError('level must be >= 0')
|
2016-02-20 16:52:06 -04:00
|
|
|
if level > 0:
|
2012-02-23 19:18:48 -04:00
|
|
|
if not isinstance(package, str):
|
2013-07-12 12:04:23 -03:00
|
|
|
raise TypeError('__package__ not set to a string')
|
2016-02-20 16:59:36 -04:00
|
|
|
elif not package:
|
|
|
|
raise ImportError('attempted relative import with no known parent '
|
|
|
|
'package')
|
2009-02-06 21:15:27 -04:00
|
|
|
if not name and level == 0:
|
2013-07-12 12:04:23 -03:00
|
|
|
raise ValueError('Empty module name')
|
2012-02-16 14:43:41 -04:00
|
|
|
|
|
|
|
|
2013-07-06 15:48:18 -03:00
|
|
|
_ERR_MSG_PREFIX = 'No module named '
|
|
|
|
_ERR_MSG = _ERR_MSG_PREFIX + '{!r}'
|
2012-02-16 14:43:41 -04:00
|
|
|
|
2012-05-17 13:55:59 -03:00
|
|
|
def _find_and_load_unlocked(name, import_):
|
2012-02-16 14:43:41 -04:00
|
|
|
path = None
|
|
|
|
parent = name.rpartition('.')[0]
|
2021-07-24 19:44:46 -03:00
|
|
|
parent_spec = None
|
2012-02-16 14:43:41 -04:00
|
|
|
if parent:
|
|
|
|
if parent not in sys.modules:
|
2012-07-31 08:14:18 -03:00
|
|
|
_call_with_frames_removed(import_, parent)
|
2012-04-02 21:33:56 -03:00
|
|
|
# Crazy side-effects!
|
|
|
|
if name in sys.modules:
|
|
|
|
return sys.modules[name]
|
2012-02-16 14:43:41 -04:00
|
|
|
parent_module = sys.modules[parent]
|
|
|
|
try:
|
|
|
|
path = parent_module.__path__
|
|
|
|
except AttributeError:
|
2022-10-06 20:43:16 -03:00
|
|
|
msg = f'{_ERR_MSG_PREFIX} {name!r}; {parent!r} is not a package'
|
2016-09-07 20:56:15 -03:00
|
|
|
raise ModuleNotFoundError(msg, name=name) from None
|
2021-07-24 19:44:46 -03:00
|
|
|
parent_spec = parent_module.__spec__
|
|
|
|
child = name.rpartition('.')[2]
|
2013-11-22 12:05:39 -04:00
|
|
|
spec = _find_spec(name, path)
|
|
|
|
if spec is None:
|
2022-10-06 20:43:16 -03:00
|
|
|
raise ModuleNotFoundError(f'{_ERR_MSG_PREFIX}{name!r}', name=name)
|
2013-11-22 12:05:39 -04:00
|
|
|
else:
|
2021-07-24 19:44:46 -03:00
|
|
|
if parent_spec:
|
|
|
|
# Temporarily add child we are currently importing to parent's
|
|
|
|
# _uninitialized_submodules for circular import tracking.
|
|
|
|
parent_spec._uninitialized_submodules.append(child)
|
|
|
|
try:
|
|
|
|
module = _load_unlocked(spec)
|
|
|
|
finally:
|
|
|
|
if parent_spec:
|
|
|
|
parent_spec._uninitialized_submodules.pop()
|
2012-02-22 19:33:05 -04:00
|
|
|
if parent:
|
|
|
|
# Set the module as an attribute on its parent.
|
|
|
|
parent_module = sys.modules[parent]
|
2020-01-22 20:42:38 -04:00
|
|
|
try:
|
|
|
|
setattr(parent_module, child, module)
|
|
|
|
except AttributeError:
|
|
|
|
msg = f"Cannot set an attribute on {parent!r} for child module {child!r}"
|
|
|
|
_warnings.warn(msg, ImportWarning)
|
2012-02-22 19:33:05 -04:00
|
|
|
return module
|
2012-02-16 14:43:41 -04:00
|
|
|
|
|
|
|
|
2017-07-21 08:00:46 -03:00
|
|
|
_NEEDS_LOADING = object()
|
|
|
|
|
|
|
|
|
2012-05-17 13:55:59 -03:00
|
|
|
def _find_and_load(name, import_):
|
2017-07-06 02:09:03 -03:00
|
|
|
"""Find and load the module."""
|
2021-08-12 15:23:29 -03:00
|
|
|
|
|
|
|
# Optimization: we avoid unneeded module locking if the module
|
|
|
|
# already exists in sys.modules and is fully initialized.
|
|
|
|
module = sys.modules.get(name, _NEEDS_LOADING)
|
|
|
|
if (module is _NEEDS_LOADING or
|
|
|
|
getattr(getattr(module, "__spec__", None), "_initializing", False)):
|
|
|
|
with _ModuleLockManager(name):
|
|
|
|
module = sys.modules.get(name, _NEEDS_LOADING)
|
|
|
|
if module is _NEEDS_LOADING:
|
|
|
|
return _find_and_load_unlocked(name, import_)
|
|
|
|
|
|
|
|
# Optimization: only call _bootstrap._lock_unlock_module() if
|
|
|
|
# module.__spec__._initializing is True.
|
|
|
|
# NOTE: because of this, initializing must be set *before*
|
|
|
|
# putting the new module in sys.modules.
|
|
|
|
_lock_unlock_module(name)
|
2017-07-10 17:52:32 -03:00
|
|
|
|
2017-07-06 02:09:03 -03:00
|
|
|
if module is None:
|
2022-10-06 20:43:16 -03:00
|
|
|
message = f'import of {name} halted; None in sys.modules'
|
2017-07-06 02:09:03 -03:00
|
|
|
raise ModuleNotFoundError(message, name=name)
|
2017-07-10 17:52:32 -03:00
|
|
|
|
2017-07-06 02:09:03 -03:00
|
|
|
return module
|
2012-05-17 13:55:59 -03:00
|
|
|
|
|
|
|
|
2012-02-16 14:43:41 -04:00
|
|
|
def _gcd_import(name, package=None, level=0):
|
|
|
|
"""Import and return the module based on its name, the package the call is
|
|
|
|
being made from, and the level adjustment.
|
|
|
|
|
|
|
|
This function represents the greatest common denominator of functionality
|
|
|
|
between import_module and __import__. This includes setting __package__ if
|
|
|
|
the loader did not.
|
|
|
|
|
|
|
|
"""
|
|
|
|
_sanity_check(name, package, level)
|
2009-02-05 22:47:33 -04:00
|
|
|
if level > 0:
|
2012-02-16 14:43:41 -04:00
|
|
|
name = _resolve_name(name, package, level)
|
2017-07-06 02:09:03 -03:00
|
|
|
return _find_and_load(name, _gcd_import)
|
2009-02-05 22:47:33 -04:00
|
|
|
|
2016-09-07 20:56:15 -03:00
|
|
|
|
2017-10-26 04:41:59 -03:00
|
|
|
def _handle_fromlist(module, fromlist, import_, *, recursive=False):
|
2012-02-16 14:43:41 -04:00
|
|
|
"""Figure out what __import__ should return.
|
2009-01-17 20:24:28 -04:00
|
|
|
|
2012-02-16 14:43:41 -04:00
|
|
|
The import_ parameter is a callable which takes the name of module to
|
|
|
|
import. It is required to decouple the function from assuming importlib's
|
|
|
|
import implementation is desired.
|
2009-01-17 20:24:28 -04:00
|
|
|
|
|
|
|
"""
|
2009-02-06 21:15:27 -04:00
|
|
|
# The hell that is fromlist ...
|
2012-02-23 21:47:57 -04:00
|
|
|
# If a package was imported, try to import stuff from fromlist.
|
2018-03-11 05:52:37 -03:00
|
|
|
for x in fromlist:
|
|
|
|
if not isinstance(x, str):
|
|
|
|
if recursive:
|
|
|
|
where = module.__name__ + '.__all__'
|
|
|
|
else:
|
|
|
|
where = "``from list''"
|
|
|
|
raise TypeError(f"Item in {where} must be str, "
|
|
|
|
f"not {type(x).__name__}")
|
|
|
|
elif x == '*':
|
|
|
|
if not recursive and hasattr(module, '__all__'):
|
|
|
|
_handle_fromlist(module, module.__all__, import_,
|
|
|
|
recursive=True)
|
|
|
|
elif not hasattr(module, x):
|
2022-10-06 20:43:16 -03:00
|
|
|
from_name = f'{module.__name__}.{x}'
|
2018-03-11 05:52:37 -03:00
|
|
|
try:
|
|
|
|
_call_with_frames_removed(import_, from_name)
|
|
|
|
except ModuleNotFoundError as exc:
|
|
|
|
# Backwards-compatibility dictates we ignore failed
|
|
|
|
# imports triggered by fromlist for modules that don't
|
|
|
|
# exist.
|
|
|
|
if (exc.name == from_name and
|
|
|
|
sys.modules.get(from_name, _NEEDS_LOADING) is not None):
|
|
|
|
continue
|
|
|
|
raise
|
2012-02-23 21:47:57 -04:00
|
|
|
return module
|
2012-02-08 19:50:22 -04:00
|
|
|
|
|
|
|
|
2012-02-16 14:43:41 -04:00
|
|
|
def _calc___package__(globals):
|
|
|
|
"""Calculate what __package__ should be.
|
|
|
|
|
|
|
|
__package__ is not guaranteed to be defined or could be set to None
|
|
|
|
to represent that its proper value is unknown.
|
|
|
|
|
|
|
|
"""
|
2016-01-22 19:25:50 -04:00
|
|
|
package = globals.get('__package__')
|
2016-01-15 17:33:03 -04:00
|
|
|
spec = globals.get('__spec__')
|
2016-01-22 19:25:50 -04:00
|
|
|
if package is not None:
|
|
|
|
if spec is not None and package != spec.parent:
|
|
|
|
_warnings.warn("__package__ != __spec__.parent "
|
|
|
|
f"({package!r} != {spec.parent!r})",
|
2022-10-05 19:00:45 -03:00
|
|
|
DeprecationWarning, stacklevel=3)
|
2016-01-22 19:25:50 -04:00
|
|
|
return package
|
|
|
|
elif spec is not None:
|
2016-01-15 17:33:03 -04:00
|
|
|
return spec.parent
|
2016-01-22 19:25:50 -04:00
|
|
|
else:
|
2016-01-15 17:33:03 -04:00
|
|
|
_warnings.warn("can't resolve package from __spec__ or __package__, "
|
|
|
|
"falling back on __name__ and __path__",
|
|
|
|
ImportWarning, stacklevel=3)
|
2012-02-16 14:43:41 -04:00
|
|
|
package = globals['__name__']
|
|
|
|
if '__path__' not in globals:
|
|
|
|
package = package.rpartition('.')[0]
|
|
|
|
return package
|
|
|
|
|
|
|
|
|
2012-08-06 17:34:44 -03:00
|
|
|
def __import__(name, globals=None, locals=None, fromlist=(), level=0):
|
2012-02-16 14:43:41 -04:00
|
|
|
"""Import a module.
|
|
|
|
|
2016-05-26 02:35:26 -03:00
|
|
|
The 'globals' argument is used to infer where the import is occurring from
|
2012-02-16 14:43:41 -04:00
|
|
|
to handle relative imports. The 'locals' argument is ignored. The
|
|
|
|
'fromlist' argument specifies what should exist as attributes on the module
|
|
|
|
being imported (e.g. ``from module import <fromlist>``). The 'level'
|
|
|
|
argument represents the package location to import from in a relative
|
|
|
|
import (e.g. ``from ..pkg import mod`` would have a 'level' of 2).
|
|
|
|
|
|
|
|
"""
|
|
|
|
if level == 0:
|
|
|
|
module = _gcd_import(name)
|
|
|
|
else:
|
2012-08-06 17:34:44 -03:00
|
|
|
globals_ = globals if globals is not None else {}
|
|
|
|
package = _calc___package__(globals_)
|
2012-02-16 14:43:41 -04:00
|
|
|
module = _gcd_import(name, package, level)
|
2012-02-23 21:47:57 -04:00
|
|
|
if not fromlist:
|
|
|
|
# Return up to the first dot in 'name'. This is complicated by the fact
|
|
|
|
# that 'name' may be relative.
|
|
|
|
if level == 0:
|
2012-05-07 16:41:59 -03:00
|
|
|
return _gcd_import(name.partition('.')[0])
|
2012-02-23 21:47:57 -04:00
|
|
|
elif not name:
|
|
|
|
return module
|
|
|
|
else:
|
2012-09-28 17:41:39 -03:00
|
|
|
# Figure out where to slice the module's name up to the first dot
|
|
|
|
# in 'name'.
|
2012-02-23 21:47:57 -04:00
|
|
|
cut_off = len(name) - len(name.partition('.')[0])
|
2012-09-28 17:41:39 -03:00
|
|
|
# Slice end needs to be positive to alleviate need to special-case
|
|
|
|
# when ``'.' not in name``.
|
2012-04-14 22:50:00 -03:00
|
|
|
return sys.modules[module.__name__[:len(module.__name__)-cut_off]]
|
2018-03-11 05:52:37 -03:00
|
|
|
elif hasattr(module, '__path__'):
|
2012-02-23 21:47:57 -04:00
|
|
|
return _handle_fromlist(module, fromlist, _gcd_import)
|
2018-03-11 05:52:37 -03:00
|
|
|
else:
|
|
|
|
return module
|
2012-02-16 14:43:41 -04:00
|
|
|
|
|
|
|
|
2013-11-22 12:05:39 -04:00
|
|
|
def _builtin_from_name(name):
|
|
|
|
spec = BuiltinImporter.find_spec(name)
|
2013-11-22 12:58:17 -04:00
|
|
|
if spec is None:
|
|
|
|
raise ImportError('no built-in module named ' + name)
|
2014-05-30 15:55:29 -03:00
|
|
|
return _load_unlocked(spec)
|
2013-11-22 12:05:39 -04:00
|
|
|
|
2012-04-20 19:04:03 -03:00
|
|
|
|
2012-04-15 17:08:47 -03:00
|
|
|
def _setup(sys_module, _imp_module):
|
2012-02-08 19:50:22 -04:00
|
|
|
"""Setup importlib by importing needed built-in modules and injecting them
|
|
|
|
into the global namespace.
|
|
|
|
|
2012-04-15 17:08:47 -03:00
|
|
|
As sys is needed for sys.modules access and _imp is needed to load built-in
|
2012-02-14 19:38:11 -04:00
|
|
|
modules, those two modules must be explicitly passed in.
|
2012-02-08 19:50:22 -04:00
|
|
|
|
|
|
|
"""
|
2015-04-13 15:21:02 -03:00
|
|
|
global _imp, sys
|
2012-04-15 17:08:47 -03:00
|
|
|
_imp = _imp_module
|
2012-02-08 19:50:22 -04:00
|
|
|
sys = sys_module
|
|
|
|
|
2013-11-22 12:05:39 -04:00
|
|
|
# Set up the spec for existing builtin/frozen modules.
|
2013-02-01 15:04:12 -04:00
|
|
|
module_type = type(sys)
|
2013-02-01 16:31:49 -04:00
|
|
|
for name, module in sys.modules.items():
|
2013-02-01 15:04:12 -04:00
|
|
|
if isinstance(module, module_type):
|
2013-11-22 12:05:39 -04:00
|
|
|
if name in sys.builtin_module_names:
|
|
|
|
loader = BuiltinImporter
|
|
|
|
elif _imp.is_frozen(name):
|
|
|
|
loader = FrozenImporter
|
|
|
|
else:
|
|
|
|
continue
|
|
|
|
spec = _spec_from_module(module, loader)
|
2014-05-30 15:55:29 -03:00
|
|
|
_init_module_attrs(spec, module)
|
2021-10-05 14:26:37 -03:00
|
|
|
if loader is FrozenImporter:
|
bpo-21736: Set __file__ on frozen stdlib modules. (gh-28656)
Currently frozen modules do not have __file__ set. In their spec, origin is set to "frozen" and they are marked as not having a location. (Similarly, for frozen packages __path__ is set to an empty list.) However, for frozen stdlib modules we are able to extrapolate __file__ as long as we can determine the stdlib directory at runtime. (We now do so since gh-28586.) Having __file__ set is helpful for a number of reasons. Likewise, having a non-empty __path__ means we can import submodules of a frozen package from the filesystem (e.g. we could partially freeze the encodings module).
This change sets __file__ (and adds to __path__) for frozen stdlib modules. It uses sys._stdlibdir (from gh-28586) and the frozen module alias information (from gh-28655). All that work is done in FrozenImporter (in Lib/importlib/_bootstrap.py).
Also, if a frozen module is imported before importlib is bootstrapped (during interpreter initialization) then we fix up that module and its spec during the importlib bootstrapping step (i.e. imporlib._bootstrap._setup()) to match what gets set by FrozenImporter, including setting the file info (if the stdlib dir is known). To facilitate this, modules imported using PyImport_ImportFrozenModule() have __origname__ set using the frozen module alias info. __origname__ is popped off during importlib bootstrap.
(To be clear, even with this change the new code to set __file__ during fixups in imporlib._bootstrap._setup() doesn't actually get triggered yet. This is because sys._stdlibdir hasn't been set yet in interpreter initialization at the point importlib is bootstrapped. However, we do fix up such modules at that point to otherwise match the result of importing through FrozenImporter, just not the __file__ and __path__ parts. Doing so will require changes in the order in which things happen during interpreter initialization. That can be addressed separately. Once it is, the file-related fixup code from this PR will kick in.)
Here are things this change does not do:
* set __file__ for non-stdlib modules (no way of knowing the parent dir)
* set __file__ if the stdlib dir is not known (nor assume the expense of finding it)
* relatedly, set __file__ if the stdlib is in a zip file
* verify that the filename set to __file__ actually exists (too expensive)
* update __path__ for frozen packages that alias a non-package (since there is no package dir)
Other things this change skips, but we may do later:
* set __file__ on modules imported using PyImport_ImportFrozenModule()
* set co_filename when we unmarshal the frozen code object while importing the module (e.g. in FrozenImporter.exec_module()) -- this would allow tracebacks to show source lines
* implement FrozenImporter.get_filename() and FrozenImporter.get_source()
https://bugs.python.org/issue21736
2021-10-14 18:32:18 -03:00
|
|
|
loader._fix_up_module(module)
|
2012-02-08 19:50:22 -04:00
|
|
|
|
2013-11-22 12:05:39 -04:00
|
|
|
# Directly load built-in modules needed during bootstrap.
|
2012-02-08 19:50:22 -04:00
|
|
|
self_module = sys.modules[__name__]
|
2017-09-18 18:50:44 -03:00
|
|
|
for builtin_name in ('_thread', '_warnings', '_weakref'):
|
2012-02-08 19:50:22 -04:00
|
|
|
if builtin_name not in sys.modules:
|
2013-11-22 12:05:39 -04:00
|
|
|
builtin_module = _builtin_from_name(builtin_name)
|
2012-02-08 19:50:22 -04:00
|
|
|
else:
|
|
|
|
builtin_module = sys.modules[builtin_name]
|
|
|
|
setattr(self_module, builtin_name, builtin_module)
|
|
|
|
|
2012-02-14 19:38:11 -04:00
|
|
|
|
2012-04-15 17:08:47 -03:00
|
|
|
def _install(sys_module, _imp_module):
|
2017-05-23 01:36:03 -03:00
|
|
|
"""Install importers for builtin and frozen modules"""
|
2012-04-15 17:08:47 -03:00
|
|
|
_setup(sys_module, _imp_module)
|
2015-05-02 22:15:18 -03:00
|
|
|
|
2012-07-28 16:33:05 -03:00
|
|
|
sys.meta_path.append(BuiltinImporter)
|
|
|
|
sys.meta_path.append(FrozenImporter)
|
2015-05-02 22:15:18 -03:00
|
|
|
|
2017-05-23 01:36:03 -03:00
|
|
|
|
|
|
|
def _install_external_importers():
|
|
|
|
"""Install importers that require external filesystem access"""
|
2015-05-16 00:54:59 -03:00
|
|
|
global _bootstrap_external
|
2015-05-02 22:15:18 -03:00
|
|
|
import _frozen_importlib_external
|
2015-05-16 00:54:59 -03:00
|
|
|
_bootstrap_external = _frozen_importlib_external
|
2015-05-02 22:15:18 -03:00
|
|
|
_frozen_importlib_external._install(sys.modules[__name__])
|