awslabs.dynamodb-mcp-server 0.1.3__py3-none-any.whl → 1.0.0__py3-none-any.whl

awslabs/dynamodb_mcp_server/common.py

@@ -1,3 +1,4 @@
+import os
 from functools import wraps
 from typing import Any, Callable, Dict, List, Literal, Optional
 from typing_extensions import TypedDict
@@ -26,6 +27,19 @@ def handle_exceptions(func: Callable) -> Callable:
     return wrapper
 
 
+def mutation_check(func):
+    """Decorator to block mutations if DDB-MCP-READONLY is set to true."""
+
+    @wraps(func)
+    async def wrapper(*args, **kwargs):
+        readonly = os.environ.get('DDB-MCP-READONLY', '').lower()
+        if readonly in ('true', '1', 'yes'):  # treat these as true
+            return {'error': 'Mutation not allowed: DDB-MCP-READONLY is set to true.'}
+        return await func(*args, **kwargs)
+
+    return wrapper
+
+
 # Type definitions
 AttributeValue = Dict[Literal['S', 'N', 'B', 'BOOL', 'NULL', 'L', 'M', 'SS', 'NS', 'BS'], Any]
 KeyAttributeValue = Dict[Literal['S', 'N', 'B'], Any]

awslabs/dynamodb_mcp_server/server.py

@@ -30,6 +30,7 @@ from awslabs.dynamodb_mcp_server.common import (
     UpdateTableInput,
     WarmThroughput,
     handle_exceptions,
+    mutation_check,
 )
 from botocore.config import Config
 from mcp.server.fastmcp import FastMCP
@@ -99,6 +100,7 @@ resource_arn: str = Field(description='The Amazon Resource Name (ARN) of the Dyn
 
 @app.tool()
 @handle_exceptions
+@mutation_check
 async def put_resource_policy(
     resource_arn: str = resource_arn,
     policy: Union[str, Dict[str, Any]] = Field(
@@ -236,6 +238,7 @@ async def query(
 
 @app.tool()
 @handle_exceptions
+@mutation_check
 async def update_item(
     table_name: str = table_name,
     key: Dict[str, KeyAttributeValue] = key,
@@ -303,6 +306,7 @@ async def get_item(
 
 @app.tool()
 @handle_exceptions
+@mutation_check
 async def put_item(
     table_name: str = table_name,
     item: Dict[str, AttributeValue] = Field(
@@ -337,6 +341,7 @@ async def put_item(
 
 @app.tool()
 @handle_exceptions
+@mutation_check
 async def delete_item(
     table_name: str = table_name,
     key: Dict[str, KeyAttributeValue] = key,
@@ -370,6 +375,7 @@ async def delete_item(
 
 @app.tool()
 @handle_exceptions
+@mutation_check
 async def update_time_to_live(
     table_name: str = table_name,
     time_to_live_specification: TimeToLiveSpecification = Field(
@@ -387,6 +393,7 @@ async def update_time_to_live(
 
 @app.tool()
 @handle_exceptions
+@mutation_check
 async def update_table(
     table_name: str = table_name,
     attribute_definitions: List[AttributeDefinition] = Field(
@@ -483,6 +490,7 @@ async def list_tables(
 
 @app.tool()
 @handle_exceptions
+@mutation_check
 async def create_table(
     table_name: str = Field(
         description='The name of the table to create.',
@@ -536,6 +544,7 @@ async def describe_table(
 
 @app.tool()
 @handle_exceptions
+@mutation_check
 async def create_backup(
     table_name: str = table_name,
     backup_name: str = Field(
@@ -602,6 +611,7 @@ async def list_backups(
 
 @app.tool()
 @handle_exceptions
+@mutation_check
 async def restore_table_from_backup(
     backup_arn: str = Field(
         description='The Amazon Resource Name (ARN) associated with the backup.',
@@ -717,6 +727,7 @@ async def describe_continuous_backups(
 
 @app.tool()
 @handle_exceptions
+@mutation_check
 async def untag_resource(
     resource_arn: str = resource_arn,
     tag_keys: List[str] = Field(description='List of tags to remove.', min_length=1),
@@ -730,6 +741,7 @@ async def untag_resource(
 
 @app.tool()
 @handle_exceptions
+@mutation_check
 async def tag_resource(
     resource_arn: str = resource_arn,
     tags: List[Tag] = Field(description='Tags to be assigned.'),
@@ -762,6 +774,7 @@ async def list_tags_of_resource(
 
 @app.tool()
 @handle_exceptions
+@mutation_check
 async def delete_table(
     table_name: str = table_name,
     region_name: str = Field(default=None, description='The aws region to run the tool'),
@@ -802,6 +815,25 @@ async def update_continuous_backups(
     return response['ContinuousBackupsDescription']
 
 
+@app.tool()
+@handle_exceptions
+async def list_imports(
+    next_token: str = Field(default=None, description='Token to fetch the next page of results.'),
+    region_name: str = Field(default=None, description='The aws region to run the tool'),
+) -> dict:
+    """Lists imports completed within the past 90 days."""
+    client = get_dynamodb_client(region_name)
+    params = {}
+    if next_token:
+        params['NextToken'] = next_token
+    params['PageSize'] = 25
+    response = client.list_imports(**params)
+    return {
+        'ImportSummaryList': response.get('ImportSummaryList', []),
+        'NextToken': response.get('NextToken'),
+    }
+
+
 def main():
     """Main entry point for the MCP server application."""
     app.run()

{awslabs_dynamodb_mcp_server-0.1.3.dist-info → awslabs_dynamodb_mcp_server-1.0.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: awslabs.dynamodb-mcp-server
-Version: 0.1.3
+Version: 1.0.0
 Summary: The official MCP Server for interacting with AWS DynamoDB
 Project-URL: homepage, https://awslabs.github.io/mcp/
 Project-URL: docs, https://awslabs.github.io/mcp/servers/dynamodb-mcp-server/
@@ -104,6 +104,7 @@ Add the MCP to your favorite agentic tools. e.g. for Amazon Q Developer CLI MCP,
       "command": "uvx",
       "args": ["awslabs.dynamodb-mcp-server@latest"],
       "env": {
+        "DDB-MCP-READONLY": "true",
         "AWS_PROFILE": "default",
         "AWS_REGION": "us-west-2",
         "FASTMCP_LOG_LEVEL": "ERROR"

awslabs_dynamodb_mcp_server-1.0.0.dist-info/RECORD

@@ -0,0 +1,10 @@
+awslabs/__init__.py,sha256=47wJeKcStxEJwX7SVVV2pnAWYR8FxcaYoT3YTmZ5Plg,674
+awslabs/dynamodb_mcp_server/__init__.py,sha256=v9gwAZRDfCdRfC-s0e9gUXPK3rzXrdHjBywl4e05cGs,616
+awslabs/dynamodb_mcp_server/common.py,sha256=TP05-UxC7b_IUfuO_Y-DsySPfkAmPl29ltgleemHfWk,10806
+awslabs/dynamodb_mcp_server/server.py,sha256=yjEvGXecHaUnwoDLy5UPy_A5repNLltduXQclejy9pA,32307
+awslabs_dynamodb_mcp_server-1.0.0.dist-info/METADATA,sha256=8yuKIKsejg7ubjyTkzzddk1w_snPyo1myTQZ8YqOJLA,5892
+awslabs_dynamodb_mcp_server-1.0.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+awslabs_dynamodb_mcp_server-1.0.0.dist-info/entry_points.txt,sha256=Vn6TvAN9d67Lsbkcs0UcIiOBI5xDpNBm_MOOzc1h-YU,88
+awslabs_dynamodb_mcp_server-1.0.0.dist-info/licenses/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+awslabs_dynamodb_mcp_server-1.0.0.dist-info/licenses/NOTICE,sha256=47UMmTFkf8rUc_JaJfdWe6NsAJQOcZNPZIL6JzU_k5U,95
+awslabs_dynamodb_mcp_server-1.0.0.dist-info/RECORD,,

awslabs_dynamodb_mcp_server-0.1.3.dist-info/RECORD

@@ -1,10 +0,0 @@
-awslabs/__init__.py,sha256=47wJeKcStxEJwX7SVVV2pnAWYR8FxcaYoT3YTmZ5Plg,674
-awslabs/dynamodb_mcp_server/__init__.py,sha256=v9gwAZRDfCdRfC-s0e9gUXPK3rzXrdHjBywl4e05cGs,616
-awslabs/dynamodb_mcp_server/common.py,sha256=RIR5d1tfWteCy7Iv-vU_6Sip2zERvOapwe6eTCH9EF0,10351
-awslabs/dynamodb_mcp_server/server.py,sha256=UYhvh-o1jdryASMGI1Km0wtFavfqcUC9uHjeJsdiidQ,31449
-awslabs_dynamodb_mcp_server-0.1.3.dist-info/METADATA,sha256=LJbrCxKxVScbcRW1wyqmT56Y1IAFpmhtYSPBof2u5bk,5856
-awslabs_dynamodb_mcp_server-0.1.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-awslabs_dynamodb_mcp_server-0.1.3.dist-info/entry_points.txt,sha256=Vn6TvAN9d67Lsbkcs0UcIiOBI5xDpNBm_MOOzc1h-YU,88
-awslabs_dynamodb_mcp_server-0.1.3.dist-info/licenses/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
-awslabs_dynamodb_mcp_server-0.1.3.dist-info/licenses/NOTICE,sha256=47UMmTFkf8rUc_JaJfdWe6NsAJQOcZNPZIL6JzU_k5U,95
-awslabs_dynamodb_mcp_server-0.1.3.dist-info/RECORD,,