Upgrade to newest tomlkit, fix type issues

Author: amyreese

dowsing/maturin.py

@@ -25,7 +25,8 @@ def get_metadata(self) -> Distribution:
 
         cargo = self.path / "Cargo.toml"
         doc = tomlkit.parse(cargo.read_text())
-        for k, v in doc["package"].items():
+        package = doc.get("package", {})
+        for k, v in package.items():
             if k == "name":
                 d.name = v
             elif k == "version":
@@ -39,7 +40,8 @@ def get_metadata(self) -> Distribution:
             # homepage
             # readme (filename)
 
-        for k, v in doc["package"]["metadata"]["maturin"].items():
+        maturin = package.get("metadata", {}).get("maturin", {})
+        for k, v in maturin.items():
             if k == "requires-python":
                 d.requires_python = v
             elif k == "classifier":

dowsing/pep517.py

@@ -26,13 +26,14 @@ def get_backend(path: Path) -> Tuple[List[str], BaseReader]:
     requires: List[str] = []
     if pyproject.exists():
         doc = tomlkit.parse(pyproject.read_text())
-        if "build-system" in doc:
-            # 1b. include any build-system requires
-            if "requires" in doc["build-system"]:
-                requires.extend(doc["build-system"]["requires"])
-            if "build-backend" in doc["build-system"]:
-                backend = doc["build-system"]["build-backend"]
-            # TODO backend-path
+        table = doc.get("build-system", {})
+
+        # 1b. include any build-system requires
+        if "requires" in table:
+            requires.extend(table["requires"])
+        if "build-backend" in table:
+            backend = table["build-backend"]
+        # TODO backend-path
 
     try:
         backend_path = KNOWN_BACKENDS[backend]

dowsing/poetry.py

@@ -42,7 +42,8 @@ def get_metadata(self) -> Distribution:
         d.packages = []
         d.packages_dict = {}
 
-        for k, v in doc["tool"]["poetry"].items():
+        poetry = doc.get("tool", {}).get("poetry", {})
+        for k, v in poetry.items():
             if k in ("homepage", "repository", "documentation"):
                 d.project_urls[k] = v
             elif k == "packages":
@@ -64,16 +65,16 @@ def get_metadata(self) -> Distribution:
                 d.packages_dict[p] = p.replace(".", "/")
                 d.packages.append(p)
 
-        for k, v in doc["tool"]["poetry"].get("dependencies", {}).items():
+        for k, v in poetry.get("dependencies", {}).items():
             if k == "python":
                 pass  # TODO translate to requires_python
             else:
                 d.requires_dist.append(k)  # TODO something with version
 
-        for k, v in doc["tool"]["poetry"].get("urls", {}).items():
+        for k, v in poetry.get("urls", {}).items():
             d.project_urls[k] = v
 
-        for k, v in doc["tool"]["poetry"].get("scripts", {}).items():
+        for k, v in poetry.get("scripts", {}).items():
             d.entry_points[k] = v
 
         d.source_mapping = d._source_mapping(self.path)

requirements.txt

@@ -1,5 +1,5 @@
 highlighter==0.1.1
 imperfect==0.3.0
 LibCST==0.3.12
-tomlkit==0.7.2
+tomlkit==0.8.0
 pkginfo==1.8.1