Allow importing wheel files. Close #541

pylint-dev · Jul 11, 2019 · 334a67d · 334a67d
1 parent 28fc86f
commit 334a67d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/ChangeLog b/ChangeLog
@@ -6,6 +6,8 @@ What's New in astroid 2.3.0?
 ============================
 Release Date: TBA
 
+* Allow importing wheel files. Close #541
+
 * Annotated AST follows PEP8 coding style when converted to string.
 
 * Fix a bug where defining a class using type() could cause a DuplicateBasesError.

diff --git a/astroid/manager.py b/astroid/manager.py
@@ -26,6 +26,9 @@
 from astroid import transforms
 
 
+ZIP_IMPORT_EXTS = (".zip", ".egg", ".whl")
+
+
 def safe_repr(obj):
     try:
         return repr(obj)
@@ -185,7 +188,7 @@ def zip_import_data(self, filepath):
         from astroid.builder import AstroidBuilder
 
         builder = AstroidBuilder(self)
-        for ext in (".zip", ".egg"):
+        for ext in ZIP_IMPORT_EXTS:
             try:
                 eggpath, resource = filepath.rsplit(ext + os.path.sep, 1)
             except ValueError: