feat: Add DynamoDB document schemas

.gitignore

@@ -4,4 +4,5 @@
 *.pyc
 .coverage
 common.sqlite3
-dist/
+dist/
+coverage.xml

pyproject.toml

@@ -21,12 +21,13 @@ optional-dependencies = { test-tools = [
     "psycopg2-binary (>=2.9,<3)",
     "requests",
     "simplejson (>=3,<4)",
-
 ], task-processor = [
     "backoff (>=2.2.1,<3.0.0)",
     "django (>4,<5)",
     "django-health-check",
     "prometheus-client (>=0.0.16)",
+], flagsmith-models = [
+    "typing_extensions",
 ] }
 authors = [
     { name = "Matthew Elwell" },
@@ -69,6 +70,7 @@ dev = [
     "djangorestframework-stubs (>=3.15.3, <4.0.0)",
     "mypy (>=1.15.0, <2.0.0)",
     "pre-commit",
+    "pydantic>=2.12.5",
     "pyfakefs (>=5.7.4, <6.0.0)",
     "pytest (>=8.3.4, <9.0.0)",
     "pytest-asyncio (>=0.25.3, <1.0.0)",
@@ -78,16 +80,16 @@ dev = [
     "pytest-httpserver (>=1.1.3, <2.0.0)",
     "pytest-mock (>=3.14.0, <4.0.0)",
     "setuptools (>=78.1.1, <79.0.0)",
-    "types-simplejson (>=3.20.0.20250326, <4.0.0)",
     "types-python-dateutil (>=2.9.0.20250516, <3.0.0)",
+    "types-simplejson (>=3.20.0.20250326, <4.0.0)",
 ]
 
 [build-system]
 requires = ["uv_build>=0.9.17,<0.10.0"]
 build-backend = "uv_build"
 
 [tool.uv.build-backend]
-module-name = ["common", "task_processor"]
+module-name = ["common", "task_processor", "flagsmith_schemas"]
 
 [tool.coverage.report]
 # Regexes for lines to exclude from consideration

src/common/test_tools/plugin.py

@@ -1,11 +1,7 @@
 from functools import partial
-from typing import Generator
+from typing import TYPE_CHECKING, Generator
 
-import prometheus_client
 import pytest
-from prometheus_client.metrics import MetricWrapperBase
-from pyfakefs.fake_filesystem import FakeFilesystem
-from pytest_django.fixtures import SettingsWrapper
 
 from common.test_tools.types import (
     AssertMetricFixture,
@@ -15,6 +11,10 @@
 )
 from task_processor.task_run_method import TaskRunMethod
 
+if TYPE_CHECKING:
+    from pyfakefs.fake_filesystem import FakeFilesystem
+    from pytest_django.fixtures import SettingsWrapper
+
 
 def pytest_addoption(parser: pytest.Parser) -> None:
     group = parser.getgroup("snapshot")
@@ -26,12 +26,14 @@ def pytest_addoption(parser: pytest.Parser) -> None:
 
 
 def assert_metric_impl() -> Generator[AssertMetricFixture, None, None]:
+    import prometheus_client
+
     registry = prometheus_client.REGISTRY
     collectors = [*registry._collector_to_names]
 
     # Reset registry state
     for collector in collectors:
-        if isinstance(collector, MetricWrapperBase):
+        if isinstance(collector, prometheus_client.metrics.MetricWrapperBase):
             collector.clear()
 
     def _assert_metric(
@@ -53,7 +55,7 @@ def _assert_metric(
 
 
 @pytest.fixture()
-def saas_mode(fs: FakeFilesystem) -> Generator[None, None, None]:
+def saas_mode(fs: "FakeFilesystem") -> Generator[None, None, None]:
     from common.core.utils import is_saas
 
     is_saas.cache_clear()
@@ -65,7 +67,7 @@ def saas_mode(fs: FakeFilesystem) -> Generator[None, None, None]:
 
 
 @pytest.fixture()
-def enterprise_mode(fs: FakeFilesystem) -> Generator[None, None, None]:
+def enterprise_mode(fs: "FakeFilesystem") -> Generator[None, None, None]:
     from common.core.utils import is_enterprise
 
     is_enterprise.cache_clear()
@@ -77,7 +79,7 @@ def enterprise_mode(fs: FakeFilesystem) -> Generator[None, None, None]:
 
 
 @pytest.fixture()
-def task_processor_mode(settings: SettingsWrapper) -> None:
+def task_processor_mode(settings: "SettingsWrapper") -> None:
     settings.TASK_PROCESSOR_MODE = True
     # The setting is supposed to be set before the metrics module is imported,
     # so reload it
@@ -101,7 +103,7 @@ def flagsmith_markers_marked(
 
 @pytest.fixture(name="run_tasks")
 def run_tasks_impl(
-    settings: SettingsWrapper,
+    settings: "SettingsWrapper",
     transactional_db: None,
     task_processor_mode: None,
 ) -> RunTasksFixture: