Use default toml file for those without poetry item

waketzheng · Jul 11, 2024 · e30bb07 · e30bb07
1 parent fd00c4d
commit e30bb07
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/fast_dev_cli/cli.py b/fast_dev_cli/cli.py
@@ -160,9 +160,13 @@ def parse_filename() -> str:
                     continue
                 return line.split('path = "', 1)[-1].split('"')[0]
         context = tomllib.loads(toml_text)
-        if (poetry_item := context["tool"]["poetry"])["version"] == "0":
+        try:
+            version_value = context["tool"]["poetry"]["version"]
+        except KeyError:
+            return TOML_FILE
+        if version_value == "0":
             try:
-                package_item = poetry_item["packages"]
+                package_item = context["tool"]["poetry"]["packages"]
             except KeyError:
                 packages = []
             else: