From e0e9b42d2ad7393acf697bccb693a78c6732538a Mon Sep 17 00:00:00 2001 From: Matt <38562764+LavMatt@users.noreply.github.com> Date: Mon, 24 Jun 2024 14:13:43 +0100 Subject: [PATCH] fix parse_owners() for when owner display name exists with a value of None --- .../data_platform_catalogue/client/graphql_helpers.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/lib/datahub-client/data_platform_catalogue/client/graphql_helpers.py b/lib/datahub-client/data_platform_catalogue/client/graphql_helpers.py index 25768d02..b0a05f45 100644 --- a/lib/datahub-client/data_platform_catalogue/client/graphql_helpers.py +++ b/lib/datahub-client/data_platform_catalogue/client/graphql_helpers.py @@ -30,8 +30,13 @@ def parse_owner(entity: dict[str, Any]) -> OwnerRef: owners = [i["owner"] for i in ownership.get("owners", [])] if owners: properties = owners[0].get("properties") or {} + display_name = ( + properties.get("displayName") + if properties.get("displayName") is not None + else properties.get("fullName", "") + ) owner_details = OwnerRef( - display_name=properties.get("displayName", properties.get("fullName", "")), + display_name=display_name, email=properties.get("email", ""), urn=owners[0].get("urn", ""), )