@aws/nx-plugin 1.0.0-rc.13 → 1.0.0-rc.14

package/src/py/dynamodb/agent-connection/__snapshots__/generator.spec.ts.snap

@@ -0,0 +1,45 @@
+// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
+
+exports[`py#dynamodb agent-connection generator > should add dynamodb serve-local dependency to agent serve-local 1`] = `
+{
+  "$schema": "../../node_modules/nx/schemas/project-schema.json",
+  "name": "my-agent",
+  "root": "packages/my-agent",
+  "targets": {
+    "agent-serve-local": {
+      "continuous": true,
+      "dependsOn": [
+        {
+          "projects": [
+            "db",
+          ],
+          "target": "serve-local",
+        },
+      ],
+      "executor": "nx:run-commands",
+    },
+  },
+}
+`;
+
+exports[`py#dynamodb agent-connection generator > should use sourceComponent name when provided 1`] = `
+{
+  "$schema": "../../node_modules/nx/schemas/project-schema.json",
+  "name": "my-agent",
+  "root": "packages/my-agent",
+  "targets": {
+    "custom-agent-serve-local": {
+      "continuous": true,
+      "dependsOn": [
+        {
+          "projects": [
+            "db",
+          ],
+          "target": "serve-local",
+        },
+      ],
+      "executor": "nx:run-commands",
+    },
+  },
+}
+`;

package/src/py/dynamodb/agent-connection/generator.d.ts

@@ -0,0 +1,10 @@
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { type GeneratorCallback, type Tree } from '@nx/devkit';
+import { type NxGeneratorInfo } from '../../../utils/nx';
+import type { PyDynamoDBAgentConnectionGeneratorSchema } from './schema';
+export declare const PY_DYNAMODB_AGENT_CONNECTION_GENERATOR_INFO: NxGeneratorInfo;
+export declare const pyDynamoDBAgentConnectionGenerator: (tree: Tree, options: PyDynamoDBAgentConnectionGeneratorSchema) => Promise<GeneratorCallback>;
+export default pyDynamoDBAgentConnectionGenerator;

package/src/py/dynamodb/agent-connection/generator.js

@@ -0,0 +1,39 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.pyDynamoDBAgentConnectionGenerator = exports.PY_DYNAMODB_AGENT_CONNECTION_GENERATOR_INFO = void 0;
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+const devkit_1 = require("@nx/devkit");
+const format_1 = require("../../../utils/format");
+const metrics_1 = require("../../../utils/metrics");
+const nx_1 = require("../../../utils/nx");
+const nxlv_python_1 = require("../../../utils/nxlv-python");
+const py_1 = require("../../../utils/py");
+exports.PY_DYNAMODB_AGENT_CONNECTION_GENERATOR_INFO = (0, nx_1.getGeneratorInfo)(__filename);
+const pyDynamoDBAgentConnectionGenerator = async (tree, options) => {
+    const sourceProject = (0, nx_1.readProjectConfigurationUnqualified)(tree, options.sourceProject);
+    const targetProject = (0, nx_1.readProjectConfigurationUnqualified)(tree, options.targetProject);
+    (0, py_1.addWorkspaceDependencyToPyProject)(tree, sourceProject.root, targetProject.name);
+    const agentName = options.sourceComponent?.name ?? 'agent';
+    const serveLocalTarget = `${agentName}-serve-local`;
+    if (sourceProject.targets?.[serveLocalTarget]) {
+        (0, nx_1.addDependencyToTargetIfNotPresent)(sourceProject, serveLocalTarget, {
+            projects: [targetProject.name],
+            target: 'serve-local',
+        });
+        (0, devkit_1.updateProjectConfiguration)(tree, sourceProject.name, sourceProject);
+    }
+    await (0, metrics_1.addGeneratorMetricsIfApplicable)(tree, [
+        exports.PY_DYNAMODB_AGENT_CONNECTION_GENERATOR_INFO,
+    ]);
+    await (0, format_1.formatFilesInSubtree)(tree);
+    return async () => {
+        (0, devkit_1.installPackagesTask)(tree);
+        await new nxlv_python_1.UVProvider(tree.root, new nxlv_python_1.Logger(), tree).install();
+    };
+};
+exports.pyDynamoDBAgentConnectionGenerator = pyDynamoDBAgentConnectionGenerator;
+exports.default = exports.pyDynamoDBAgentConnectionGenerator;
+//# sourceMappingURL=generator.js.map

package/src/py/dynamodb/agent-connection/generator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"generator.js","sourceRoot":"","sources":["../../../../../../../packages/nx-plugin/src/py/dynamodb/agent-connection/generator.ts"],"names":[],"mappings":";;;AAAA;;;GAGG;AACH,uCAKoB;AACpB,kDAA6D;AAC7D,oDAAyE;AACzE,0CAK2B;AAC3B,4DAAgE;AAChE,0CAAsE;AAGzD,QAAA,2CAA2C,GACtD,IAAA,qBAAgB,EAAC,UAAU,CAAC,CAAC;AAExB,MAAM,kCAAkC,GAAG,KAAK,EACrD,IAAU,EACV,OAAiD,EACrB,EAAE;IAC9B,MAAM,aAAa,GAAG,IAAA,wCAAmC,EACvD,IAAI,EACJ,OAAO,CAAC,aAAa,CACtB,CAAC;IACF,MAAM,aAAa,GAAG,IAAA,wCAAmC,EACvD,IAAI,EACJ,OAAO,CAAC,aAAa,CACtB,CAAC;IAEF,IAAA,sCAAiC,EAC/B,IAAI,EACJ,aAAa,CAAC,IAAI,EAClB,aAAa,CAAC,IAAK,CACpB,CAAC;IAEF,MAAM,SAAS,GAAG,OAAO,CAAC,eAAe,EAAE,IAAI,IAAI,OAAO,CAAC;IAC3D,MAAM,gBAAgB,GAAG,GAAG,SAAS,cAAc,CAAC;IAEpD,IAAI,aAAa,CAAC,OAAO,EAAE,CAAC,gBAAgB,CAAC,EAAE,CAAC;QAC9C,IAAA,sCAAiC,EAAC,aAAa,EAAE,gBAAgB,EAAE;YACjE,QAAQ,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC;YAC9B,MAAM,EAAE,aAAa;SACtB,CAAC,CAAC;QACH,IAAA,mCAA0B,EAAC,IAAI,EAAE,aAAa,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;IACtE,CAAC;IAED,MAAM,IAAA,yCAA+B,EAAC,IAAI,EAAE;QAC1C,mDAA2C;KAC5C,CAAC,CAAC;IACH,MAAM,IAAA,6BAAoB,EAAC,IAAI,CAAC,CAAC;IACjC,OAAO,KAAK,IAAI,EAAE;QAChB,IAAA,4BAAmB,EAAC,IAAI,CAAC,CAAC;QAC1B,MAAM,IAAI,wBAAU,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,oBAAM,EAAE,EAAE,IAAI,CAAC,CAAC,OAAO,EAAE,CAAC;IAChE,CAAC,CAAC;AACJ,CAAC,CAAC;AAtCW,QAAA,kCAAkC,sCAsC7C;AAEF,kBAAe,0CAAkC,CAAC"}

package/src/py/dynamodb/agent-connection/schema.d.ts

@@ -0,0 +1,14 @@
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { ComponentMetadata } from '../../../utils/nx';
+
+/**
+ * TypeScript types for options defined in schema.json
+ */
+export interface PyDynamoDBAgentConnectionGeneratorSchema {
+  sourceProject: string;
+  targetProject: string;
+  sourceComponent?: ComponentMetadata;
+}

package/src/py/dynamodb/agent-connection/schema.json

@@ -0,0 +1,22 @@
+{
+  "$schema": "https://json-schema.org/schema",
+  "$id": "py#dynamodb#agent-connection",
+  "title": "py#dynamodb#agent-connection",
+  "description": "Connect a py#agent to a py#dynamodb project",
+  "type": "object",
+  "properties": {
+    "sourceProject": {
+      "type": "string",
+      "description": "The project containing the py#agent to connect from"
+    },
+    "targetProject": {
+      "type": "string",
+      "description": "The py#dynamodb project to connect to"
+    },
+    "sourceComponent": {
+      "type": "string",
+      "description": "The agent component name"
+    }
+  },
+  "required": ["sourceProject", "targetProject"]
+}

package/src/py/dynamodb/fast-api-connection/__snapshots__/generator.spec.ts.snap

@@ -0,0 +1,23 @@
+// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
+
+exports[`py#dynamodb fast-api-connection generator > should add dynamodb serve-local dependency to fast-api serve-local 1`] = `
+{
+  "$schema": "../../node_modules/nx/schemas/project-schema.json",
+  "name": "api",
+  "root": "packages/api",
+  "targets": {
+    "serve-local": {
+      "continuous": true,
+      "dependsOn": [
+        {
+          "projects": [
+            "db",
+          ],
+          "target": "serve-local",
+        },
+      ],
+      "executor": "nx:run-commands",
+    },
+  },
+}
+`;

package/src/py/dynamodb/fast-api-connection/generator.d.ts

@@ -0,0 +1,10 @@
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { type GeneratorCallback, type Tree } from '@nx/devkit';
+import { type NxGeneratorInfo } from '../../../utils/nx';
+import type { PyDynamoDBFastApiConnectionGeneratorSchema } from './schema';
+export declare const PY_DYNAMODB_FAST_API_CONNECTION_GENERATOR_INFO: NxGeneratorInfo;
+export declare const pyDynamoDBFastApiConnectionGenerator: (tree: Tree, options: PyDynamoDBFastApiConnectionGeneratorSchema) => Promise<GeneratorCallback>;
+export default pyDynamoDBFastApiConnectionGenerator;

package/src/py/dynamodb/fast-api-connection/generator.js

@@ -0,0 +1,37 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.pyDynamoDBFastApiConnectionGenerator = exports.PY_DYNAMODB_FAST_API_CONNECTION_GENERATOR_INFO = void 0;
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+const devkit_1 = require("@nx/devkit");
+const format_1 = require("../../../utils/format");
+const metrics_1 = require("../../../utils/metrics");
+const nx_1 = require("../../../utils/nx");
+const nxlv_python_1 = require("../../../utils/nxlv-python");
+const py_1 = require("../../../utils/py");
+exports.PY_DYNAMODB_FAST_API_CONNECTION_GENERATOR_INFO = (0, nx_1.getGeneratorInfo)(__filename);
+const pyDynamoDBFastApiConnectionGenerator = async (tree, options) => {
+    const sourceProject = (0, nx_1.readProjectConfigurationUnqualified)(tree, options.sourceProject);
+    const targetProject = (0, nx_1.readProjectConfigurationUnqualified)(tree, options.targetProject);
+    (0, py_1.addWorkspaceDependencyToPyProject)(tree, sourceProject.root, targetProject.name);
+    if (sourceProject.targets?.['serve-local']) {
+        (0, nx_1.addDependencyToTargetIfNotPresent)(sourceProject, 'serve-local', {
+            projects: [targetProject.name],
+            target: 'serve-local',
+        });
+        (0, devkit_1.updateProjectConfiguration)(tree, sourceProject.name, sourceProject);
+    }
+    await (0, metrics_1.addGeneratorMetricsIfApplicable)(tree, [
+        exports.PY_DYNAMODB_FAST_API_CONNECTION_GENERATOR_INFO,
+    ]);
+    await (0, format_1.formatFilesInSubtree)(tree);
+    return async () => {
+        (0, devkit_1.installPackagesTask)(tree);
+        await new nxlv_python_1.UVProvider(tree.root, new nxlv_python_1.Logger(), tree).install();
+    };
+};
+exports.pyDynamoDBFastApiConnectionGenerator = pyDynamoDBFastApiConnectionGenerator;
+exports.default = exports.pyDynamoDBFastApiConnectionGenerator;
+//# sourceMappingURL=generator.js.map

package/src/py/dynamodb/fast-api-connection/generator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"generator.js","sourceRoot":"","sources":["../../../../../../../packages/nx-plugin/src/py/dynamodb/fast-api-connection/generator.ts"],"names":[],"mappings":";;;AAAA;;;GAGG;AACH,uCAKoB;AACpB,kDAA6D;AAC7D,oDAAyE;AACzE,0CAK2B;AAC3B,4DAAgE;AAChE,0CAAsE;AAGzD,QAAA,8CAA8C,GACzD,IAAA,qBAAgB,EAAC,UAAU,CAAC,CAAC;AAExB,MAAM,oCAAoC,GAAG,KAAK,EACvD,IAAU,EACV,OAAmD,EACvB,EAAE;IAC9B,MAAM,aAAa,GAAG,IAAA,wCAAmC,EACvD,IAAI,EACJ,OAAO,CAAC,aAAa,CACtB,CAAC;IACF,MAAM,aAAa,GAAG,IAAA,wCAAmC,EACvD,IAAI,EACJ,OAAO,CAAC,aAAa,CACtB,CAAC;IAEF,IAAA,sCAAiC,EAC/B,IAAI,EACJ,aAAa,CAAC,IAAI,EAClB,aAAa,CAAC,IAAK,CACpB,CAAC;IAEF,IAAI,aAAa,CAAC,OAAO,EAAE,CAAC,aAAa,CAAC,EAAE,CAAC;QAC3C,IAAA,sCAAiC,EAAC,aAAa,EAAE,aAAa,EAAE;YAC9D,QAAQ,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC;YAC9B,MAAM,EAAE,aAAa;SACtB,CAAC,CAAC;QACH,IAAA,mCAA0B,EAAC,IAAI,EAAE,aAAa,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;IACtE,CAAC;IAED,MAAM,IAAA,yCAA+B,EAAC,IAAI,EAAE;QAC1C,sDAA8C;KAC/C,CAAC,CAAC;IACH,MAAM,IAAA,6BAAoB,EAAC,IAAI,CAAC,CAAC;IACjC,OAAO,KAAK,IAAI,EAAE;QAChB,IAAA,4BAAmB,EAAC,IAAI,CAAC,CAAC;QAC1B,MAAM,IAAI,wBAAU,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,oBAAM,EAAE,EAAE,IAAI,CAAC,CAAC,OAAO,EAAE,CAAC;IAChE,CAAC,CAAC;AACJ,CAAC,CAAC;AAnCW,QAAA,oCAAoC,wCAmC/C;AAEF,kBAAe,4CAAoC,CAAC"}

package/src/py/dynamodb/fast-api-connection/schema.d.ts

@@ -0,0 +1,12 @@
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+/**
+ * TypeScript types for options defined in schema.json
+ */
+export interface PyDynamoDBFastApiConnectionGeneratorSchema {
+  sourceProject: string;
+  targetProject: string;
+}

package/src/py/dynamodb/fast-api-connection/schema.json

@@ -0,0 +1,18 @@
+{
+  "$schema": "https://json-schema.org/schema",
+  "$id": "py#dynamodb#fast-api-connection",
+  "title": "py#dynamodb#fast-api-connection",
+  "description": "Connect a py#fast-api project to a py#dynamodb project",
+  "type": "object",
+  "properties": {
+    "sourceProject": {
+      "type": "string",
+      "description": "The py#fast-api project to connect from"
+    },
+    "targetProject": {
+      "type": "string",
+      "description": "The py#dynamodb project to connect to"
+    }
+  },
+  "required": ["sourceProject", "targetProject"]
+}

package/src/py/dynamodb/files/__name__/__init__.py.template

@@ -0,0 +1,6 @@
+from .entities import BaseModel, ExampleModel
+
+__all__ = [
+    'BaseModel',
+    'ExampleModel',
+]

package/src/py/dynamodb/files/__name__/client.py.template

@@ -0,0 +1,39 @@
+import json
+import os
+from typing import Any, cast
+
+# Generated: must match runtimeConfigKey in config.json
+runtime_config_key = '<%= runtimeConfigKey %>'
+
+# config.json is only present in local dev (not in the deployed bundle),
+# so calls to _serve_local must be guarded by is_local().
+def _serve_local() -> dict:
+    config_path = os.path.join(os.path.dirname(__file__), '..', 'config.json')
+    with open(config_path) as f:
+        return json.load(f)['serveLocal']
+
+
+def is_local() -> bool:
+    return os.environ.get("SERVE_LOCAL") == "true"
+
+
+_table_name: str | None = None
+
+
+def get_table_name() -> str:
+    global _table_name
+    if is_local():
+        return _serve_local()['tableName']
+    if _table_name is None:
+        from aws_lambda_powertools.utilities.parameters import AppConfigProvider
+
+        app_id = os.environ.get("RUNTIME_CONFIG_APP_ID")
+        if not app_id:
+            raise ValueError("RUNTIME_CONFIG_APP_ID environment variable is not set")
+        provider = AppConfigProvider(application=app_id, environment="default")
+        config = cast(dict[str, Any], provider.get("dynamodb", transform="json") or {})
+        table_name = (config.get(runtime_config_key) or {}).get("tableName")
+        if not table_name:
+            raise ValueError("Could not resolve table name from runtime config")
+        _table_name = table_name
+    return _table_name

package/src/py/dynamodb/files/__name__/entities/__init__.py.template

@@ -0,0 +1,4 @@
+from .base import BaseModel
+from .example import ExampleModel
+
+__all__ = ['BaseModel', 'ExampleModel']

package/src/py/dynamodb/files/__name__/entities/base.py.template

@@ -0,0 +1,63 @@
+import os
+
+from pynamodb.attributes import DiscriminatorAttribute, UnicodeAttribute
+from pynamodb.indexes import AllProjection, GlobalSecondaryIndex
+from pynamodb.models import Model
+
+from ..client import _serve_local, get_table_name, is_local
+
+
+class Gsi1Index(GlobalSecondaryIndex):
+    class Meta:
+        index_name = 'gsi1pk-gsi1sk-index'
+        projection = AllProjection()
+
+    gsi1pk = UnicodeAttribute(hash_key=True)
+    gsi1sk = UnicodeAttribute(range_key=True)
+
+
+class Gsi2Index(GlobalSecondaryIndex):
+    class Meta:
+        index_name = 'gsi2pk-gsi2sk-index'
+        projection = AllProjection()
+
+    gsi2pk = UnicodeAttribute(hash_key=True)
+    gsi2sk = UnicodeAttribute(range_key=True)
+
+
+class BaseModel(Model):
+    """Base model for all DynamoDB entities in this table.
+
+    All entity models MUST extend this class. It wires up the correct DynamoDB
+    table name at runtime — resolving from AWS AppConfig when deployed, or from
+    config.json when running locally via DynamoDB Local — by overriding
+    _get_connection(). Without this, PynamoDB will not know which table to use.
+
+    Also uses PynamoDB polymorphism to support multiple entity types in a single
+    table via DiscriminatorAttribute.
+    See https://pynamodb.readthedocs.io/en/stable/polymorphism.html
+    """
+
+    class Meta:
+        table_name: str
+        region = os.environ.get("AWS_REGION", 'us-east-1')
+        host = f"http://localhost:{_serve_local()['port']}" if is_local() else None
+        aws_access_key_id = 'UNUSED' if is_local() else None
+        aws_secret_access_key = 'UNUSED' if is_local() else None
+
+    entity_type = DiscriminatorAttribute()
+
+    pk = UnicodeAttribute(hash_key=True)
+    sk = UnicodeAttribute(range_key=True)
+    gsi1pk = UnicodeAttribute(null=True)
+    gsi1sk = UnicodeAttribute(null=True)
+    gsi2pk = UnicodeAttribute(null=True)
+    gsi2sk = UnicodeAttribute(null=True)
+
+    gsi1_index = Gsi1Index()
+    gsi2_index = Gsi2Index()
+
+    @classmethod
+    def _get_connection(cls):
+        cls.Meta.table_name = get_table_name()
+        return super()._get_connection()

package/src/py/dynamodb/files/__name__/entities/example.py.template

@@ -0,0 +1,64 @@
+from collections.abc import Iterator
+from datetime import UTC, datetime
+
+from pynamodb.attributes import UnicodeAttribute
+
+from .base import BaseModel
+
+
+# Example entity using PynamoDB - replace with your own entities
+# See https://pynamodb.readthedocs.io/en/stable/tutorial.html for documentation
+class ExampleModel(BaseModel, discriminator="ExampleModel"):
+    """Example entity demonstrating single-table design with GSI access patterns.
+
+    Key design:
+      pk=EXAMPLE#{id}, sk=EXAMPLE#{id}
+      gsi1pk=CATEGORY#{category}, gsi1sk=EXAMPLE#{id}
+      gsi2pk=EXAMPLE, gsi2sk={created_at ISO-8601}
+    """
+
+    name = UnicodeAttribute()
+    category = UnicodeAttribute()
+    created_at = UnicodeAttribute()
+    updated_at = UnicodeAttribute()
+
+    @classmethod
+    def make_pk(cls, id: str) -> str:
+        return f"EXAMPLE#{id}"
+
+    # ── Writes ────────────────────────────────────────────────────────────────
+    @classmethod
+    def create(cls, id: str, name: str, category: str) -> ExampleModel:
+        now = datetime.now(UTC).isoformat()
+        item = cls(
+            pk=cls.make_pk(id),
+            sk=cls.make_pk(id),
+            gsi1pk=f"CATEGORY#{category}",
+            gsi1sk=cls.make_pk(id),
+            gsi2pk="EXAMPLE",
+            gsi2sk=now,
+            name=name,
+            category=category,
+            created_at=now,
+            updated_at=now,
+        )
+        item.save()
+        return item
+
+    # ── Primary index ─────────────────────────────────────────────────────────
+    @classmethod
+    def get_by_id(cls, id: str) -> ExampleModel:
+        """Exact lookup by pk + sk."""
+        return cls.get(cls.make_pk(id), cls.make_pk(id))
+
+    # ── gsi1_index: partition=category, sort=id ───────────────────────────────
+    @classmethod
+    def list_by_category(cls, category: str) -> Iterator[ExampleModel]:
+        """All items in a category."""
+        return cls.gsi1_index.query(f"CATEGORY#{category}")
+
+    # ── gsi2_index: partition=type, sort=created_at ───────────────────────────
+    @classmethod
+    def list_by_date(cls, limit: int | None = None) -> Iterator[ExampleModel]:
+        """All examples, newest first."""
+        return cls.gsi2_index.query("EXAMPLE", limit=limit, scan_index_forward=False)

package/src/py/dynamodb/files/config.json.template

@@ -0,0 +1,24 @@
+{
+  "runtimeConfigKey": "<%= runtimeConfigKey %>",
+  "serveLocal": {
+    "port": <%= localDynamoDBPort %>,
+    "tableName": "<%= localTableName %>",
+    "image": "public.ecr.aws/aws-dynamodb-local/aws-dynamodb-local:latest",
+    "containerName": "<%= containerName %>",
+    "containerEngine": "<%= containerEngine %>"
+  },
+  "tableConfig": {
+    "globalSecondaryIndexes": [
+      {
+        "indexName": "gsi1pk-gsi1sk-index",
+        "partitionKey": "gsi1pk",
+        "sortKey": "gsi1sk"
+      },
+      {
+        "indexName": "gsi2pk-gsi2sk-index",
+        "partitionKey": "gsi2pk",
+        "sortKey": "gsi2sk"
+      }
+    ]
+  }
+}

package/src/py/dynamodb/generator.d.ts

@@ -0,0 +1,10 @@
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { type GeneratorCallback, type Tree } from '@nx/devkit';
+import { type NxGeneratorInfo } from '../../utils/nx';
+import type { PyDynamoDBGeneratorSchema } from './schema';
+export declare const PY_DYNAMODB_GENERATOR_INFO: NxGeneratorInfo;
+export declare const pyDynamoDBGenerator: (tree: Tree, options: PyDynamoDBGeneratorSchema) => Promise<GeneratorCallback>;
+export default pyDynamoDBGenerator;

package/src/py/dynamodb/generator.js

@@ -0,0 +1,106 @@
+"use strict";
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.pyDynamoDBGenerator = exports.PY_DYNAMODB_GENERATOR_INFO = void 0;
+const tslib_1 = require("tslib");
+const node_path_1 = require("node:path");
+const devkit_1 = require("@nx/devkit");
+const containers_1 = require("../../utils/containers");
+const dynamodb_constructs_1 = require("../../utils/dynamodb-constructs/dynamodb-constructs");
+const format_1 = require("../../utils/format");
+const iac_1 = require("../../utils/iac");
+const metrics_1 = require("../../utils/metrics");
+const names_1 = require("../../utils/names");
+const npm_scope_1 = require("../../utils/npm-scope");
+const nx_1 = require("../../utils/nx");
+const nxlv_python_1 = require("../../utils/nxlv-python");
+const port_1 = require("../../utils/port");
+const py_1 = require("../../utils/py");
+const shared_constructs_1 = require("../../utils/shared-constructs");
+const shared_constructs_constants_1 = require("../../utils/shared-constructs-constants");
+const shared_dynamodb_scripts_1 = require("../../utils/shared-dynamodb-scripts");
+const generator_1 = tslib_1.__importStar(require("../project/generator"));
+exports.PY_DYNAMODB_GENERATOR_INFO = (0, nx_1.getGeneratorInfo)(__filename);
+const pyDynamoDBGenerator = async (tree, options) => {
+    const nameClassName = (0, names_1.toClassName)(options.name);
+    const localTableName = `${(0, npm_scope_1.getNpmScope)(tree)}-${(0, names_1.kebabCase)(options.tableName ?? options.name)}`;
+    const containerEngine = await (0, containers_1.resolveContainers)(tree, 'inherit');
+    const { fullyQualifiedName, dir, normalizedModuleName } = (0, generator_1.getPyProjectDetails)(tree, {
+        name: options.name,
+        directory: options.directory,
+        subDirectory: options.subDirectory,
+    });
+    if (!(0, nx_1.projectExists)(tree, fullyQualifiedName)) {
+        await (0, generator_1.default)(tree, {
+            name: options.name,
+            directory: options.directory,
+            subDirectory: options.subDirectory,
+            type: 'library',
+        });
+    }
+    const projectConfig = (0, devkit_1.readProjectConfiguration)(tree, fullyQualifiedName);
+    const localDynamoDBPort = (0, port_1.assignSharedPort)(tree, projectConfig, shared_constructs_constants_1.DYNAMODB_GENERATOR_IDS, 8000);
+    const containerName = `${(0, npm_scope_1.getNpmScope)(tree)}-dynamodb`;
+    const templateOptions = {
+        name: normalizedModuleName,
+        runtimeConfigKey: nameClassName,
+        localDynamoDBPort,
+        localTableName,
+        containerName,
+        containerEngine,
+    };
+    (0, devkit_1.generateFiles)(tree, (0, devkit_1.joinPathFragments)(__dirname, 'files'), dir, templateOptions);
+    await (0, shared_dynamodb_scripts_1.sharedDynamoDBScriptsGenerator)(tree);
+    const scriptsDir = (0, node_path_1.relative)(dir, (0, devkit_1.joinPathFragments)(shared_constructs_constants_1.PACKAGES_DIR, shared_constructs_constants_1.SHARED_SCRIPTS_DIR, 'src', 'dynamodb'));
+    projectConfig.targets['pull-image'] = {
+        executor: 'nx:run-commands',
+        options: {
+            command: `tsx ${scriptsDir}/pull-image.ts`,
+            cwd: '{projectRoot}',
+        },
+    };
+    projectConfig.targets['serve-local'] = {
+        executor: 'nx:run-commands',
+        continuous: true,
+        dependsOn: ['pull-image'],
+        options: {
+            commands: [
+                `tsx ${scriptsDir}/start-container.ts`,
+                `tsx ${scriptsDir}/create-local-table.ts`,
+            ],
+            parallel: true,
+            cwd: '{projectRoot}',
+        },
+    };
+    (0, devkit_1.updateProjectConfiguration)(tree, fullyQualifiedName, projectConfig);
+    (0, nx_1.addGeneratorMetadata)(tree, fullyQualifiedName, exports.PY_DYNAMODB_GENERATOR_INFO);
+    if (options.infra !== 'none') {
+        const iac = await (0, iac_1.resolveIac)(tree, options.iac);
+        await (0, shared_constructs_1.sharedConstructsGenerator)(tree, { iac });
+        await (0, dynamodb_constructs_1.addDynamoDBInfra)(tree, {
+            iac,
+            projectName: fullyQualifiedName,
+            nameClassName,
+            nameKebabCase: (0, names_1.kebabCase)(options.name),
+            tableName: localTableName,
+            projectRoot: dir,
+        });
+    }
+    (0, py_1.addDependenciesToPyProjectToml)(tree, dir, [
+        'pynamodb',
+        'boto3',
+        'aws-lambda-powertools',
+    ]);
+    await (0, metrics_1.addGeneratorMetricsIfApplicable)(tree, [exports.PY_DYNAMODB_GENERATOR_INFO]);
+    await (0, format_1.formatFilesInSubtree)(tree);
+    return async () => {
+        (0, devkit_1.installPackagesTask)(tree);
+        await new nxlv_python_1.UVProvider(tree.root, new nxlv_python_1.Logger(), tree).install();
+    };
+};
+exports.pyDynamoDBGenerator = pyDynamoDBGenerator;
+exports.default = exports.pyDynamoDBGenerator;
+//# sourceMappingURL=generator.js.map

package/src/py/dynamodb/generator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"generator.js","sourceRoot":"","sources":["../../../../../../packages/nx-plugin/src/py/dynamodb/generator.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;;AAEH,yCAAqC;AACrC,uCAQoB;AACpB,uDAA2D;AAC3D,6FAAuF;AACvF,+CAA0D;AAC1D,yCAA6C;AAC7C,iDAAsE;AACtE,6CAA2D;AAC3D,qDAAoD;AACpD,uCAKwB;AACxB,yDAA6D;AAC7D,2CAAoD;AACpD,uCAAgE;AAChE,qEAA0E;AAC1E,yFAIiD;AACjD,iFAAqF;AACrF,0EAA+E;AAGlE,QAAA,0BAA0B,GACrC,IAAA,qBAAgB,EAAC,UAAU,CAAC,CAAC;AAExB,MAAM,mBAAmB,GAAG,KAAK,EACtC,IAAU,EACV,OAAkC,EACN,EAAE;IAC9B,MAAM,aAAa,GAAG,IAAA,mBAAW,EAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IAChD,MAAM,cAAc,GAAG,GAAG,IAAA,uBAAW,EAAC,IAAI,CAAC,IAAI,IAAA,iBAAS,EAAC,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;IAC9F,MAAM,eAAe,GAAG,MAAM,IAAA,8BAAiB,EAAC,IAAI,EAAE,SAAS,CAAC,CAAC;IACjE,MAAM,EAAE,kBAAkB,EAAE,GAAG,EAAE,oBAAoB,EAAE,GAAG,IAAA,+BAAmB,EAC3E,IAAI,EACJ;QACE,IAAI,EAAE,OAAO,CAAC,IAAI;QAClB,SAAS,EAAE,OAAO,CAAC,SAAS;QAC5B,YAAY,EAAE,OAAO,CAAC,YAAY;KACnC,CACF,CAAC;IAEF,IAAI,CAAC,IAAA,kBAAa,EAAC,IAAI,EAAE,kBAAkB,CAAC,EAAE,CAAC;QAC7C,MAAM,IAAA,mBAAkB,EAAC,IAAI,EAAE;YAC7B,IAAI,EAAE,OAAO,CAAC,IAAI;YAClB,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,YAAY,EAAE,OAAO,CAAC,YAAY;YAClC,IAAI,EAAE,SAAS;SAChB,CAAC,CAAC;IACL,CAAC;IAED,MAAM,aAAa,GAAG,IAAA,iCAAwB,EAAC,IAAI,EAAE,kBAAkB,CAAC,CAAC;IAEzE,MAAM,iBAAiB,GAAG,IAAA,uBAAgB,EACxC,IAAI,EACJ,aAAa,EACb,oDAAsB,EACtB,IAAI,CACL,CAAC;IAEF,MAAM,aAAa,GAAG,GAAG,IAAA,uBAAW,EAAC,IAAI,CAAC,WAAW,CAAC;IAEtD,MAAM,eAAe,GAAG;QACtB,IAAI,EAAE,oBAAoB;QAC1B,gBAAgB,EAAE,aAAa;QAC/B,iBAAiB;QACjB,cAAc;QACd,aAAa;QACb,eAAe;KAChB,CAAC;IAEF,IAAA,sBAAa,EACX,IAAI,EACJ,IAAA,0BAAiB,EAAC,SAAS,EAAE,OAAO,CAAC,EACrC,GAAG,EACH,eAAe,CAChB,CAAC;IAEF,MAAM,IAAA,wDAA8B,EAAC,IAAI,CAAC,CAAC;IAE3C,MAAM,UAAU,GAAG,IAAA,oBAAQ,EACzB,GAAG,EACH,IAAA,0BAAiB,EAAC,0CAAY,EAAE,gDAAkB,EAAE,KAAK,EAAE,UAAU,CAAC,CACvE,CAAC;IAEF,aAAa,CAAC,OAAO,CAAC,YAAY,CAAC,GAAG;QACpC,QAAQ,EAAE,iBAAiB;QAC3B,OAAO,EAAE;YACP,OAAO,EAAE,OAAO,UAAU,gBAAgB;YAC1C,GAAG,EAAE,eAAe;SACrB;KACF,CAAC;IACF,aAAa,CAAC,OAAO,CAAC,aAAa,CAAC,GAAG;QACrC,QAAQ,EAAE,iBAAiB;QAC3B,UAAU,EAAE,IAAI;QAChB,SAAS,EAAE,CAAC,YAAY,CAAC;QACzB,OAAO,EAAE;YACP,QAAQ,EAAE;gBACR,OAAO,UAAU,qBAAqB;gBACtC,OAAO,UAAU,wBAAwB;aAC1C;YACD,QAAQ,EAAE,IAAI;YACd,GAAG,EAAE,eAAe;SACrB;KACF,CAAC;IAEF,IAAA,mCAA0B,EAAC,IAAI,EAAE,kBAAkB,EAAE,aAAa,CAAC,CAAC;IACpE,IAAA,yBAAoB,EAAC,IAAI,EAAE,kBAAkB,EAAE,kCAA0B,CAAC,CAAC;IAE3E,IAAI,OAAO,CAAC,KAAK,KAAK,MAAM,EAAE,CAAC;QAC7B,MAAM,GAAG,GAAG,MAAM,IAAA,gBAAU,EAAC,IAAI,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC;QAChD,MAAM,IAAA,6CAAyB,EAAC,IAAI,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC;QAC/C,MAAM,IAAA,sCAAgB,EAAC,IAAI,EAAE;YAC3B,GAAG;YACH,WAAW,EAAE,kBAAkB;YAC/B,aAAa;YACb,aAAa,EAAE,IAAA,iBAAS,EAAC,OAAO,CAAC,IAAI,CAAC;YACtC,SAAS,EAAE,cAAc;YACzB,WAAW,EAAE,GAAG;SACjB,CAAC,CAAC;IACL,CAAC;IAED,IAAA,mCAA8B,EAAC,IAAI,EAAE,GAAG,EAAE;QACxC,UAAU;QACV,OAAO;QACP,uBAAuB;KACxB,CAAC,CAAC;IAEH,MAAM,IAAA,yCAA+B,EAAC,IAAI,EAAE,CAAC,kCAA0B,CAAC,CAAC,CAAC;IAE1E,MAAM,IAAA,6BAAoB,EAAC,IAAI,CAAC,CAAC;IACjC,OAAO,KAAK,IAAI,EAAE;QAChB,IAAA,4BAAmB,EAAC,IAAI,CAAC,CAAC;QAC1B,MAAM,IAAI,wBAAU,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,oBAAM,EAAE,EAAE,IAAI,CAAC,CAAC,OAAO,EAAE,CAAC;IAChE,CAAC,CAAC;AACJ,CAAC,CAAC;AA7GW,QAAA,mBAAmB,uBA6G9B;AAEF,kBAAe,2BAAmB,CAAC"}

package/src/py/dynamodb/mcp-server-connection/__snapshots__/generator.spec.ts.snap

@@ -0,0 +1,45 @@
+// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
+
+exports[`py#dynamodb mcp-server-connection generator > should add dynamodb serve-local dependency to mcp-server serve-local 1`] = `
+{
+  "$schema": "../../node_modules/nx/schemas/project-schema.json",
+  "name": "my-mcp-server",
+  "root": "packages/my-mcp-server",
+  "targets": {
+    "mcp-server-serve-local": {
+      "continuous": true,
+      "dependsOn": [
+        {
+          "projects": [
+            "db",
+          ],
+          "target": "serve-local",
+        },
+      ],
+      "executor": "nx:run-commands",
+    },
+  },
+}
+`;
+
+exports[`py#dynamodb mcp-server-connection generator > should use sourceComponent name when provided 1`] = `
+{
+  "$schema": "../../node_modules/nx/schemas/project-schema.json",
+  "name": "my-mcp-server",
+  "root": "packages/my-mcp-server",
+  "targets": {
+    "custom-mcp-server-serve-local": {
+      "continuous": true,
+      "dependsOn": [
+        {
+          "projects": [
+            "db",
+          ],
+          "target": "serve-local",
+        },
+      ],
+      "executor": "nx:run-commands",
+    },
+  },
+}
+`;

package/src/py/dynamodb/mcp-server-connection/generator.d.ts

@@ -0,0 +1,10 @@
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { type GeneratorCallback, type Tree } from '@nx/devkit';
+import { type NxGeneratorInfo } from '../../../utils/nx';
+import type { PyDynamoDBMcpServerConnectionGeneratorSchema } from './schema';
+export declare const PY_DYNAMODB_MCP_SERVER_CONNECTION_GENERATOR_INFO: NxGeneratorInfo;
+export declare const pyDynamoDBMcpServerConnectionGenerator: (tree: Tree, options: PyDynamoDBMcpServerConnectionGeneratorSchema) => Promise<GeneratorCallback>;
+export default pyDynamoDBMcpServerConnectionGenerator;