Json array to cvs.

Author: wenfeng.wf

src/alibabacloud_rds_openapi_mcp_server/server.py

@@ -2,6 +2,7 @@
 import logging
 import os
 import sys
+import csv
 from datetime import datetime
 from typing import Dict, Any, List
 
@@ -15,7 +16,7 @@
 from utils import (transform_to_iso_8601,
                    transform_to_datetime,
                    transform_perf_key,
-                   compress_json_array,
+                   json_array_to_csv,
                    get_rds_client,
                    get_vpc_client,
                    get_bill_client)
@@ -40,9 +41,12 @@ async def describe_db_instances(region_id: str):
     """
     client = get_rds_client(region_id)
     try:
-        request = rds_20140815_models.DescribeDBInstancesRequest(region_id=region_id)
+        request = rds_20140815_models.DescribeDBInstancesRequest(
+            region_id=region_id,
+            page_size=100
+        )
         response = client.describe_dbinstances(request)
-        return response.body.to_map()
+        return json_array_to_csv(response.body.items.dbinstance)
     except Exception as e:
         raise e
 
@@ -113,7 +117,7 @@ def _compress_performance(performance_value, max_items=10):
         response = client.describe_dbinstance_performance(request)
         responses = []
         for perf_key in response.body.performance_keys.performance_key:
-            perf_key_info = f"""Key={perf_key.key}; Unit={perf_key.unit}; ValueFormat={perf_key.value_format}; Values={compress_json_array([item.to_map() for item in _compress_performance(perf_key.values.performance_value)])}"""
+            perf_key_info = f"""Key={perf_key.key}; Unit={perf_key.unit}; ValueFormat={perf_key.value_format}; Values={json_array_to_csv(_compress_performance(perf_key.values.performance_value))}"""
             responses.append(perf_key_info)
         return responses
     except Exception as e:
@@ -905,12 +909,8 @@ async def describe_db_instance_parameters(
 
                 db_instance_parameters[db_instance_id] = {
                     "ParamGroupInfo": response.body.param_group_info.to_map(),
-                    "ConfigParameters": compress_json_array([
-                        config_parameter.to_map() for config_parameter in response.body.config_parameters.dbinstance_parameter
-                    ]),
-                    "RunningParameters": compress_json_array([
-                        running_parameter.to_map() for running_parameter in response.body.running_parameters.dbinstance_parameter
-                    ])
+                    "ConfigParameters": json_array_to_csv(response.body.config_parameters.dbinstance_parameter),
+                    "RunningParameters": json_array_to_csv(response.body.running_parameters.dbinstance_parameter)
                 }
         return db_instance_parameters
     except Exception as e:
@@ -978,7 +978,7 @@ async def describe_bills(
                             "BillingItemUsage": item.usage,
                         }
                     )
-            res[billing_cycle] = compress_json_array(item_filters)
+            res[billing_cycle] = json_array_to_csv(item_filters)
         return res
     except Exception as e:
         raise e

src/alibabacloud_rds_openapi_mcp_server/utils.py

@@ -1,5 +1,7 @@
+import csv
 import os
 from datetime import datetime, timezone
+from io import StringIO
 
 from alibabacloud_bssopenapi20171214.client import Client as BssOpenApi20171214Client
 from alibabacloud_rds20140815.client import Client as RdsClient
@@ -66,13 +68,31 @@ def transform_perf_key(db_type: str, perf_keys: list[str]):
     return perf_key_after_transform
 
 
-def compress_json_array(json_array: list[dict]):
-    if not json_array or len(json_array) == 0:
+def json_array_to_csv(data):
+    if not data or not isinstance(data, list):
         return ""
-    compress_str = ";".join(json_array[0].keys())
-    for item in json_array:
-        compress_str += "|" + ";".join([str(item[key] if key in item else "") for key in json_array[0].keys()])
-    return compress_str
+
+    fieldnames = set()
+    for item in data:
+        if isinstance(item, dict):
+            fieldnames.update(item.keys())
+        elif hasattr(item, 'to_map'):
+            fieldnames.update(item.to_map().keys())
+
+    if not fieldnames:
+        return ""
+
+    output = StringIO()
+    writer = csv.DictWriter(output, fieldnames=sorted(fieldnames))
+
+    writer.writeheader()
+    for item in data:
+        if isinstance(item, dict):
+            writer.writerow({k: v if v is not None else '' for k, v in item.items()})
+        elif hasattr(item, 'to_map'):
+            writer.writerow({k: v if v is not None else '' for k, v in item.to_map().items()})
+
+    return output.getvalue()
 
 
 def get_rds_client(region_id: str):