|
7 | 7 | from datahub.metadata.schema_classes import EntityChangeEventClass as EntityChangeEvent |
8 | 8 | from datahub.utilities.urns.urn import Urn |
9 | 9 |
|
| 10 | +from datahub_actions.utils.datahub_util import make_entity_url |
10 | 11 | from datahub_actions.utils.name_resolver import ( |
11 | 12 | get_entity_name_from_urn, |
12 | 13 | get_entity_qualifier_from_urn, |
@@ -108,20 +109,20 @@ def get_message_from_entity_change_event( |
108 | 109 | parent_specialized_type = get_entity_qualifier_from_urn( |
109 | 110 | str(parent_entity_urn), datahub_graph |
110 | 111 | ) |
111 | | - parent_entity_url = f"{datahub_base_url}/{parent_entity_urn.get_type()}/{parent_entity_urn}/Schema?schemaFilter={entity_name}" |
| 112 | + parent_entity_url = f"{make_entity_url(str(parent_entity_urn), parent_entity_urn.get_type(), datahub_base_url)}/Schema?schemaFilter={entity_name}" |
112 | 113 | entity_message_trailer = f"{entity_name} of {parent_specialized_type} {make_url_with_title(title=parent_entity_name, url=parent_entity_url, channel=channel)}" |
113 | 114 | elif event.entityType == "dataFlow": |
114 | | - entity_url = f"{datahub_base_url}/pipelines/{event.entityUrn}" |
| 115 | + entity_url = make_entity_url(event.entityUrn, "pipelines", datahub_base_url) |
115 | 116 | entity_message_trailer = make_url_with_title( |
116 | 117 | title=entity_name, url=entity_url, channel=channel |
117 | 118 | ) |
118 | 119 | elif event.entityType == "dataJob": |
119 | | - entity_url = f"{datahub_base_url}/tasks/{event.entityUrn}" |
| 120 | + entity_url = make_entity_url(event.entityUrn, "tasks", datahub_base_url) |
120 | 121 | entity_message_trailer = make_url_with_title( |
121 | 122 | title=entity_name, url=entity_url, channel=channel |
122 | 123 | ) |
123 | 124 | else: |
124 | | - entity_url = f"{datahub_base_url}/{event.entityType}/{event.entityUrn}" |
| 125 | + entity_url = make_entity_url(event.entityUrn, event.entityType, datahub_base_url) |
125 | 126 | entity_message_trailer = make_url_with_title( |
126 | 127 | title=entity_name, url=entity_url, channel=channel |
127 | 128 | ) |
|
0 commit comments