Fix pylint warnings

Author: Michael Christensen

src/amazon-keyspaces-mcp-server/awslabs/amazon_keyspaces_mcp_server/server.py

@@ -371,7 +371,7 @@ async def list_keyspaces(
     description='Lists all tables in a specified keyspace - args: keyspace',
 )
 async def list_tables(
-    input: KeyspaceInput,
+    input: KeyspaceInput,  # pylint: disable=redefined-builtin
     ctx: Optional[Context] = None,
 ) -> str:
     """Lists all tables in a specified keyspace."""
@@ -384,7 +384,7 @@ async def list_tables(
     description='Gets detailed information about a keyspace - args: keyspace',
 )
 async def describe_keyspace(
-    input: KeyspaceInput,
+    input: KeyspaceInput,  # pylint: disable=redefined-builtin
     ctx: Optional[Context] = None,
 ) -> str:
     """Gets detailed information about a keyspace."""
@@ -397,7 +397,7 @@ async def describe_keyspace(
     description='Gets detailed information about a table - args: keyspace, table',
 )
 async def describe_table(
-    input: TableInput,
+    input: TableInput,  # pylint: disable=redefined-builtin
     ctx: Optional[Context] = None,
 ) -> str:
     """Gets detailed information about a table."""
@@ -410,7 +410,7 @@ async def describe_table(
     description='Executes a read-only SELECT query against the database - args: keyspace, query',
 )
 async def execute_query(
-    input: QueryInput,
+    input: QueryInput,  # pylint: disable=redefined-builtin
     ctx: Optional[Context] = None,
 ) -> str:
     """Executes a read-only (SELECT) query against the database."""
@@ -423,7 +423,7 @@ async def execute_query(
     description='Analyzes the performance characteristics of a CQL query - args: keyspace, query',
 )
 async def analyze_query_performance(
-    input: QueryInput,
+    input: QueryInput,  # pylint: disable=redefined-builtin
     ctx: Optional[Context] = None,
 ) -> str:
     """Analyzes the performance characteristics of a CQL query."""
@@ -732,17 +732,17 @@ async def _handle_analyze_query_performance(
 
 def main():
     """Run the MCP server."""
-    import asyncio
-    
+    import asyncio  # pylint: disable=import-outside-toplevel
+
     # Validate connection before starting server
     try:
         proxy = asyncio.run(get_proxy())
         asyncio.run(proxy.schema_service.cassandra_client.get_session())
         logger.success('Successfully validated database connection')
-    except Exception as e:
+    except Exception as e:  # pylint: disable=broad-exception-caught
         logger.error(f'Failed to connect to database: {e}')
         sys.exit(1)
-    
+
     mcp.run()
 
 

src/amazon-keyspaces-mcp-server/tests/test_client.py

@@ -222,7 +222,9 @@ async def test_ssl_context_load_error(self, mock_ssl, mock_cluster_class):
         mock_ssl.SSLError = ssl.SSLError
 
         # Make load_verify_locations raise an exception
-        mock_ssl_context.load_verify_locations.side_effect = FileNotFoundError('Certificate not found')
+        mock_ssl_context.load_verify_locations.side_effect = FileNotFoundError(
+            'Certificate not found'
+        )
 
         # Create the client and get session
         client = UnifiedCassandraClient(self.keyspaces_config)
@@ -728,7 +730,9 @@ async def test_execute_read_only_query_with_column_error(self):
         mock_column_names = ['id', 'bad_column']
         mock_row = Mock()
         mock_row.id = 1
-        type(mock_row).bad_column = property(lambda self: (_ for _ in ()).throw(ValueError('Bad column')))
+        type(mock_row).bad_column = property(
+            lambda self: (_ for _ in ()).throw(ValueError('Bad column'))
+        )
 
         mock_result_set = Mock()
         mock_result_set.column_names = mock_column_names