toml-combine 0.1.7__py3-none-any.whl → 0.1.9__py3-none-any.whl

toml_combine/combiner.py

@@ -5,7 +5,7 @@ import dataclasses
 import itertools
 from collections.abc import Mapping, Sequence
 from functools import partial
-from typing import Any
+from typing import Any, TypeVar, Union
 
 from . import exceptions
 
@@ -24,7 +24,7 @@ class Output:
 
 @dataclasses.dataclass()
 class Override:
-    when: Mapping[str, str | list[str]]
+    when: Mapping[str, list[str]]
     config: Mapping[str, Any]
 
     def __str__(self) -> str:
@@ -48,9 +48,12 @@ def wrap_in_list(value: str | list[str]) -> list[str]:
     return value
 
 
+T = TypeVar("T", bound=Union[str, list[str]])
+
+
 def clean_dimensions_dict(
-    to_sort: Mapping[str, str | list[str]], clean: dict[str, list[str]], type: str
-) -> dict[str, str]:
+    to_sort: Mapping[str, T], clean: dict[str, list[str]], type: str
+) -> dict[str, T]:
     """
     Recreate a dictionary of dimension values with the same order as the
     dimensions list.
@@ -148,7 +151,9 @@ def build_config(config: dict[str, Any]) -> Config:
         except KeyError:
             raise exceptions.MissingOverrideCondition(id=override)
 
-        conditions = tuple((k, tuple(wrap_in_list(v))) for k, v in when.items())
+        when = {k: wrap_in_list(v) for k, v in when.items()}
+
+        conditions = tuple((k, tuple(v)) for k, v in when.items())
         if conditions in seen_conditions:
             raise exceptions.DuplicateError(type="override", id=when)
 
@@ -176,7 +181,6 @@ def build_config(config: dict[str, Any]) -> Config:
             output[key] = wrap_in_list(output[key])
 
         for cartesian_product in itertools.product(*output.values()):
-            # Create a dictionary with the same keys as when
             single_output = dict(zip(output.keys(), cartesian_product))
 
             conditions = tuple(single_output.items())
@@ -200,6 +204,20 @@ def build_config(config: dict[str, Any]) -> Config:
     )
 
 
+def output_matches_override(output: Output, override: Override) -> bool:
+    """
+    Check if the values in the override match the output dimensions.
+    """
+    for dim, values in override.when.items():
+        if dim not in output.dimensions:
+            return False
+
+        if output.dimensions[dim] not in values:
+            return False
+
+    return True
+
+
 def generate_output(
     default: Mapping[str, Any], overrides: Sequence[Override], output: Output
 ) -> dict[str, Any]:
@@ -207,10 +225,8 @@ def generate_output(
     # Apply each matching override
     for override in overrides:
         # Check if all dimension values in the override match
-        if all(
-            override.when.get(dim) == output.dimensions.get(dim)
-            for dim in override.when.keys()
-        ):
+
+        if output_matches_override(output=output, override=override):
             result = merge_configs(result, override.config)
 
     return {"dimensions": output.dimensions, **result}

{toml_combine-0.1.7.dist-info → toml_combine-0.1.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: toml-combine
-Version: 0.1.7
+Version: 0.1.9
 Summary: A tool for combining complex configurations in TOML format.
 Author-email: Joachim Jablon <ewjoachim@gmail.com>
 License-Expression: MIT
@@ -146,6 +146,41 @@ This example is simple because `name` is a natural choice for the key. In some c
 the choice is less natural, but you can always decide to name the elements of your
 list and use that name as a key. Also, yes, you'll loose ordering.
 
+### CLI
+
+```console
+$ toml-combine {path/to/config.toml}
+```
+
+Generates all the outputs described by the given TOML config.
+
+Note that you can restrict generation to some dimension values by passing
+`--{dimension}={value}`
+
+## Lib
+
+```python
+import toml_combine
+
+
+result = toml_combine.combine(
+        config_file=config_file,
+        environment=["production", "staging"],
+        type="job",
+        job=["manage", "special-command"],
+    )
+
+print(result)
+{
+  "production-job-manage": {...},
+  "production-job-special-command": {...},
+  "staging-job-manage": {...},
+  "staging-job-special-command": {...},
+}
+```
+
+You can pass either `config` (TOML string or dict) or `config_file` (`pathlib.Path` or string path) to `combine()`. Additional `kwargs` restrict the output.
+
 ### A bigger example
 
 ```toml
@@ -181,39 +216,90 @@ container.port = 8080
 name = "service-dev"
 when.environment = "dev"
 container.env.DEBUG = true
-```
-
-### CLI
 
-```console
-$ toml-combine {path/to/config.toml}
+[[override]]
+when.environment = ["staging", "dev"]
+when.service = "backend"
+container.env.ENABLE_EXPENSIVE_MONITORING = false
 ```
 
-Generates all the outputs described by the given TOML config.
-
-Note that you can restrict generation to some dimension values by passing
-`--{dimension}={value}`
+This produces the following configs:
 
-## Lib
-
-```python
-import toml_combine
-
-
-result = toml_combine.combine(
-        config_file=config_file,
-        environment=["production", "staging"],
-        type="job",
-        job=["manage", "special-command"],
-    )
-
-print(result)
+```json
 {
-  "production-job-manage": {...},
-  "production-job-special-command": {...},
-  "staging-job-manage": {...},
-  "staging-job-special-command": {...},
+  "production-frontend-eu": {
+    "dimensions": {
+      "environment": "production",
+      "service": "frontend",
+      "region": "eu"
+    },
+    "registry": "gcr.io/my-project/",
+    "service_account": "my-service-account",
+    "name": "service-frontend",
+    "container": {
+      "image_name": "my-image-frontend"
+    }
+  },
+  "production-backend-eu": {
+    "dimensions": {
+      "environment": "production",
+      "service": "backend",
+      "region": "eu"
+    },
+    "registry": "gcr.io/my-project/",
+    "service_account": "my-service-account",
+    "name": "service-backend",
+    "container": {
+      "image_name": "my-image-backend",
+      "port": 8080
+    }
+  },
+  "staging-frontend-eu": {
+    "dimensions": {
+      "environment": "staging",
+      "service": "frontend",
+      "region": "eu"
+    },
+    "registry": "gcr.io/my-project/",
+    "service_account": "my-service-account",
+    "name": "service-frontend",
+    "container": {
+      "image_name": "my-image-frontend"
+    }
+  },
+  "staging-backend-eu": {
+    "dimensions": {
+      "environment": "staging",
+      "service": "backend",
+      "region": "eu"
+    },
+    "registry": "gcr.io/my-project/",
+    "service_account": "my-service-account",
+    "name": "service-backend",
+    "container": {
+      "image_name": "my-image-backend",
+      "port": 8080,
+      "env": {
+        "ENABLE_EXPENSIVE_MONITORING": false
+      }
+    }
+  },
+  "dev-backend": {
+    "dimensions": {
+      "environment": "dev",
+      "service": "backend"
+    },
+    "registry": "gcr.io/my-project/",
+    "service_account": "my-service-account",
+    "name": "service-backend",
+    "container": {
+      "env": {
+        "DEBUG": true,
+        "ENABLE_EXPENSIVE_MONITORING": false
+      },
+      "image_name": "my-image-backend",
+      "port": 8080
+    }
+  }
 }
 ```
-
-You can pass either `config` (TOML string or dict) or `config_file` (`pathlib.Path` or string path) to `combine()`. Additional `kwargs` restrict the output.

{toml_combine-0.1.7.dist-info → toml_combine-0.1.9.dist-info}/RECORD

@@ -1,10 +1,10 @@
 toml_combine/__init__.py,sha256=l7i0GkM9k7cc__wj1yqK5XjpB3IJ0jqFU63tqKMuYlY,1625
 toml_combine/__main__.py,sha256=hmF8N8xX6UEApzbKTVZ-4E1HU5-rjgUkdXNLO-mF6vo,100
 toml_combine/cli.py,sha256=MZrAEP4wt6f9Qn0TEXIjeLoQMlvQulFpkMciwU8GRO4,2328
-toml_combine/combiner.py,sha256=98iWUnkbzDVGEkw-dXFZEt5G7io5SQzayceuTU2hRvo,7315
+toml_combine/combiner.py,sha256=_sSOCVBlv8ljAdMd6BEthS6rxiCnt-VV9rMEPSK_Cvk,7643
 toml_combine/exceptions.py,sha256=SepRFDxeWQEbD88jhF5g7laZSSULthho83BpW8u9RWs,897
 toml_combine/toml.py,sha256=_vCINvfJeS3gWid35Pmm3Yz4xyJ8LpKJRHL0axSU8nk,384
-toml_combine-0.1.7.dist-info/METADATA,sha256=vUsTB2Phm6t8GMpu28aPPsIxkSNOb0nBIFnQuKJSj_Y,6401
-toml_combine-0.1.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-toml_combine-0.1.7.dist-info/entry_points.txt,sha256=dXUQNom54uZt_7ylEG81iNYMamYpaFo9-ItcZJU6Uzc,58
-toml_combine-0.1.7.dist-info/RECORD,,
+toml_combine-0.1.9.dist-info/METADATA,sha256=ti8e_ng-_KdRPCAJybModogRhP4guqH8UilQTeXQHAk,8375
+toml_combine-0.1.9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+toml_combine-0.1.9.dist-info/entry_points.txt,sha256=dXUQNom54uZt_7ylEG81iNYMamYpaFo9-ItcZJU6Uzc,58
+toml_combine-0.1.9.dist-info/RECORD,,