Robustify version parsing, restore pypi version guard

Author: elprans

.github/workflows/publish.yml

@@ -48,10 +48,30 @@ jobs:
             exit 1
           fi
 
+      - name: Check if version already published
+        id: pypi
+        env:
+          PACKAGE_NAME: ${{ steps.package.outputs.name }}
+        run: |
+          VERSION=$(python scripts/get-version.py "$PACKAGE_NAME")
+          HTTP_STATUS=$(curl -s -o /dev/null -w "%{http_code}" "https://pypi.org/pypi/${PACKAGE_NAME}/${VERSION}/json")
+          if [ "$HTTP_STATUS" = "200" ]; then
+            echo "$PACKAGE_NAME $VERSION already exists on PyPI, skipping publish"
+            echo "should-publish=false" >> "$GITHUB_OUTPUT"
+          elif [ "$HTTP_STATUS" = "404" ]; then
+            echo "$PACKAGE_NAME $VERSION not found on PyPI, proceeding with publish"
+            echo "should-publish=true" >> "$GITHUB_OUTPUT"
+          else
+            echo "Unexpected PyPI response for $PACKAGE_NAME $VERSION: $HTTP_STATUS"
+            exit 1
+          fi
+
       - name: Build package
+        if: steps.pypi.outputs.should-publish == 'true'
         env:
           PACKAGE_NAME: ${{ steps.package.outputs.name }}
         run: uv build --package "$PACKAGE_NAME" --no-sources
 
       - name: Publish package
+        if: steps.pypi.outputs.should-publish == 'true'
         run: uv publish

scripts/get-version.py

@@ -11,12 +11,21 @@
 def read_version(path: Path) -> str:
     module = ast.parse(path.read_text(encoding="utf-8"), filename=str(path))
     for node in module.body:
-        if not isinstance(node, ast.Assign):
+        value: ast.expr | None
+        if isinstance(node, ast.Assign) and any(
+            isinstance(target, ast.Name) and target.id == "__version__" for target in node.targets
+        ):
+            value = node.value
+        elif (
+            isinstance(node, ast.AnnAssign)
+            and isinstance(node.target, ast.Name)
+            and node.target.id == "__version__"
+        ):
+            value = node.value
+        else:
             continue
-        if not any(isinstance(target, ast.Name) and target.id == "__version__" for target in node.targets):
-            continue
-        if isinstance(node.value, ast.Constant) and isinstance(node.value.value, str):
-            return node.value.value
+        if isinstance(value, ast.Constant) and isinstance(value.value, str):
+            return value.value
     raise RuntimeError(f"could not find __version__ in {path}")