ran formatter

Author: kenliao94

Dockerfile

@@ -25,7 +25,7 @@ RUN --mount=type=cache,target=/root/.cache/uv \
 FROM python:3.12-slim-bookworm
 
 WORKDIR /app
- 
+
 COPY --from=uv --chown=app:app /app/.venv /app/.venv
 
 # Place executables in the environment at the front of the path

README.md

@@ -38,12 +38,12 @@ https://smithery.ai/server/@kenliao94/mcp-server-rabbitmq
         "command": "uv",
         "args": [
             "--directory",
-            "/path/to/repo/mcp-server-rabbitmq", 
-            "run", 
+            "/path/to/repo/mcp-server-rabbitmq",
+            "run",
             "mcp-server-rabbitmq",
             "--rabbitmq-host",
             "<hostname ex. test.rabbit.com, localhost>",
-            "--port", 
+            "--port",
             "<port number ex. 5672>",
             "--username",
             "<rabbitmq username>",

glama.json

@@ -3,4 +3,4 @@
     "maintainers": [
       "kenliao94"
     ]
-}
+}

mcp_server_rabbitmq/admin.py

@@ -3,18 +3,19 @@
 
 import requests
 
+from mcp_server_rabbitmq.connection import validate_rabbitmq_name
+
 
 class RabbitMQAdmin:
     def __init__(self, host: str, port: int, username: str, password: str, use_tls: bool):
         self.protocol = "https" if use_tls else "http"
         self.base_url = f"{self.protocol}://{host}:{port}/api"
         self.auth = base64.b64encode(f"{username}:{password}".encode()).decode()
-        self.headers = {
-            "Authorization": f"Basic {self.auth}",
-            "Content-Type": "application/json"
-        }
+        self.headers = {"Authorization": f"Basic {self.auth}", "Content-Type": "application/json"}
 
-    def _make_request(self, method: str, endpoint: str, data: Optional[Dict] = None) -> requests.Response:
+    def _make_request(
+        self, method: str, endpoint: str, data: Optional[Dict] = None
+    ) -> requests.Response:
         url = f"{self.base_url}/{endpoint}"
         response = requests.request(method, url, headers=self.headers, json=data, verify=True)
         response.raise_for_status()
@@ -60,15 +61,19 @@ def delete_exchange(self, exchange: str, vhost: str = "/") -> None:
         vhost_encoded = requests.utils.quote(vhost, safe="")
         self._make_request("DELETE", f"exchanges/{vhost_encoded}/{exchange}")
 
-    def get_bindings(self, queue: Optional[str] = None, exchange: Optional[str] = None, vhost: str = "/") -> List[Dict]:
+    def get_bindings(
+        self, queue: Optional[str] = None, exchange: Optional[str] = None, vhost: str = "/"
+    ) -> List[Dict]:
         """Get bindings, optionally filtered by queue or exchange"""
         vhost_encoded = requests.utils.quote(vhost, safe="")
         if queue:
             validate_rabbitmq_name(queue, "Queue name")
             response = self._make_request("GET", f"queues/{vhost_encoded}/{queue}/bindings")
         elif exchange:
             validate_rabbitmq_name(exchange, "Exchange name")
-            response = self._make_request("GET", f"exchanges/{vhost_encoded}/{exchange}/bindings/source")
+            response = self._make_request(
+                "GET", f"exchanges/{vhost_encoded}/{exchange}/bindings/source"
+            )
         else:
             response = self._make_request("GET", f"bindings/{vhost_encoded}")
         return response.json()

mcp_server_rabbitmq/connection.py

@@ -8,7 +8,7 @@ def __init__(self, host: str, port: int, username: str, password: str, use_tls:
         self.protocol = "amqps" if use_tls else "amqp"
         self.url = f"{self.protocol}://{username}:{password}@{host}:{port}"
         self.parameters = pika.URLParameters(self.url)
-        
+
         if use_tls:
             ssl_context = ssl.SSLContext(ssl.PROTOCOL_TLSv1_2)
             ssl_context.set_ciphers("ECDHE+AESGCM:!ECDSA")
@@ -19,11 +19,14 @@ def get_channel(self) -> tuple[pika.BlockingConnection, pika.channel.Channel]:
         channel = connection.channel()
         return connection, channel
 
+
 def validate_rabbitmq_name(name: str, field_name: str) -> None:
     """Validate RabbitMQ queue/exchange names"""
     if not name or not name.strip():
         raise ValueError(f"{field_name} cannot be empty")
     if not all(c.isalnum() or c in "-_.:" for c in name):
-        raise ValueError(f"{field_name} can only contain letters, digits, hyphen, underscore, period, or colon")
+        raise ValueError(
+            f"{field_name} can only contain letters, digits, hyphen, underscore, period, or colon"
+        )
     if len(name) > 255:
         raise ValueError(f"{field_name} must be less than 255 characters")

mcp_server_rabbitmq/handlers.py

@@ -10,31 +10,41 @@ def handle_enqueue(rabbitmq: RabbitMQConnection, queue: str, message: str):
     channel.basic_publish(exchange="", routing_key=queue, body=message)
     connection.close()
 
+
 def handle_fanout(rabbitmq: RabbitMQConnection, exchange: str, message: str):
     connection, channel = rabbitmq.get_channel()
     channel.exchange_declare(exchange=exchange, exchange_type="fanout")
     channel.basic_publish(exchange=exchange, routing_key="", body=message)
     connection.close()
 
+
 def handle_list_queues(rabbitmq_admin: RabbitMQAdmin) -> List[str]:
     result = rabbitmq_admin.list_queues()
     return [queue["name"] for queue in result]
 
+
 def handle_list_exchanges(rabbitmq_admin: RabbitMQAdmin) -> List[str]:
     result = rabbitmq_admin.list_exchanges()
     return [exchange["name"] for exchange in result]
 
+
 def handle_get_queue_info(rabbitmq_admin: RabbitMQAdmin, queue: str, vhost: str = "/") -> dict:
     return rabbitmq_admin.get_queue_info(queue, vhost)
 
+
 def handle_delete_queue(rabbitmq_admin: RabbitMQAdmin, queue: str, vhost: str = "/") -> None:
     rabbitmq_admin.delete_queue(queue, vhost)
 
+
 def handle_purge_queue(rabbitmq_admin: RabbitMQAdmin, queue: str, vhost: str = "/") -> None:
     rabbitmq_admin.purge_queue(queue, vhost)
 
+
 def handle_delete_exchange(rabbitmq_admin: RabbitMQAdmin, exchange: str, vhost: str = "/") -> None:
     rabbitmq_admin.delete_exchange(exchange, vhost)
 
-def handle_get_exchange_info(rabbitmq_admin: RabbitMQAdmin, exchange: str, vhost: str = "/") -> dict:
+
+def handle_get_exchange_info(
+    rabbitmq_admin: RabbitMQAdmin, exchange: str, vhost: str = "/"
+) -> dict:
     return rabbitmq_admin.get_exchange_info(exchange, vhost)

mcp_server_rabbitmq/server.py

@@ -26,10 +26,7 @@ def __init__(self):
         self.logger.setLevel(logging.INFO)
         logging.basicConfig(
             format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
-            handlers=[
-                logging.FileHandler("server.log"),
-                logging.StreamHandler()
-            ]
+            handlers=[logging.FileHandler("server.log"), logging.StreamHandler()],
         )
 
         # Initialize FastMCP
@@ -57,8 +54,11 @@ def enqueue(queue: str, message: str) -> str:
             validate_rabbitmq_name(queue, "Queue name")
             try:
                 rabbitmq = RabbitMQConnection(
-                    self.rabbitmq_host, self.rabbitmq_port, self.rabbitmq_username, 
-                    self.rabbitmq_password, self.rabbitmq_use_tls
+                    self.rabbitmq_host,
+                    self.rabbitmq_port,
+                    self.rabbitmq_username,
+                    self.rabbitmq_password,
+                    self.rabbitmq_use_tls,
                 )
                 handle_enqueue(rabbitmq, queue, message)
                 return "Message successfully enqueued"
@@ -72,8 +72,11 @@ def fanout(exchange: str, message: str) -> str:
             validate_rabbitmq_name(exchange, "Exchange name")
             try:
                 rabbitmq = RabbitMQConnection(
-                    self.rabbitmq_host, self.rabbitmq_port, self.rabbitmq_username,
-                    self.rabbitmq_password, self.rabbitmq_use_tls
+                    self.rabbitmq_host,
+                    self.rabbitmq_port,
+                    self.rabbitmq_username,
+                    self.rabbitmq_password,
+                    self.rabbitmq_use_tls,
                 )
                 handle_fanout(rabbitmq, exchange, message)
                 return "Message successfully published to exchange"
@@ -86,8 +89,11 @@ def list_queues() -> str:
             """List all the queues in the broker."""
             try:
                 admin = RabbitMQAdmin(
-                    self.rabbitmq_host, self.rabbitmq_api_port, self.rabbitmq_username,
-                    self.rabbitmq_password, self.rabbitmq_use_tls
+                    self.rabbitmq_host,
+                    self.rabbitmq_api_port,
+                    self.rabbitmq_username,
+                    self.rabbitmq_password,
+                    self.rabbitmq_use_tls,
                 )
                 result = handle_list_queues(admin)
                 return str(result)
@@ -100,8 +106,11 @@ def list_exchanges() -> str:
             """List all the exchanges in the broker."""
             try:
                 admin = RabbitMQAdmin(
-                    self.rabbitmq_host, self.rabbitmq_api_port, self.rabbitmq_username,
-                    self.rabbitmq_password, self.rabbitmq_use_tls
+                    self.rabbitmq_host,
+                    self.rabbitmq_api_port,
+                    self.rabbitmq_username,
+                    self.rabbitmq_password,
+                    self.rabbitmq_use_tls,
                 )
                 result = handle_list_exchanges(admin)
                 return str(result)
@@ -114,8 +123,11 @@ def get_queue_info(queue: str, vhost: str = "/") -> str:
             """Get detailed information about a specific queue."""
             try:
                 admin = RabbitMQAdmin(
-                    self.rabbitmq_host, self.rabbitmq_api_port, self.rabbitmq_username,
-                    self.rabbitmq_password, self.rabbitmq_use_tls
+                    self.rabbitmq_host,
+                    self.rabbitmq_api_port,
+                    self.rabbitmq_username,
+                    self.rabbitmq_password,
+                    self.rabbitmq_use_tls,
                 )
                 validate_rabbitmq_name(queue, "Queue name")
                 result = handle_get_queue_info(admin, queue, vhost)
@@ -129,8 +141,11 @@ def delete_queue(queue: str, vhost: str = "/") -> str:
             """Delete a specific queue."""
             try:
                 admin = RabbitMQAdmin(
-                    self.rabbitmq_host, self.rabbitmq_api_port, self.rabbitmq_username,
-                    self.rabbitmq_password, self.rabbitmq_use_tls
+                    self.rabbitmq_host,
+                    self.rabbitmq_api_port,
+                    self.rabbitmq_username,
+                    self.rabbitmq_password,
+                    self.rabbitmq_use_tls,
                 )
                 validate_rabbitmq_name(queue, "Queue name")
                 handle_delete_queue(admin, queue, vhost)
@@ -144,8 +159,11 @@ def purge_queue(queue: str, vhost: str = "/") -> str:
             """Remove all messages from a specific queue."""
             try:
                 admin = RabbitMQAdmin(
-                    self.rabbitmq_host, self.rabbitmq_api_port, self.rabbitmq_username,
-                    self.rabbitmq_password, self.rabbitmq_use_tls
+                    self.rabbitmq_host,
+                    self.rabbitmq_api_port,
+                    self.rabbitmq_username,
+                    self.rabbitmq_password,
+                    self.rabbitmq_use_tls,
                 )
                 validate_rabbitmq_name(queue, "Queue name")
                 handle_purge_queue(admin, queue, vhost)
@@ -159,8 +177,11 @@ def delete_exchange(exchange: str, vhost: str = "/") -> str:
             """Delete a specific exchange."""
             try:
                 admin = RabbitMQAdmin(
-                    self.rabbitmq_host, self.rabbitmq_api_port, self.rabbitmq_username,
-                    self.rabbitmq_password, self.rabbitmq_use_tls
+                    self.rabbitmq_host,
+                    self.rabbitmq_api_port,
+                    self.rabbitmq_username,
+                    self.rabbitmq_password,
+                    self.rabbitmq_use_tls,
                 )
                 validate_rabbitmq_name(exchange, "Exchange name")
                 handle_delete_exchange(admin, exchange, vhost)
@@ -174,8 +195,11 @@ def get_exchange_info(exchange: str, vhost: str = "/") -> str:
             """Get detailed information about a specific exchange."""
             try:
                 admin = RabbitMQAdmin(
-                    self.rabbitmq_host, self.rabbitmq_api_port, self.rabbitmq_username,
-                    self.rabbitmq_password, self.rabbitmq_use_tls
+                    self.rabbitmq_host,
+                    self.rabbitmq_api_port,
+                    self.rabbitmq_username,
+                    self.rabbitmq_password,
+                    self.rabbitmq_use_tls,
                 )
                 validate_rabbitmq_name(exchange, "Exchange name")
                 result = handle_get_exchange_info(admin, exchange, vhost)
@@ -212,16 +236,22 @@ def main():
     parser.add_argument("--port", type=int, required=True, help="Port of the RabbitMQ host")
     parser.add_argument("--username", type=str, required=True, help="Username for the connection")
     parser.add_argument("--password", type=str, required=True, help="Password for the connection")
-    parser.add_argument("--use-tls", type=bool, default=False, help="Is the connection using TLS/SSL")
-    parser.add_argument("--api-port", type=int, default=15672, help="Port for the RabbitMQ management API")
+    parser.add_argument(
+        "--use-tls", type=bool, default=False, help="Is the connection using TLS/SSL"
+    )
+    parser.add_argument(
+        "--api-port", type=int, default=15672, help="Port for the RabbitMQ management API"
+    )
     parser.add_argument("--sse", action="store_true", help="Use SSE transport")
-    parser.add_argument("--server-port", type=int, default=8888, help="Port to run the MCP server on")
+    parser.add_argument(
+        "--server-port", type=int, default=8888, help="Port to run the MCP server on"
+    )
 
     args = parser.parse_args()
-    
+
     server = RabbitMQMCPServer()
     server.run(args)
 
 
 if __name__ == "__main__":
-    main()       
+    main()

pyproject.toml

@@ -67,7 +67,7 @@ ignore = ["E203", "E501"]
 known-first-party = ["mcp_server_rabbitmq"]
 
 [tool.ruff.format]
-quote-style = "single"
+quote-style = "double"
 indent-style = "space"
 line-ending = "auto"
 

tests/test_connection.py

@@ -14,10 +14,9 @@ class TestRabbitMQConnection:
     def test_connection_initialization(self, mock_connection):
         """Test that the connection is initialized correctly."""
         mock_connection.return_value = MagicMock()
-        
         connection = RabbitMQConnection("localhost", 5672, "guest", "guest", False)
-        
         assert connection is not None
+        _, _ = connection.get_channel()
         mock_connection.assert_called_once()
 
 
@@ -36,12 +35,12 @@ def test_validate_rabbitmq_name_invalid(self):
         """Test that invalid names fail validation."""
         with pytest.raises(ValueError):
             validate_rabbitmq_name("", "Test")
-        
+
         with pytest.raises(ValueError):
             validate_rabbitmq_name("invalid/name", "Test")
-        
+
         with pytest.raises(ValueError):
             validate_rabbitmq_name("invalid\\name", "Test")
-        
+
         with pytest.raises(ValueError):
             validate_rabbitmq_name("invalid*name", "Test")