Ruud
10 years ago
20 changed files with 523 additions and 215 deletions
@ -1,3 +1,107 @@ |
|||
""" |
|||
Copyright (c) Donald Stufft, pip, and individual contributors |
|||
|
|||
Permission is hereby granted, free of charge, to any person obtaining |
|||
a copy of this software and associated documentation files (the |
|||
"Software"), to deal in the Software without restriction, including |
|||
without limitation the rights to use, copy, modify, merge, publish, |
|||
distribute, sublicense, and/or sell copies of the Software, and to |
|||
permit persons to whom the Software is furnished to do so, subject to |
|||
the following conditions: |
|||
|
|||
The above copyright notice and this permission notice shall be |
|||
included in all copies or substantial portions of the Software. |
|||
|
|||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, |
|||
EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF |
|||
MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND |
|||
NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE |
|||
LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION |
|||
OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION |
|||
WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. |
|||
""" |
|||
from __future__ import absolute_import |
|||
|
|||
from . import urllib3 |
|||
import sys |
|||
|
|||
|
|||
class VendorAlias(object): |
|||
|
|||
def __init__(self, package_names): |
|||
self._package_names = package_names |
|||
self._vendor_name = __name__ |
|||
self._vendor_pkg = self._vendor_name + "." |
|||
self._vendor_pkgs = [ |
|||
self._vendor_pkg + name for name in self._package_names |
|||
] |
|||
|
|||
def find_module(self, fullname, path=None): |
|||
if fullname.startswith(self._vendor_pkg): |
|||
return self |
|||
|
|||
def load_module(self, name): |
|||
# Ensure that this only works for the vendored name |
|||
if not name.startswith(self._vendor_pkg): |
|||
raise ImportError( |
|||
"Cannot import %s, must be a subpackage of '%s'." % ( |
|||
name, self._vendor_name, |
|||
) |
|||
) |
|||
|
|||
if not (name == self._vendor_name or |
|||
any(name.startswith(pkg) for pkg in self._vendor_pkgs)): |
|||
raise ImportError( |
|||
"Cannot import %s, must be one of %s." % ( |
|||
name, self._vendor_pkgs |
|||
) |
|||
) |
|||
|
|||
# Check to see if we already have this item in sys.modules, if we do |
|||
# then simply return that. |
|||
if name in sys.modules: |
|||
return sys.modules[name] |
|||
|
|||
# Check to see if we can import the vendor name |
|||
try: |
|||
# We do this dance here because we want to try and import this |
|||
# module without hitting a recursion error because of a bunch of |
|||
# VendorAlias instances on sys.meta_path |
|||
real_meta_path = sys.meta_path[:] |
|||
try: |
|||
sys.meta_path = [ |
|||
m for m in sys.meta_path |
|||
if not isinstance(m, VendorAlias) |
|||
] |
|||
__import__(name) |
|||
module = sys.modules[name] |
|||
finally: |
|||
# Re-add any additions to sys.meta_path that were made while |
|||
# during the import we just did, otherwise things like |
|||
# requests.packages.urllib3.poolmanager will fail. |
|||
for m in sys.meta_path: |
|||
if m not in real_meta_path: |
|||
real_meta_path.append(m) |
|||
|
|||
# Restore sys.meta_path with any new items. |
|||
sys.meta_path = real_meta_path |
|||
except ImportError: |
|||
# We can't import the vendor name, so we'll try to import the |
|||
# "real" name. |
|||
real_name = name[len(self._vendor_pkg):] |
|||
try: |
|||
__import__(real_name) |
|||
module = sys.modules[real_name] |
|||
except ImportError: |
|||
raise ImportError("No module named '%s'" % (name,)) |
|||
|
|||
# If we've gotten here we've found the module we're looking for, either |
|||
# as part of our vendored package, or as the real name, so we'll add |
|||
# it to sys.modules as the vendored name so that we don't have to do |
|||
# the lookup again. |
|||
sys.modules[name] = module |
|||
|
|||
# Finally, return the loaded module |
|||
return module |
|||
|
|||
|
|||
sys.meta_path.append(VendorAlias(["urllib3", "chardet"])) |
|||
|
|||
Loading…
Reference in new issue