[bugfix]Adapt the method of writing objects to the database.

Author: Potterluo

test/common/capture_utils.py

@@ -1,3 +1,6 @@
+import dataclasses
+import functools
+from collections.abc import Mapping
 from typing import Any, Dict, List
 
 from common.db_utils import write_to_db
@@ -42,14 +45,65 @@ def post_process(table_name: str, **kwargs) -> List[Dict[str, Any]]:
     return []
 
 
+def _ensure_list(obj):
+    """
+    Ensure the object is returned as a list.
+    """
+    if isinstance(obj, list):
+        return obj
+    if isinstance(obj, (str, bytes, Mapping)):
+        return [obj]
+    if hasattr(obj, "__iter__") and not hasattr(obj, "__len__"):  # 如 generator
+        return list(obj)
+    return [obj]
+
+
+def _to_dict(obj: Any) -> Dict[str, Any]:
+    """
+    Convert various object types to a dictionary for DB writing.
+    """
+    if isinstance(obj, Mapping):
+        return dict(obj)
+    if dataclasses.is_dataclass(obj):
+        return dataclasses.asdict(obj)
+    if hasattr(obj, "_asdict"):  # namedtuple
+        return obj._asdict()
+    if hasattr(obj, "__dict__"):
+        return vars(obj)
+    raise TypeError(f"Cannot convert {type(obj)} to dict for DB writing")
+
+
+def proj_process(table_name: str, **kwargs) -> List[Dict[str, Any]]:
+    if "_proj" not in kwargs:
+        return []
+
+    name = kwargs.get("_name", table_name)
+    raw_input = kwargs["_proj"]
+    raw_results = _ensure_list(raw_input)
+
+    processed_results = []
+    for result in raw_results:
+        try:
+            dict_result = _to_dict(result)
+            write_to_db(name, dict_result)
+            processed_results.append(dict_result)
+        except Exception as e:
+            raise ValueError(f"Failed to process item in _proj: {e}") from e
+
+    return processed_results
+
+
 # ---------------- decorator ----------------
 def export_vars(func):
+    @functools.wraps(func)
     def wrapper(*args, **kwargs):
         result = func(*args, **kwargs)
-        # If the function returns a dict containing '_data' or 'data', post-process it
+        # If the function returns a dict containing '_data' or '_proj', post-process it
         if isinstance(result, dict):
-            if "_data" in result or "data" in result:
+            if "_data" in result:
                 return post_process(func.__name__, **result)
+            if "_proj" in result:
+                return proj_process(func.__name__, **result)
         # Otherwise return unchanged
         return result
 
@@ -63,33 +117,6 @@ def capture():
     return {"name": "demo", "_data": {"accuracy": 0.1, "loss": 0.3}}
 
 
-@export_vars
-def capture_list():
-    """All lists via '_name' + '_data'"""
-    return {
-        "_name": "demo",
-        "_data": {
-            "accuracy": [0.1, 0.2, 0.3],
-            "loss": [0.1, 0.2, 0.3],
-        },
-    }
-
-
-@export_vars
-def capture_mix():
-    """Mixed single + lists via '_name' + '_data'"""
-    return {
-        "_name": "demo",
-        "_data": {
-            "length": 10086,  # single value
-            "accuracy": [0.1, 0.2, 0.3],  # list
-            "loss": [0.1, 0.2, 0.3],  # list
-        },
-    }
-
-
 # quick test
 if __name__ == "__main__":
     print("capture():      ", capture())
-    print("capture_list(): ", capture_list())
-    print("capture_mix():  ", capture_mix())

test/common/db.sql

@@ -0,0 +1,8 @@
+CREATE TABLE test_results (
+    id INT AUTO_INCREMENT PRIMARY KEY,
+    test_case VARCHAR(255) NOT NULL,
+    status VARCHAR(50) NOT NULL,
+    error TEXT,
+    test_build_id VARCHAR(100) NOT NULL,
+    created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
+);

test/config.yaml

@@ -9,7 +9,7 @@ reports:
 
 database:
   backup: "results/"
-  enabled: true
+  enabled: false
   host: "127.0.0.1"
   port: 3306
   name: "ucm_pytest"

test/suites/E2E/test_demo_performance.py

@@ -41,18 +41,65 @@ def test_divide_by_zero(self, calc):
 
 
 @pytest.mark.feature("capture")  # pytest must be the top
-@export_vars
-def test_capture_mix():
-    """Mixed single + lists via '_name' + '_data'"""
-    assert 1 == 1
-    return {
-        "_name": "demo",
-        "_data": {
-            "length": 10086,  # single value
-            "accuracy": [0.1, 0.2, 0.3],  # list
-            "loss": [0.1, 0.2, 0.3],  # list
-        },
-    }
+class TestCapture:
+    @export_vars
+    def test_capture_mix(self):
+        """Mixed single + lists via '_name' + '_data'"""
+        assert 1 == 1
+        return {
+            "_name": "capture_demo",
+            "_data": {
+                "length": 10086,  # single value
+                "accuracy": [0.1, 0.2, 0.3],  # list
+                "loss": [0.1, 0.2, 0.3],  # list
+            },
+        }
+
+    @export_vars
+    def test_capture_proj(self):
+        """Mixed single + lists via '_name' + '_proj'"""
+        return {
+            "_name": "capture_demo",
+            "_proj": {
+                {"length": 10086, "accuracy": 0.1, "loss": 0.1},
+                {"length": 10086, "accuracy": 0.2, "loss": 0.2},
+                {"length": 10086, "accuracy": 0.3, "loss": 0.3},
+            },
+        }
+
+    @export_vars
+    def test_capture_proj(self):
+        """Mixed single + lists via '_name' + '_proj'"""
+
+        class Result:
+            def __init__(self, length, accuracy, loss):
+                self.length = length
+                self.accuracy = accuracy
+                self.loss = loss
+
+        return {
+            "_name": "capture_demo",
+            "_proj": Result(10010, 0.1, 0.1),
+        }
+
+    @export_vars
+    def test_capture_list_proj(self):
+        """Mixed single + lists via '_name' + '_proj'"""
+
+        class Result:
+            def __init__(self, length, accuracy, loss):
+                self.length = length
+                self.accuracy = accuracy
+                self.loss = loss
+
+        return {
+            "_name": "capture_demo",
+            "_proj": [
+                Result(10086, 0.1, 0.1),
+                Result(10086, 0.2, 0.2),
+                Result(10086, 0.3, 0.3),
+            ],
+        }
 
 
 # ---------------- Read Config Example ----------------