test:delete unusual test file (#299)

1.delete usual test file
2.format code

Author: csunny

pilot/common/sql_database.py

@@ -268,6 +268,31 @@ def __query(self, session, query, fetch: str = "all"):
             result.insert(0, field_names)
             return result
 
+    def query_ex(self, session, query, fetch: str = "all"):
+        """
+        only for query
+        Args:
+            session:
+            query:
+            fetch:
+        Returns:
+        """
+        print(f"Query[{query}]")
+        if not query:
+            return []
+        cursor = session.execute(text(query))
+        if cursor.returns_rows:
+            if fetch == "all":
+                result = cursor.fetchall()
+            elif fetch == "one":
+                result = cursor.fetchone()[0]  # type: ignore
+            else:
+                raise ValueError("Fetch parameter must be either 'one' or 'all'")
+            field_names = list(i[0:] for i in cursor.keys())
+
+            result = list(result)
+            return field_names, result
+
     def run(self, session, command: str, fetch: str = "all") -> List:
         """Execute a SQL command and return a string representing the results."""
         print("SQL:" + command)

pilot/connections/rdbms/py_study/__init__.py

@@ -2,6 +2,8 @@
 import os
 import uuid
 from typing import Dict, NamedTuple, List
+from decimal import Decimal
+
 from pilot.scene.base_message import (
     HumanMessage,
     ViewMessage,
@@ -17,6 +19,7 @@
 from pilot.scene.chat_dashboard.data_preparation.report_schma import (
     ChartData,
     ReportData,
+    ValueItem,
 )
 
 CFG = Config()
@@ -74,16 +77,49 @@ def do_action(self, prompt_response):
         chart_datas: List[ChartData] = []
         for chart_item in prompt_response:
             try:
-                datas = self.database.run(self.db_connect, chart_item.sql)
+                field_names, datas = self.database.query_ex(
+                    self.db_connect, chart_item.sql
+                )
+                values: List[ValueItem] = []
+                data_map = {}
+                field_map = {}
+                index = 0
+                for field_name in field_names:
+                    data_map.update({f"{field_name}": [row[index] for row in datas]})
+                    index += 1
+                    if not data_map[field_name]:
+                        field_map.update({f"{field_name}": False})
+                    else:
+                        field_map.update(
+                            {
+                                f"{field_name}": all(
+                                    isinstance(item, (int, float, Decimal))
+                                    for item in data_map[field_name]
+                                )
+                            }
+                        )
+
+                for field_name in field_names[1:]:
+                    if not field_map[field_name]:
+                        print("more than 2 non-numeric column")
+                    else:
+                        for data in datas:
+                            value_item = ValueItem(
+                                name=data[0],
+                                type=field_name,
+                                value=data[field_names.index(field_name)],
+                            )
+                            values.append(value_item)
+
                 chart_datas.append(
                     ChartData(
                         chart_uid=str(uuid.uuid1()),
                         chart_name=chart_item.title,
                         chart_type=chart_item.showcase,
                         chart_desc=chart_item.thoughts,
                         chart_sql=chart_item.sql,
-                        column_name=datas[0],
-                        values=datas,
+                        column_name=field_names,
+                        values=values,
                     )
                 )
             except Exception as e: