Feat: avg query time + last used at timestamp (#281)

gouline · Oct 12, 2024 · 7904041 · 7904041
1 parent 78e8133
commit 7904041
Showing 1 changed file with 18 additions and 1 deletion.
diff --git a/dbtmetabase/_exposures.py b/dbtmetabase/_exposures.py
@@ -112,7 +112,9 @@ def extract_exposures(
                 depends = set()
                 native_query = ""
                 header = ""
-
+                average_query_time_ms = None
+                last_used_at_utc = None
+
                 entity: Mapping
                 if item["model"] == "card":
                     card_entity = self.metabase.find_card(uid=item["id"])
@@ -128,6 +130,8 @@ def extract_exposures(
                     result = self.__extract_card_exposures(ctx, card=entity)
                     depends.update(result["depends"])
                     native_query = result["native_query"]
+                    average_query_time_ms = entity.get('average_query_time')
+                    last_used_at_utc = entity['last_used_at']
 
                 elif item["model"] == "dashboard":
                     dashboard_entity = self.metabase.find_dashboard(uid=item["id"])
@@ -190,6 +194,8 @@ def extract_exposures(
                             created_at=entity["created_at"],
                             creator_name=creator_name or "",
                             creator_email=creator_email or "",
+                            last_used_at_utc=last_used_at_utc,
+                            average_query_time_ms=average_query_time_ms,
                             native_query=native_query,
                             depends_on=sorted(
                                 [
@@ -310,6 +316,8 @@ def __format_exposure(
         created_at: str,
         creator_name: str,
         creator_email: str,
+        last_used_at_utc: Optional[str],
+        average_query_time_ms: Optional[float],
         native_query: Optional[str],
         depends_on: Iterable[str],
         tags: Optional[Sequence[str]],
@@ -346,6 +354,11 @@ def __format_exposure(
             + f"Created On: __{created_at}__"
         )
 
+        if average_query_time_ms:
+            average_query_time_s = average_query_time_ms // 1000
+            average_query_time = f"{int(average_query_time_s // 60)}:{str(int(average_query_time_s % 60)).zfill(2)}"
+        else:
+            average_query_time = "n.a."
         exposure = {
             "name": name,
             "label": label,
@@ -360,6 +373,10 @@ def __format_exposure(
                 "email": creator_email,
             },
             "depends_on": list(depends_on),
+            "meta": {
+                "avg_query_time": average_query_time,
+                "last_used_at_utc": last_used_at_utc or 'n.a.'
+            }
         }
 
         if tags: