fix

sbdchd · Oct 17, 2023 · fdffbe6 · fdffbe6
1 parent a7d591e
commit fdffbe6
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 11 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -27,6 +27,7 @@ types-PyYAML = "^6.0.1"
 types-pytz = "^2021.3.0"
 psycopg2-binary = "^2.9.3"
 ruff = "^0.0.286"
+tomlkit = "^0.12.1"
 
 [tool.black]
 line-length = 88

diff --git a/s/update-version b/s/update-version
@@ -5,14 +5,7 @@
 from pathlib import Path
 import re
 import argparse
-import tomllib
-
-
-def find_version() -> str:
-    pyproject = Path(".").parent.parent / "pyproject.toml"
-    pyproject_toml = tomllib.loads(pyproject)
-    return pyproject_toml["tool"]["poetry"]["version"]
-
+import tomlkit
 
 def main():
     parser = argparse.ArgumentParser()
@@ -22,9 +15,11 @@ def main():
     pyproject_path = Path(".").parent.parent / "pyproject.toml"
     pyproject = pyproject_path.read_text()
 
-    current_version = tomllib.loads(pyproject)["tool"]["poetry"]["version"]
+    current_version = tomlkit.loads(pyproject)["tool"]["poetry"]["version"]
 
     major, minor, patch = [int(x) for x in current_version.split(".")]
+
+
     match args.version_increment:
         case "major":
             major += 1