feat: unwrap resulting item

Makefile

@@ -67,8 +67,11 @@ generate-test-outputs:
 	@rm -rf tests/outputs/with_hooks_dir
 	@uv run efemel process "**/*.py" --cwd tests/inputs/basic --out tests/outputs/with_hooks_dir --hooks tests/hooks/multiple
 
-	@rm -rf tests/outputs/with_hooks_dir
-	@uv run efemel process "**/*.py" --cwd tests/inputs/process_data --out tests/outputs/process_data --pick user_data
+	@rm -rf tests/outputs/process_data_pick
+	@uv run efemel process "**/*.py" --cwd tests/inputs/process_data --out tests/outputs/process_data_pick --pick user_data
+
+	@rm -rf tests/outputs/process_data_unwrap
+	@uv run efemel process "**/*.py" --cwd tests/inputs/process_data --out tests/outputs/process_data_unwrap --unwrap user_data
 
 
 # Clean build artifacts and cache files

efemel/cli.py

@@ -56,7 +56,13 @@ def info():
   multiple=True,
   help="Pick specific dictionary keys to extract (can be used multiple times)",
 )
-def process(file_pattern, out, flatten, cwd, env, workers, hooks, pick):
+@click.option(
+  "--unwrap",
+  "-u",
+  multiple=True,
+  help="Extract specific values from the processed data dictionary, merging them (can be used multiple times)",
+)
+def process(file_pattern, out, flatten, cwd, env, workers, hooks, pick, unwrap):
   """Process Python files and extract public dictionary variables to JSON.
 
   FILE_PATTERN: Glob pattern to match Python files (e.g., "**/*.py")
@@ -75,6 +81,9 @@ def process(file_pattern, out, flatten, cwd, env, workers, hooks, pick):
   if pick:
     hooks_manager.add("process_data", process_data_hooks.pick_data(pick))
 
+  if unwrap:
+    hooks_manager.add("process_data", process_data_hooks.unwrap_data(unwrap))
+
   # Load user-defined hooks if a path is specified
   if hooks:
     if os.path.isfile(hooks):

efemel/hooks/process_data.py

@@ -7,3 +7,26 @@ def _pick(context):
     context["data"] = {key: value for key, value in context["data"].items() if key in keys}
 
   return _pick
+
+
+def unwrap_data(keys):
+  """Extracts specific values from the processed data dictionary, merges them."""
+
+  def _unwrap(context):
+    if not keys:
+      return
+
+    merged = {}
+
+    for key in keys:
+      if key not in context["data"]:
+        continue
+
+      value = context["data"][key]
+
+      if isinstance(value, dict):
+        merged.update(value)
+
+    context["data"] = merged
+
+  return _unwrap

tests/outputs/process_data_unwrap/data.json

@@ -0,0 +1,10 @@
+{
+  "admin": {
+    "name": "Admin User",
+    "role": "admin"
+  },
+  "guest": {
+    "name": "Guest User",
+    "role": "guest"
+  }
+}

tests/outputs/with_hooks_dir/simple_test_1_2_3.json

@@ -0,0 +1,10 @@
+{
+  "test_dict": {
+    "key": "value",
+    "number": 42
+  },
+  "config": {
+    "app": "test",
+    "version": "1.0"
+  }
+}

tests/outputs/with_hooks_dir/test_data_1_2_3.json

@@ -0,0 +1,33 @@
+{
+  "config": {
+    "name": "test_app",
+    "version": "1.0.0",
+    "debug": true,
+    "features": [
+      "auth",
+      "api",
+      "ui"
+    ]
+  },
+  "settings": {
+    "database": {
+      "host": "localhost",
+      "port": 5432,
+      "name": "app_db"
+    },
+    "cache": {
+      "type": "redis",
+      "ttl": 3600
+    }
+  },
+  "user_data": {
+    "admin": {
+      "name": "Admin User",
+      "role": "admin"
+    },
+    "guest": {
+      "name": "Guest User",
+      "role": "guest"
+    }
+  }
+}

tests/outputs/with_hooks_dir/test_dir/config_1_2_3.json

@@ -0,0 +1,10 @@
+{
+  "app_config": {
+    "app_name": "my_app",
+    "port": 8080
+  },
+  "database_config": {
+    "url": "postgresql://localhost/mydb",
+    "pool_size": 10
+  }
+}

tests/outputs/with_hooks_dir/test_dir/subdir/nested_1_2_3.json

@@ -0,0 +1,14 @@
+{
+  "nested_data": {
+    "level": "deep",
+    "items": [
+      1,
+      2,
+      3
+    ]
+  },
+  "metadata": {
+    "created_by": "test",
+    "timestamp": "2025-01-01"
+  }
+}

tests/outputs/with_hooks_dir/test_dir/utils_1_2_3.json

@@ -0,0 +1 @@
+{}

tests/test_cli.py

@@ -51,9 +51,15 @@ def get_test_scenarios():
     {
       "name": "process data - pick",
       "inputs_dir": test_dir / "inputs/process_data",
-      "outputs_dir": test_dir / "outputs/process_data",
+      "outputs_dir": test_dir / "outputs/process_data_pick",
       "process_args": ["--pick", "user_data"],
     },
+    {
+      "name": "process data - unwrap",
+      "inputs_dir": test_dir / "inputs/process_data",
+      "outputs_dir": test_dir / "outputs/process_data_unwrap",
+      "process_args": ["--unwrap", "user_data"],
+    },
   ]