From 60039c0af3b72d669a170147c9c6732f2a84b1e2 Mon Sep 17 00:00:00 2001
From: Petr <petr@keboola.com>
Date: Sun, 15 Mar 2026 02:03:27 +0100
Subject: [PATCH] Add direct catalog URL to YAML header (metric/table entity
 links)

Source line now links directly to the entity in OpenMetadata:
- metrics: https://datacatalog.../metric/UniqueVisitors
- tables:  https://datacatalog.../table/bigquery.project.dataset.table
---
 src/catalog_export.py | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/src/catalog_export.py b/src/catalog_export.py
index 30a21e3..cb457c0 100644
--- a/src/catalog_export.py
+++ b/src/catalog_export.py
@@ -68,21 +68,26 @@ def _get_docs_dir() -> Path:
     return docs_dir
 
 
-def _yaml_header(catalog_url: str, fqn: str = "") -> str:
+def _yaml_header(catalog_url: str, fqn: str = "", entity_type: str = "") -> str:
     """
     Generate AUTO-GENERATED header comment for YAML files.
 
     Args:
         catalog_url: OpenMetadata instance URL
         fqn: Optional entity FQN for reference
+        entity_type: Entity type for catalog URL (e.g., "metric", "table")
 
     Returns:
         Multi-line comment string
     """
     timestamp = datetime.now(timezone.utc).strftime("%Y-%m-%dT%H:%M:%SZ")
+    if fqn and entity_type:
+        source_url = f"{catalog_url}/{entity_type}/{fqn}"
+    else:
+        source_url = catalog_url
     lines = [
         AUTO_GENERATED_MARKER,
-        CATALOG_SOURCE_LINE.format(url=catalog_url),
+        CATALOG_SOURCE_LINE.format(url=source_url),
         GENERATED_LINE.format(timestamp=timestamp),
         DO_NOT_EDIT_LINE,
     ]
@@ -166,7 +171,7 @@ def export_metrics(
 
             # Write individual metric YAML
             file_path = cat_dir / f"{metric_name}.yml"
-            header = _yaml_header(catalog_url, fqn)
+            header = _yaml_header(catalog_url, fqn, entity_type="metric")
             # Wrap in list to match MetricParser expected format
             yaml_content = yaml.dump(
                 [yaml_dict],
@@ -283,7 +288,7 @@ def export_tables(
 
             # Write table YAML
             file_path = tables_dir / f"{table_config.name}.yml"
-            header = _yaml_header(catalog_url, fqn)
+            header = _yaml_header(catalog_url, fqn, entity_type="table")
             yaml_content = yaml.dump(
                 yaml_dict,
                 default_flow_style=False,