awslabs.dynamodb-mcp-server 1.0.9__py3-none-any.whl → 2.0.2__py3-none-any.whl

awslabs/__init__.py

@@ -14,3 +14,23 @@
 
 # This file is part of the awslabs namespace.
 # It is intentionally minimal to support PEP 420 namespace packages.
+
+# Namespace Package Configuration
+#
+# This line resolves namespace conflicts when multiple packages share the 'awslabs' namespace prefix.
+# Without this configuration, test suites fail and build issues occur because Python cannot properly
+# resolve which package owns the 'awslabs' namespace when both 'awslabs.dynamodb-mcp-server' and
+# 'awslabs.mysql-mcp-server' are installed in the same environment.
+#
+# The extend_path() function implements PEP 420 namespace packages, allowing multiple distributions
+# to contribute modules to the same namespace. This ensures that:
+# 1. Both DynamoDB and MySQL MCP servers can coexist in the same Python environment
+# 2. Import statements like 'from awslabs.dynamodb_mcp_server import ...' work correctly
+# 3. Test discovery and execution functions properly across both packages
+# 4. Build processes complete successfully without namespace collision errors
+#
+# This is the standard solution for namespace packages in Python and is required for proper
+# multi-package namespace support in the awslabs ecosystem.
+
+# Extend namespace to include installed packages
+__path__ = __import__('pkgutil').extend_path(__path__, __name__)

awslabs/dynamodb_mcp_server/__init__.py

@@ -14,4 +14,4 @@
 
 """awslabs.dynamodb-mcp-server"""
 
-__version__ = '1.0.9'
+__version__ = '2.0.2'

awslabs/dynamodb_mcp_server/common.py

@@ -14,8 +14,7 @@
 
 import os
 from functools import wraps
-from typing import Any, Callable, Dict, List, Literal, Optional
-from typing_extensions import TypedDict
+from typing import Callable
 
 
 def handle_exceptions(func: Callable) -> Callable:
@@ -52,283 +51,3 @@ def mutation_check(func):
         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]
-
-# Return value enums
-ReturnValue = Literal['NONE', 'ALL_OLD', 'UPDATED_OLD', 'ALL_NEW', 'UPDATED_NEW']
-ReturnConsumedCapacity = Literal['INDEXES', 'TOTAL', 'NONE']
-ReturnItemCollectionMetrics = Literal['SIZE', 'NONE']
-Select = Literal['ALL_ATTRIBUTES', 'ALL_PROJECTED_ATTRIBUTES', 'SPECIFIC_ATTRIBUTES', 'COUNT']
-
-
-class ScanInput(TypedDict, total=False):
-    """Parameters for Scan operation."""
-
-    TableName: str  # required
-    IndexName: Optional[str]
-    AttributesToGet: Optional[List[str]]  # Legacy parameter
-    Limit: Optional[int]
-    Select: Optional[Select]
-    ScanFilter: Optional[
-        Dict[str, AttributeValue]
-    ]  # Legacy parameter (must use AttributeValue format e.g. {'S': 'value'})
-    ConditionalOperator: Optional[Literal['AND', 'OR']]  # Legacy parameter
-    ExclusiveStartKey: Optional[
-        Dict[str, KeyAttributeValue]
-    ]  # Primary key attributes in AttributeValue format e.g. {'S': 'value'}
-    ReturnConsumedCapacity: Optional[ReturnConsumedCapacity]
-    TotalSegments: Optional[int]
-    Segment: Optional[int]
-    ProjectionExpression: Optional[str]
-    FilterExpression: Optional[str]
-    ExpressionAttributeNames: Optional[Dict[str, str]]
-    ExpressionAttributeValues: Optional[
-        Dict[str, AttributeValue]
-    ]  # values must use AttributeValue format e.g. {'S': 'value'}
-    ConsistentRead: Optional[bool]
-
-
-class QueryInput(TypedDict, total=False):
-    """Parameters for Query operation."""
-
-    TableName: str  # required
-    IndexName: Optional[str]
-    Select: Optional[Select]
-    AttributesToGet: Optional[List[str]]  # Legacy parameter
-    Limit: Optional[int]
-    ConsistentRead: Optional[bool]
-    KeyConditionExpression: Optional[str]
-    FilterExpression: Optional[str]
-    ProjectionExpression: Optional[str]
-    ExpressionAttributeNames: Optional[Dict[str, str]]
-    ExpressionAttributeValues: Optional[
-        Dict[str, AttributeValue]
-    ]  # values must use AttributeValue format e.g. {'S': 'value'}
-    ScanIndexForward: Optional[bool]
-    ExclusiveStartKey: Optional[
-        Dict[str, KeyAttributeValue]
-    ]  # Primary key attributes in AttributeValue format e.g. {'S': 'value'}
-    ReturnConsumedCapacity: Optional[ReturnConsumedCapacity]
-
-
-class DeleteItemInput(TypedDict, total=False):
-    """Parameters for DeleteItem operation."""
-
-    TableName: str  # required
-    Key: Dict[
-        str, KeyAttributeValue
-    ]  # required - primary key attributes in AttributeValue format e.g. {'S': 'value'}
-    ConditionExpression: Optional[str]
-    ExpressionAttributeNames: Optional[Dict[str, str]]
-    ExpressionAttributeValues: Optional[
-        Dict[str, AttributeValue]
-    ]  # values must use AttributeValue format e.g. {'S': 'value'}
-    ReturnConsumedCapacity: Optional[ReturnConsumedCapacity]
-    ReturnItemCollectionMetrics: Optional[ReturnItemCollectionMetrics]
-    ReturnValues: Optional[ReturnValue]
-    ReturnValuesOnConditionCheckFailure: Optional[Literal['ALL_OLD', 'NONE']]
-
-
-class UpdateItemInput(TypedDict, total=False):
-    """Parameters for UpdateItem operation."""
-
-    TableName: str  # required
-    Key: Dict[
-        str, KeyAttributeValue
-    ]  # required - primary key attributes in AttributeValue format e.g. {'S': 'value'}
-    UpdateExpression: Optional[str]
-    ConditionExpression: Optional[str]
-    ExpressionAttributeNames: Optional[Dict[str, str]]
-    ExpressionAttributeValues: Optional[
-        Dict[str, AttributeValue]
-    ]  # values must use AttributeValue format e.g. {'S': 'value'}
-    ReturnConsumedCapacity: Optional[ReturnConsumedCapacity]
-    ReturnItemCollectionMetrics: Optional[ReturnItemCollectionMetrics]
-    ReturnValues: Optional[ReturnValue]
-    ReturnValuesOnConditionCheckFailure: Optional[Literal['ALL_OLD', 'NONE']]
-
-
-class GetItemInput(TypedDict, total=False):
-    """Parameters for GetItem operation."""
-
-    TableName: str  # required
-    Key: Dict[
-        str, KeyAttributeValue
-    ]  # required - primary key attributes in AttributeValue format e.g. {'S': 'value'}
-    AttributesToGet: Optional[List[str]]
-    ConsistentRead: Optional[bool]
-    ExpressionAttributeNames: Optional[Dict[str, str]]
-    ProjectionExpression: Optional[str]
-    ReturnConsumedCapacity: Optional[ReturnConsumedCapacity]
-
-
-class PutItemInput(TypedDict, total=False):
-    """Parameters for PutItem operation."""
-
-    TableName: str  # required
-    Item: Dict[
-        str, AttributeValue
-    ]  # required - maps attribute name to AttributeValue (must use AttributeValue format e.g. {'S': 'value'})
-    ConditionExpression: Optional[str]
-    ExpressionAttributeNames: Optional[Dict[str, str]]
-    ExpressionAttributeValues: Optional[
-        Dict[str, AttributeValue]
-    ]  # values must use AttributeValue format e.g. {'S': 'value'}
-    ReturnConsumedCapacity: Optional[ReturnConsumedCapacity]
-    ReturnItemCollectionMetrics: Optional[ReturnItemCollectionMetrics]
-    ReturnValues: Optional[ReturnValue]
-    ReturnValuesOnConditionCheckFailure: Optional[Literal['ALL_OLD', 'NONE']]
-
-
-class AttributeDefinition(TypedDict):
-    AttributeName: str
-    AttributeType: Literal['S', 'N', 'B']
-
-
-class KeySchemaElement(TypedDict):
-    AttributeName: str
-    KeyType: Literal['HASH', 'RANGE']
-
-
-class ProvisionedThroughput(TypedDict):
-    ReadCapacityUnits: int
-    WriteCapacityUnits: int
-
-
-class Projection(TypedDict, total=False):
-    ProjectionType: Literal['KEYS_ONLY', 'INCLUDE', 'ALL']
-    NonKeyAttributes: List[str]
-
-
-class OnDemandThroughput(TypedDict, total=False):
-    MaxReadRequestUnits: int
-    MaxWriteRequestUnits: int
-
-
-class WarmThroughput(TypedDict, total=False):
-    ReadUnitsPerSecond: int
-    WriteUnitsPerSecond: int
-
-
-class GlobalSecondaryIndex(TypedDict, total=False):
-    IndexName: str  # required
-    KeySchema: List[KeySchemaElement]  # required
-    Projection: Projection  # required
-    ProvisionedThroughput: ProvisionedThroughput
-    OnDemandThroughput: OnDemandThroughput
-
-
-class GlobalSecondaryIndexUpdateAction(TypedDict, total=False):
-    IndexName: str
-    ProvisionedThroughput: ProvisionedThroughput
-    OnDemandThroughput: OnDemandThroughput
-    WarmThroughput: WarmThroughput
-
-
-class GlobalSecondaryIndexDeleteAction(TypedDict):
-    IndexName: str
-
-
-class GlobalSecondaryIndexUpdate(TypedDict, total=False):
-    Create: GlobalSecondaryIndex
-    Delete: GlobalSecondaryIndexDeleteAction
-    Update: GlobalSecondaryIndexUpdateAction
-
-
-class StreamSpecification(TypedDict, total=False):
-    StreamEnabled: bool
-    StreamViewType: Literal['KEYS_ONLY', 'NEW_IMAGE', 'OLD_IMAGE', 'NEW_AND_OLD_IMAGES']
-
-
-class Tag(TypedDict):
-    Key: str
-    Value: str
-
-
-class SSESpecification(TypedDict, total=False):
-    """Set Enabled to true for AWS managed key (KMS charges apply). set it to false for AWS owned key."""
-
-    Enabled: bool
-    SSEType: Literal['KMS']
-    KMSMasterKeyId: str
-
-
-class TimeToLiveSpecification(TypedDict):
-    AttributeName: str  # The name of the TTL attribute used to store the expiration time for items
-    Enabled: bool  # Indicates whether TTL is enabled (true) or disabled (false) on the table
-
-
-class GetResourcePolicyInput(TypedDict):
-    ResourceArn: str  # The Amazon Resource Name (ARN) of the DynamoDB resource to which the policy is attached
-
-
-class PutResourcePolicyInput(TypedDict, total=False):
-    Policy: str  # An AWS resource-based policy document in JSON format
-    ResourceArn: str  # The Amazon Resource Name (ARN) of the DynamoDB resource to which the policy will be attached
-    ConfirmRemoveSelfResourceAccess: (
-        bool  # Set to true to confirm removing your permissions to change the policy in the future
-    )
-    ExpectedRevisionId: str  # A string value for conditional updates of your policy
-
-
-class OnDemandThroughputOverride(TypedDict):
-    MaxReadRequestUnits: int
-
-
-class ProvisionedThroughputOverride(TypedDict):
-    ReadCapacityUnits: int
-
-
-class ReplicaCreate(TypedDict, total=False):
-    RegionName: str
-    KMSMasterKeyId: str
-
-
-class ReplicaDelete(TypedDict):
-    RegionName: str
-
-
-class ReplicaUpdate(TypedDict, total=False):
-    KMSMasterKeyId: str
-    OnDemandThroughputOverride: OnDemandThroughputOverride
-    ProvisionedThroughputOverride: ProvisionedThroughputOverride
-    RegionName: str
-    TableClassOverride: Literal['STANDARD', 'STANDARD_INFREQUENT_ACCESS']
-
-
-class ReplicationGroupUpdate(TypedDict, total=False):
-    Create: ReplicaCreate
-    Update: ReplicaUpdate
-    Delete: ReplicaDelete
-
-
-class CreateTableInput(TypedDict, total=False):
-    """Parameters for CreateTable operation."""
-
-    TableName: str  # required
-    AttributeDefinitions: List[AttributeDefinition]  # required
-    KeySchema: List[KeySchemaElement]  # required
-    BillingMode: Literal['PROVISIONED', 'PAY_PER_REQUEST']
-    GlobalSecondaryIndexes: List[GlobalSecondaryIndex]
-    ProvisionedThroughput: ProvisionedThroughput
-
-
-class UpdateTableInput(TypedDict, total=False):
-    """Parameters for UpdateTable operation."""
-
-    TableName: str  # required
-    AttributeDefinitions: List[AttributeDefinition]
-    BillingMode: Literal['PROVISIONED', 'PAY_PER_REQUEST']
-    DeletionProtectionEnabled: bool
-    GlobalSecondaryIndexUpdates: List[GlobalSecondaryIndexUpdate]
-    OnDemandThroughput: OnDemandThroughput
-    ProvisionedThroughput: ProvisionedThroughput
-    ReplicaUpdates: List[ReplicationGroupUpdate]
-    SSESpecification: SSESpecification
-    StreamSpecification: StreamSpecification
-    TableClass: Literal['STANDARD', 'STANDARD_INFREQUENT_ACCESS']
-    WarmThroughput: WarmThroughput

awslabs/dynamodb_mcp_server/database_analysis_queries.py

@@ -0,0 +1,210 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Source Database Analysis SQL Query Resources for DynamoDB Data Modeling."""
+
+from typing import Any, Dict
+
+
+# SQL Query Templates for MySQL
+mysql_analysis_queries = {
+    'performance_schema_check': {
+        'name': 'Performance Schema Status Check',
+        'description': 'Returns the status of the performance_schema system variable (ON/OFF)',
+        'sql': 'SELECT @@performance_schema;',
+        'parameters': [],
+    },
+    'query_pattern_analysis': {
+        'name': 'Query Pattern Analysis',
+        'description': 'Returns query patterns from Performance Schema with execution counts, calculated RPS, average execution time, average rows examined per execution, scan counts, execution timeframes, and SQL complexity classification',
+        'sql': """SELECT
+  -- Basic pattern information
+  DIGEST_TEXT as query_pattern,
+  COUNT_STAR as frequency,
+  -- Timing information
+  FIRST_SEEN as first_seen,
+  LAST_SEEN as last_seen
+FROM performance_schema.events_statements_summary_by_digest
+WHERE SCHEMA_NAME = '{target_database}'
+AND COUNT_STAR > 0
+AND LAST_SEEN >= DATE_SUB(NOW(), INTERVAL {pattern_analysis_days} DAY)
+-- Exclude system and administrative queries
+AND DIGEST_TEXT NOT LIKE 'SET%'
+AND DIGEST_TEXT NOT LIKE 'USE %'
+AND DIGEST_TEXT NOT LIKE 'SHOW%'
+AND DIGEST_TEXT NOT LIKE '/* RDS Data API */%'
+AND DIGEST_TEXT NOT LIKE '%information_schema%'
+AND DIGEST_TEXT NOT LIKE '%performance_schema%'
+AND DIGEST_TEXT NOT LIKE '%mysql.%'
+AND DIGEST_TEXT NOT LIKE 'SELECT @@%'
+AND DIGEST_TEXT NOT LIKE '%sys.%'
+AND DIGEST_TEXT NOT LIKE 'select ?'
+AND DIGEST_TEXT NOT LIKE '%mysql.general_log%'
+AND DIGEST_TEXT NOT LIKE 'DESCRIBE %'
+AND DIGEST_TEXT NOT LIKE 'EXPLAIN %'
+AND DIGEST_TEXT NOT LIKE '%configured_database%'
+AND DIGEST_TEXT NOT LIKE 'FLUSH %'
+AND DIGEST_TEXT NOT LIKE 'RESET %'
+AND DIGEST_TEXT NOT LIKE 'OPTIMIZE %'
+AND DIGEST_TEXT NOT LIKE 'ANALYZE %'
+AND DIGEST_TEXT NOT LIKE 'CHECK %'
+AND DIGEST_TEXT NOT LIKE 'REPAIR %'
+AND DIGEST_TEXT NOT LIKE '%@@default_storage_engine%'
+AND DIGEST_TEXT NOT LIKE '%@%:=%'
+AND DIGEST_TEXT NOT LIKE '%MD5%'
+AND DIGEST_TEXT NOT LIKE '%SHA%'
+AND DIGEST_TEXT NOT LIKE '%CONCAT_WS%'
+ORDER BY frequency DESC;""",
+        'parameters': ['target_database', 'pattern_analysis_days'],
+    },
+    'table_analysis': {
+        'name': 'Table Structure Analysis',
+        'description': 'Returns table statistics including row counts, data size in MB, index size in MB, column counts, foreign key counts, and creation/modification timestamps',
+        'sql': """SELECT
+  TABLE_NAME,
+  TABLE_ROWS,
+  -- Storage sizes in MB
+  ROUND(DATA_LENGTH/1024/1024, 2) as datamb,
+  ROUND(INDEX_LENGTH/1024/1024, 2) as indexmb,
+  -- Count columns in this table
+  (SELECT COUNT(*) FROM information_schema.COLUMNS c
+   WHERE c.TABLE_SCHEMA = t.TABLE_SCHEMA AND c.TABLE_NAME = t.TABLE_NAME) as columncount,
+  -- Count foreign keys in this table
+  (SELECT COUNT(*) FROM information_schema.KEY_COLUMN_USAGE k
+   WHERE k.TABLE_SCHEMA = t.TABLE_SCHEMA AND k.TABLE_NAME = t.TABLE_NAME
+   AND k.REFERENCED_TABLE_NAME IS NOT NULL) as fkcount,
+  CREATE_TIME,
+  UPDATE_TIME
+FROM information_schema.TABLES t
+WHERE TABLE_SCHEMA = '{target_database}'
+ORDER BY TABLE_ROWS DESC;""",
+        'parameters': ['target_database'],
+    },
+    'column_analysis': {
+        'name': 'Column Information Analysis',
+        'description': 'Returns all column definitions including table name, column name, data type, nullability, key type, default value, and extra attributes',
+        'sql': """SELECT
+  TABLE_NAME,
+  COLUMN_NAME,
+  COLUMN_TYPE,
+  IS_NULLABLE,
+  COLUMN_KEY,
+  COLUMN_DEFAULT,
+  EXTRA
+FROM information_schema.COLUMNS
+WHERE TABLE_SCHEMA = '{target_database}'
+ORDER BY TABLE_NAME, ORDINAL_POSITION;""",
+        'parameters': ['target_database'],
+    },
+    'index_analysis': {
+        'name': 'Index Statistics Analysis',
+        'description': 'Returns index structure details including table name, index name, column name, uniqueness flag, and column position within each index',
+        'sql': """SELECT
+  TABLE_NAME,
+  INDEX_NAME,
+  COLUMN_NAME,
+  NON_UNIQUE,
+  SEQ_IN_INDEX
+FROM information_schema.STATISTICS
+WHERE TABLE_SCHEMA = '{target_database}'
+ORDER BY TABLE_NAME, INDEX_NAME, SEQ_IN_INDEX;""",
+        'parameters': ['target_database'],
+    },
+    'foreign_key_analysis': {
+        'name': 'Foreign Key Relationship Analysis',
+        'description': 'Returns foreign key relationships including constraint names, child/parent table and column mappings, referential action rules, and estimated relationship cardinality',
+        'sql': """SELECT
+  kcu.CONSTRAINT_NAME,
+  kcu.TABLE_NAME as child_table,
+  kcu.COLUMN_NAME as child_column,
+  kcu.REFERENCED_TABLE_NAME as parent_table,
+  kcu.REFERENCED_COLUMN_NAME as parent_column,
+  rc.UPDATE_RULE,
+  rc.DELETE_RULE,
+  -- Estimate relationship cardinality based on unique constraints
+  CASE
+    WHEN EXISTS (
+      SELECT 1 FROM information_schema.STATISTICS s
+      WHERE s.TABLE_SCHEMA = '{target_database}'
+      AND s.TABLE_NAME = kcu.TABLE_NAME
+      AND s.COLUMN_NAME = kcu.COLUMN_NAME
+      AND s.NON_UNIQUE = 0  -- Unique constraint exists
+      AND (SELECT COUNT(*) FROM information_schema.KEY_COLUMN_USAGE kcu2
+           WHERE kcu2.CONSTRAINT_NAME = s.INDEX_NAME
+           AND kcu2.TABLE_SCHEMA = s.TABLE_SCHEMA) = 1  -- Single column constraint
+    ) THEN '1:1 or 1:0..1'
+    ELSE '1:Many'
+  END as estimated_cardinality
+FROM information_schema.KEY_COLUMN_USAGE kcu
+LEFT JOIN information_schema.REFERENTIAL_CONSTRAINTS rc
+  ON kcu.CONSTRAINT_NAME = rc.CONSTRAINT_NAME
+  AND kcu.CONSTRAINT_SCHEMA = rc.CONSTRAINT_SCHEMA
+WHERE kcu.TABLE_SCHEMA = '{target_database}'
+  AND kcu.REFERENCED_TABLE_NAME IS NOT NULL  -- Only foreign key constraints
+ORDER BY kcu.TABLE_NAME, kcu.COLUMN_NAME;""",
+        'parameters': ['target_database'],
+    },
+    'database_objects': {
+        'name': 'Database Objects Summary',
+        'description': 'Returns object counts and concatenated names grouped by object type: tables, triggers, stored procedures, and functions',
+        'sql': """SELECT
+  'Tables' as object_type,
+  COUNT(*) as count,
+  GROUP_CONCAT(TABLE_NAME) as names
+FROM information_schema.TABLES
+WHERE TABLE_SCHEMA = '{target_database}'
+UNION ALL
+SELECT
+  'Triggers' as object_type,
+  COUNT(*) as count,
+  COALESCE(GROUP_CONCAT(TRIGGER_NAME), 'None') as names
+FROM information_schema.TRIGGERS
+WHERE TRIGGER_SCHEMA = '{target_database}'
+UNION ALL
+SELECT
+  'Stored Procedures' as object_type,
+  COUNT(*) as count,
+  COALESCE(GROUP_CONCAT(ROUTINE_NAME), 'None') as names
+FROM information_schema.ROUTINES
+WHERE ROUTINE_SCHEMA = '{target_database}'
+AND ROUTINE_TYPE = 'PROCEDURE'
+UNION ALL
+SELECT
+  'Functions' as object_type,
+  COUNT(*) as count,
+  COALESCE(GROUP_CONCAT(ROUTINE_NAME), 'None') as names
+FROM information_schema.ROUTINES
+WHERE ROUTINE_SCHEMA = '{target_database}'
+AND ROUTINE_TYPE = 'FUNCTION';""",
+        'parameters': ['target_database'],
+    },
+}
+
+
+def get_query_resource(query_name: str, max_query_results: int, **params) -> Dict[str, Any]:
+    """Get a SQL query resource with parameters substituted."""
+    if query_name not in mysql_analysis_queries:
+        raise ValueError(f"Query '{query_name}' not found")
+
+    query_info = mysql_analysis_queries[query_name].copy()
+
+    # Substitute parameters in SQL
+    if params:
+        query_info['sql'] = query_info['sql'].format(**params)
+
+    # Apply LIMIT to all queries
+    sql = query_info['sql'].rstrip(';')
+    query_info['sql'] = f'{sql} LIMIT {max_query_results};'
+
+    return query_info