refactor: clean up lineage-related functions and improve URN parsing safety

Author: seyoung4503

data_utils/datahub_source.py

@@ -2,6 +2,9 @@
 from datahub.emitter.mcp import MetadataChangeProposalWrapper
 from datahub.metadata.schema_classes import DatasetPropertiesClass, SchemaMetadataClass
 from datahub.emitter.rest_emitter import DatahubRestEmitter
+from datahub.ingestion.graph.client import DatahubClientConfig, DataHubGraph
+from datahub.metadata.schema_classes import UpstreamLineageClass
+from collections import defaultdict
 import requests
 
 
@@ -94,8 +97,6 @@ def get_table_lineage(
         if degree_values is None:
             degree_values = ["1", "2"]
 
-        from datahub.ingestion.graph.client import DatahubClientConfig, DataHubGraph
-
         graph = DataHubGraph(DatahubClientConfig(server=self.gms_server))
 
         query = """
@@ -153,10 +154,6 @@ def get_column_lineage(self, urn):
             }
         """
 
-        from datahub.ingestion.graph.client import DatahubClientConfig, DataHubGraph
-        from datahub.metadata.schema_classes import UpstreamLineageClass
-        from collections import defaultdict
-
         # DataHub 연결 및 lineage 가져오기
         graph = DataHubGraph(DatahubClientConfig(server=self.gms_server))
         result = graph.get_aspect(entity_urn=urn, aspect_type=UpstreamLineageClass)