Handle distribution.files being None

python-poetry · Oct 24, 2022 · ff0cb20 · ff0cb20
1 parent 5c578a2
commit ff0cb20
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/poetry/utils/env.py b/src/poetry/utils/env.py
@@ -328,8 +328,8 @@ def find_distribution_files_with_suffix(
         for distribution in self.distributions(
             name=distribution_name, writable_only=writable_only
         ):
-            assert distribution.files is not None
-            for file in distribution.files:
+            files = [] if distribution.files is None else distribution.files
+            for file in files:
                 if file.name.endswith(suffix):
                     yield Path(
                         distribution.locate_file(file),  # type: ignore[no-untyped-call]
@@ -341,8 +341,8 @@ def find_distribution_files_with_name(
         for distribution in self.distributions(
             name=distribution_name, writable_only=writable_only
         ):
-            assert distribution.files is not None
-            for file in distribution.files:
+            files = [] if distribution.files is None else distribution.files
+            for file in files:
                 if file.name == name:
                     yield Path(
                         distribution.locate_file(file),  # type: ignore[no-untyped-call]
@@ -372,8 +372,8 @@ def remove_distribution_files(self, distribution_name: str) -> list[Path]:
         for distribution in self.distributions(
             name=distribution_name, writable_only=True
         ):
-            assert distribution.files is not None
-            for file in distribution.files:
+            files = [] if distribution.files is None else distribution.files
+            for file in files:
                 path = Path(
                     distribution.locate_file(file),  # type: ignore[no-untyped-call]
                 )