[feature/PI-49-logging_with_step_chain]

.pre-commit-config.yaml

@@ -78,6 +78,7 @@ repos:
         entry: json\.loads
         language: pygrep
         types: [python]
+        exclude: src/layers/event/json/__init__.py
 
   - repo: local
     hooks:
@@ -86,6 +87,7 @@ repos:
         entry: json\.load
         language: pygrep
         types: [python]
+        exclude: src/layers/event/json/__init__.py
 
   - repo: local
     hooks:

pyproject.toml

@@ -16,6 +16,7 @@ packages = [
 [tool.poetry.dependencies]
 python = "3.11.4"
 pydantic = "^2.4.2"
+nhs-context-logging = "^0.2.7"
 
 [tool.poetry.group.dev.dependencies]
 pre-commit = "^3.4.0"
@@ -25,6 +26,8 @@ behave = "^1.2.6"
 pytest = "^7.4.2"
 pytest-custom-exit-code = "^0.3.0"
 sh = "^2.0.6"
+hypothesis = "^6.87.3"
+aws-lambda-powertools = { extras = ["aws-sdk"], version = "^2.25.1" }
 
 [build-system]
 requires = ["poetry-core"]

src/api/createProduct/index.py

@@ -0,0 +1,37 @@
+from event.environment import BaseEnvironment
+from event.logging.logger import setup_logger
+from event.logging.step_decorators import logging_step_decorators
+from event.response.steps import response_steps
+from event.step_chain import StepChain
+from event.versioning.steps import get_steps_for_requested_version, versioning_steps
+
+
+class Environment(BaseEnvironment):
+    SOMETHING: str
+
+
+cache = {**Environment.model_construct().model_dump()}
+step_decorators = [*logging_step_decorators]
+pre_steps = [*versioning_steps]
+post_steps = [*response_steps]
+
+
+def handler(event: dict, context=None):
+    setup_logger(service_name=__file__)
+
+    pre_step_chain = StepChain(
+        step_chain=versioning_steps, step_decorators=step_decorators
+    )
+    pre_step_chain.run(init={"event": event, "__file__": __file__})
+
+    if isinstance(pre_step_chain.result, Exception):
+        result = pre_step_chain.result
+    else:
+        steps = pre_step_chain.data[get_steps_for_requested_version]
+        step_chain = StepChain(step_chain=steps, step_decorators=step_decorators)
+        step_chain.run(cache=cache, init=event)
+        result = step_chain.result
+
+    post_step_chain = StepChain(step_chain=post_steps, step_decorators=step_decorators)
+    post_step_chain.run(init=result)
+    return post_step_chain.result

src/api/createProduct/src/v1/steps.py

@@ -0,0 +1,9 @@
+def a(data, cache):
+    return {"blah", "hi"}
+
+
+def b(data, cache):
+    return {"a's result": data[a]}
+
+
+steps = [a, b]

src/api/createProduct/tests/test_index.py

@@ -0,0 +1,14 @@
+import pytest
+
+from api.createProduct.index import handler
+
+
+@pytest.mark.parametrize(
+    "version",
+    [
+        "1",
+    ],
+)
+def test_index(version):
+    result = handler(event={"headers": {"version": version}})
+    assert result == "OK"