Support type-hinted python version string

uclahs-cds · Oct 2, 2024 · 5269c31 · 5269c31
1 parent b5a2e22
commit 5269c31
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/bumpchanges/updatefiles.py b/bumpchanges/updatefiles.py
@@ -19,6 +19,7 @@
         version             # Literal `version`
         (?:__)?             # Optional literal `__`
         (?P=vquote)         # `'`, `"`, or nothing (back-reference to `vquote`)
+        (?:\s*:\s*str\b)?   # Optional literal `: str` (for python typing)
         (?:
             \s*             # Any whitespace
             [=:]?           # `=`, `:`, or nothing

diff --git a/tests/test_versionfiles.py b/tests/test_versionfiles.py
@@ -24,6 +24,11 @@
         """__version__ = 1.1.0""",
         """__version__ = 2.3.4""",
     ),
+    (
+        # Python-style with typing
+        """__version__: str = '1.1.0'""",
+        """__version__: str = '2.3.4'""",
+    ),
     (
         # Plain string version
         '''version = "1.1.0"''',