Add modify port support

Author: wenfeng.wf

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "alibabacloud-rds-openapi-mcp-server"
-version = "2.0.4"
+version = "2.0.5"
 description = "MCP server for RDS Services via OPENAPI."
 readme = "README.md"
 requires-python = ">=3.12"

src/alibabacloud_rds_openapi_mcp_server/server.py

@@ -24,11 +24,12 @@
                    json_array_to_csv,
                    get_rds_client,
                    get_vpc_client,
-                   get_bill_client, get_das_client, convert_datetime_to_timestamp)
+                   get_bill_client, get_das_client, convert_datetime_to_timestamp,
+                   parse_args)
 
 logger = logging.getLogger(__name__)
 
-mcp = FastMCP("Alibaba Cloud RDS OPENAPI")
+mcp = FastMCP("Alibaba Cloud RDS OPENAPI", **(parse_args(sys.argv)))
 
 
 class OpenAPIError(Exception):
@@ -1483,7 +1484,7 @@ async def show_create_table(
 
 
 def main():
-    mcp.run(transport=os.getenv('SERVER_TRANSPORT', 'sse'))
+    mcp.run(transport=os.getenv('SERVER_TRANSPORT', 'stdio'))
 
 
 if __name__ == '__main__':

src/alibabacloud_rds_openapi_mcp_server/utils.py

@@ -49,6 +49,22 @@
 }
 
 
+def parse_args(argv):
+    args = {}
+    i = 1
+    while i < len(argv):
+        arg = argv[i]
+        if arg.startswith('--'):
+            key = arg[2:]
+            if i + 1 < len(argv) and not argv[i + 1].startswith('--'):
+                args[key] = argv[i+1]
+                i += 2
+            else:
+                args[key] = True
+                i += 1
+    return args
+
+
 def transform_to_iso_8601(dt: datetime, timespec: str):
     return dt.astimezone(timezone.utc).isoformat(timespec=timespec).replace("+00:00", "Z")