fix to handle optional dev-dependencies (#129)

apoclyps · Dec 15, 2022 · 78bf12f · 78bf12f
1 parent d73eee5
commit 78bf12f
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/deps/pipenv/resolver.py b/deps/pipenv/resolver.py
@@ -88,12 +88,15 @@ def extract_package_versions(self, repo: str, content: str | None) -> dict:
         if not content:
             return self.versions_by_service
 
-        data = toml.loads(content)
+        data: dict = toml.loads(content)
+        poetry_sections: dict = data.get("tool", {}).get("poetry", {})
 
-        for name, version in data["tool"]["poetry"]["dependencies"].items():
+        for name, version in poetry_sections.get("dependencies", {}).items():
+            version = version.replace("=", "")
             self._add_version_by_service(repo=repo, name=name, version=version)
 
-        for name, version in data["tool"]["poetry"]["dev-dependencies"].items():
+        for name, version in poetry_sections.get("dev-dependencies", {}).items():
+            version = version.replace("=", "")
             self._add_version_by_service(repo=repo, name=name, version=version)
 
         return self.versions_by_service