Improve pdm_build.py (#273)

* Improve `pdm_build.py`

Signed-off-by: Yu Ishikawa <[email protected]>

* Add a nox test

Signed-off-by: Yu Ishikawa <[email protected]>

---------

Signed-off-by: Yu Ishikawa <[email protected]>

Author: yu-iskw

noxfile.py

@@ -50,3 +50,27 @@ def run_cli_deprecations(session):
         env={"UV_PROJECT_ENVIRONMENT": session.virtualenv.location},
     )
     session.run("dbt-autofix", "deprecations")
+
+
+@nox.session(python=["3.10", "3.11", "3.12", "3.13"], venv_backend="uv")
+def test_pre_commit_installation(session):
+    """Test that dbt-autofix can be installed as a pre-commit hook"""
+    session.run_install(
+        "uv",
+        "sync",
+        "--extra=test",
+        f"--python={session.virtualenv.location}",
+        env={"UV_PROJECT_ENVIRONMENT": session.virtualenv.location},
+    )
+    # Use try-repo with a non-existent file to test installation without execution.
+    # This avoids the requirement for a dbt_project.yml file while still
+    # triggering the pdm_build.py logic we want to verify.
+    session.run(
+        "pre-commit",
+        "try-repo",
+        ".",
+        "dbt-autofix-check",
+        "--files",
+        "non_existent_file",
+        "--verbose",
+    )

pdm_build.py

@@ -1,3 +1,5 @@
+import sys
+
 from pdm.backend.hooks.version import SCMVersion
 
 
@@ -14,6 +16,32 @@ def format_version(version: SCMVersion) -> str:
 def pdm_build_initialize(context):
     metadata = context.config.metadata
     if metadata["name"] == "dbt-autofix" and "dbt-fusion-package-tools" in metadata["dependencies"]:
-        new_package_version = metadata["version"]
-        dependency_idx = metadata["dependencies"].index("dbt-fusion-package-tools")
-        metadata["dependencies"][dependency_idx] = f"dbt-fusion-package-tools=={new_package_version}"
+        version = str(metadata["version"])
+        # Use stderr to ensure visibility in pip output
+        sys.stderr.write(f"[dbt-autofix build] Detected version: {version}\n")
+
+        # Only pin strictly for real releases.
+        # Fallback versions (0.0...) or development versions (.post, .dev, .rc, +, alpha, beta, etc.)
+        # should not be pinned as they likely won't exist on PyPI.
+        is_release = not (
+            ".post" in version
+            or ".dev" in version
+            or "rc" in version
+            or "+" in version
+            or "a" in version
+            or "b" in version
+            or "alpha" in version
+            or "beta" in version
+            or version.startswith("0.0")
+        )
+
+        if is_release:
+            sys.stderr.write(
+                f"[dbt-autofix build] Release version detected. Pinning dbt-fusion-package-tools to =={version}\n"
+            )
+            dependency_idx = metadata["dependencies"].index("dbt-fusion-package-tools")
+            metadata["dependencies"][dependency_idx] = f"dbt-fusion-package-tools=={version}"
+        else:
+            sys.stderr.write(
+                "[dbt-autofix build] Non-release/fallback version detected. Skipping strict dependency pin.\n"
+            )