csrd-utils 0.3.51__tar.gz → 0.3.54__tar.gz

{csrd_utils-0.3.51 → csrd_utils-0.3.54}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: csrd-utils
-Version: 0.3.51
+Version: 0.3.54
 Summary: CLI utilities for csrd service generation and feature augmentation
 Project-URL: Repository, https://github.com/csrd-api/fastapi-common
 Project-URL: Documentation, https://github.com/csrd-api/fastapi-common/tree/main/packages/utils

{csrd_utils-0.3.51 → csrd_utils-0.3.54}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "csrd-utils"
-version = "0.3.51"
+version = "0.3.54"
 description = "CLI utilities for csrd service generation and feature augmentation"
 license = { text = "MIT" }
 requires-python = ">=3.12"

{csrd_utils-0.3.51 → csrd_utils-0.3.54}/src/csrd_utils/__main__.py

@@ -881,6 +881,37 @@ def _ensure_unique_ports(services: list[ServiceSpec], preserve_name: str) -> Non
         used.add(port)
 
 
+def _sync_workspace_service_feature(
+    workspace_root: Path, service_name: str, feature_name: str
+) -> Path | None:
+    spec = _load_workspace_cluster_spec(workspace_root)
+    services = spec.get("services", [])
+    target = next(
+        (
+            cast(ServiceSpec, dict(service))
+            for service in services
+            if str(service.get("name", "")).strip() == service_name
+        ),
+        None,
+    )
+    if target is None:
+        return None
+
+    features = list(target.get("features", []))
+    if feature_name not in features:
+        features.append(feature_name)
+    target["features"] = features
+
+    requires = list(target.get("requires", []))
+    if feature_name in {"caching", "workers"} and "redis" not in requires:
+        requires.append("redis")
+    target["requires"] = requires
+
+    updated_spec = _upsert_workspace_service(spec, target, workspace_root.name)
+    _save_workspace_cluster_spec(workspace_root, updated_spec)
+    return generate_cluster(updated_spec, workspace_root, force=True)
+
+
 def _maybe_update_workspace_cluster(
     workspace_root: Path,
     service_config: GenerateServiceArgs,
@@ -1315,6 +1346,13 @@ def main() -> int:
     if args.plan:
         print("Plan validated.")
     else:
+        feature_workspace_root = _workspace_root_for_service(resolved_service)
+        if feature_workspace_root is not None:
+            refreshed_cluster = _sync_workspace_service_feature(
+                feature_workspace_root, resolved_service.name, args.feature
+            )
+            if refreshed_cluster is not None:
+                print(f"Refreshed cluster scaffold at: {refreshed_cluster}")
         print(f"Feature '{args.feature}' added.")
         if changes:
             for change in changes:

{csrd_utils-0.3.51 → csrd_utils-0.3.54}/src/csrd_utils/augmentor.py

@@ -80,6 +80,15 @@ class ServiceAugmentor:
             return workspace_root / "tests"
         return self.service_root / "tests"
 
+    @property
+    def _pkg_import_name(self) -> str:
+        return self._pkg_dir.name
+
+    def _rewrite_src_app_refs(self, content: str) -> str:
+        """Map legacy feature fragment imports to the concrete service package name."""
+        pkg = self._pkg_import_name
+        return content.replace("src.app.", f"{pkg}.").replace("src.app", pkg)
+
     def load_feature(self, feature_name: str) -> bool:
         """
         Load and validate feature manifest.
@@ -247,6 +256,11 @@ class ServiceAugmentor:
         if not fragment_path.exists():
             return True  # Not required
 
+        if self._workspace_root is not None:
+            # Workspace compose is generated from cluster spec, not merged via legacy fragments.
+            print("✓ Skipped docker-compose.yml fragment merge (workspace-managed compose)")
+            return True
+
         try:
             dc_path = self._docker_compose_path()
             self.backup_file(dc_path)
@@ -343,7 +357,7 @@ class ServiceAugmentor:
             self.backup_file(conftest_path)
 
             base = conftest_path.read_text()
-            fragment = fragment_path.read_text()
+            fragment = self._rewrite_src_app_refs(fragment_path.read_text())
             merged = ConfTestMerger.merge(base, fragment)
 
             conftest_path.write_text(merged)
@@ -387,8 +401,11 @@ class ServiceAugmentor:
                     # Create parent dirs
                     dest_file.parent.mkdir(parents=True, exist_ok=True)
 
-                    # Copy file
-                    shutil.copy2(src_file, dest_file)
+                    if src_file.suffix == ".py":
+                        rendered = self._rewrite_src_app_refs(src_file.read_text())
+                        dest_file.write_text(rendered)
+                    else:
+                        shutil.copy2(src_file, dest_file)
                     self.applied_changes.append(f"Created {rel_path}")
 
             print("✓ Created feature files")
@@ -433,7 +450,11 @@ class ServiceAugmentor:
                         dest_file = self._tests_dir / rel_path
 
                     dest_file.parent.mkdir(parents=True, exist_ok=True)
-                    shutil.copy2(src_file, dest_file)
+                    if src_file.suffix == ".py":
+                        rendered = self._rewrite_src_app_refs(src_file.read_text())
+                        dest_file.write_text(rendered)
+                    else:
+                        shutil.copy2(src_file, dest_file)
                     self.applied_changes.append(f"Created {rel_path}")
 
             print("✓ Created test files")

{csrd_utils-0.3.51 → csrd_utils-0.3.54}/src/csrd_utils/cluster.py

@@ -232,8 +232,14 @@ def _infer_infra(services: Iterable[ServiceSpec]) -> set[str]:
         if service_type == "worker":
             broker = service.get("broker", "redis")
             infra.add("redis" if broker == "redis" else "rabbitmq")
+        if "workers" in service.get("features", []):
+            broker = service.get("broker", "redis")
+            infra.add("redis" if broker == "redis" else "rabbitmq")
         if "caching" in service.get("features", []):
             infra.add("redis")
+        for required in service.get("requires", []):
+            if required in INFRA_NAMES:
+                infra.add(required)
     return infra
 
 
@@ -258,7 +264,7 @@ def _service_environment(service: ServiceSpec, services: list[ServiceSpec]) -> d
             env["DB_PASSWORD"] = "${DB_PASSWORD:-change_me}"
             env["DB_NAME"] = "${DB_NAME:-service_db}"
 
-    if service["type"] == "worker":
+    if service["type"] == "worker" or "workers" in service.get("features", []):
         broker = service.get("broker", "redis")
         if broker == "redis":
             env["CELERY_BROKER_URL"] = "redis://redis:6379/0"
@@ -306,6 +312,9 @@ def render_compose(spec: ClusterSpec) -> dict[str, Any]:
         if service["type"] == "worker":
             requires.add("redis" if service.get("broker", "redis") == "redis" else "rabbitmq")
 
+        if "workers" in service.get("features", []):
+            requires.add("redis" if service.get("broker", "redis") == "redis" else "rabbitmq")
+
         if "caching" in service.get("features", []):
             requires.add("redis")
 
@@ -432,14 +441,14 @@ def _build_readme(spec: ClusterSpec) -> str:
             "From the workspace root, run all tests:",
             "",
             "```bash",
-            "pytest",
+            "uv run pytest",
             "```",
             "",
             "Run tests for a specific service:",
             "",
             "```bash",
-            "pytest tests/unit/{service_name}",
-            "pytest tests/acceptance/{service_name}",
+            "uv run pytest tests/unit/{service_name}",
+            "uv run pytest tests/acceptance/{service_name}",
             "```",
             "",
             "## Run",