Group poetry.lock and pyproject.toml in frozen dataclass

Author: ArBridgeman

exasol/toolbox/nox/_lint.py

@@ -11,6 +11,7 @@
 from nox import Session
 
 from exasol.toolbox.nox._shared import python_files
+from exasol.toolbox.util.dependencies.poetry_dependencies import PoetryFiles
 from noxconfig import PROJECT_CONFIG
 
 
@@ -140,7 +141,7 @@ def security_lint(session: Session) -> None:
 @nox.session(name="lint:dependencies", python=False)
 def dependency_check(session: Session) -> None:
     """Checks if only valid sources of dependencies are used"""
-    content = Path(PROJECT_CONFIG.root, "pyproject.toml").read_text()
+    content = Path(PROJECT_CONFIG.root, PoetryFiles.pyproject_toml).read_text()
     dependencies = Dependencies.parse(content)
     console = rich.console.Console()
     if illegal := dependencies.illegal:

exasol/toolbox/nox/_release.py

@@ -14,6 +14,7 @@
     check_for_config_attribute,
 )
 from exasol.toolbox.nox.plugin import NoxTasks
+from exasol.toolbox.util.dependencies.poetry_dependencies import PoetryFiles
 from exasol.toolbox.util.git import Git
 from exasol.toolbox.util.release.changelog import Changelogs
 from exasol.toolbox.util.version import (
@@ -142,7 +143,7 @@ def prepare_release(session: Session) -> None:
         return
 
     changed_files += [
-        PROJECT_CONFIG.root / "pyproject.toml",
+        PROJECT_CONFIG.root / PoetryFiles.pyproject_toml,
         PROJECT_CONFIG.version_file,
     ]
     results = pm.hook.prepare_release_add_files(session=session, config=PROJECT_CONFIG)

exasol/toolbox/util/dependencies/poetry_dependencies.py

@@ -3,8 +3,9 @@
 import subprocess
 import tempfile
 from collections import OrderedDict
+from dataclasses import dataclass
 from pathlib import Path
-from typing import Optional
+from typing import Final
 
 import tomlkit
 from pydantic import (
@@ -28,7 +29,16 @@ class PoetryGroup(BaseModel):
     toml_section: str | None
 
 
-PYPROJECT_TOML = "pyproject.toml"
+@dataclass(frozen=True)
+class PoetryFiles:
+    pyproject_toml: Final[str] = "pyproject.toml"
+    poetry_lock: Final[str] = "poetry.lock"
+
+    @property
+    def files(self) -> tuple[str, ...]:
+        return tuple(self.__dict__.values())
+
+
 TRANSITIVE_GROUP = PoetryGroup(name="transitive", toml_section=None)
 
 
@@ -39,7 +49,7 @@ class PoetryToml(BaseModel):
 
     @classmethod
     def load_from_toml(cls, working_directory: Path) -> PoetryToml:
-        file_path = working_directory / PYPROJECT_TOML
+        file_path = working_directory / PoetryFiles.pyproject_toml
         if not file_path.exists():
             raise ValueError(f"File not found: {file_path}")
 
@@ -169,6 +179,6 @@ def get_dependencies_from_latest_tag() -> (
     path = PROJECT_CONFIG.root.relative_to(Git.toplevel())
     with tempfile.TemporaryDirectory() as tmpdir_str:
         tmpdir = Path(tmpdir_str)
-        for file in ("poetry.lock", PYPROJECT_TOML):
+        for file in PoetryFiles().files:
             Git.checkout(latest_tag, path / file, tmpdir / file)
         return get_dependencies(working_directory=tmpdir)