chore: 코드 스타일 개선 및 불필요한 공백 제거

Author: ehddnr301

data_utils/datahub_services/__init__.py

@@ -6,7 +6,7 @@
 주요 구성요소:
 - DataHubBaseClient: 기본 연결 및 통신
 - MetadataService: 메타데이터, 리니지, URN 관련 기능
-- QueryService: 쿼리 관련 기능  
+- QueryService: 쿼리 관련 기능
 - GlossaryService: 용어집 관련 기능
 """
 
@@ -15,9 +15,4 @@
 from .query_service import QueryService
 from .glossary_service import GlossaryService
 
-__all__ = [
-    'DataHubBaseClient',
-    'MetadataService', 
-    'QueryService',
-    'GlossaryService'
-]
+__all__ = ["DataHubBaseClient", "MetadataService", "QueryService", "GlossaryService"]

data_utils/datahub_services/base_client.py

@@ -11,11 +11,11 @@
 
 class DataHubBaseClient:
     """DataHub 기본 클라이언트 클래스"""
-    
+
     def __init__(self, gms_server="http://localhost:8080", extra_headers={}):
         """
         DataHub 클라이언트 초기화
-        
+
         Args:
             gms_server (str): DataHub GMS 서버 URL
             extra_headers (dict): 추가 HTTP 헤더
@@ -26,7 +26,7 @@ def __init__(self, gms_server="http://localhost:8080", extra_headers={}):
 
         self.gms_server = gms_server
         self.extra_headers = extra_headers
-        
+
         # DataHub 클라이언트 초기화
         self.emitter = DatahubRestEmitter(
             gms_server=gms_server, extra_headers=extra_headers
@@ -36,10 +36,10 @@ def __init__(self, gms_server="http://localhost:8080", extra_headers={}):
     def _is_valid_gms_server(self, gms_server):
         """
         GMS 서버 주소의 유효성을 검사하는 함수
-        
+
         Args:
             gms_server (str): 검사할 GMS 서버 URL
-            
+
         Returns:
             bool: 서버가 유효한 경우 True
         """
@@ -57,26 +57,26 @@ def _is_valid_gms_server(self, gms_server):
     def execute_graphql_query(self, query, variables=None):
         """
         GraphQL 쿼리 실행
-        
+
         Args:
             query (str): GraphQL 쿼리 문자열
             variables (dict, optional): 쿼리 변수
-            
+
         Returns:
             dict: GraphQL 응답
         """
         headers = {"Content-Type": "application/json"}
         payload = {"query": query}
-        
+
         if variables:
             payload["variables"] = variables
-            
+
         response = requests.post(
             f"{self.gms_server}/api/graphql",
             json=payload,
             headers=headers,
         )
-        
+
         if response.status_code == 200:
             return response.json()
         else:
@@ -92,4 +92,4 @@ def get_datahub_graph(self):
 
     def get_urns(self):
         """필터를 적용하여 데이터셋의 URN 가져오기"""
-        return self.datahub_graph.get_urns_by_filter()
+        return self.datahub_graph.get_urns_by_filter()

data_utils/datahub_services/glossary_service.py

@@ -14,11 +14,11 @@
 
 class GlossaryService:
     """용어집 관련 서비스 클래스"""
-    
+
     def __init__(self, client: DataHubBaseClient):
         """
         용어집 서비스 초기화
-        
+
         Args:
             client (DataHubBaseClient): DataHub 기본 클라이언트
         """
@@ -191,4 +191,4 @@ def get_glossary_terms_by_urn(self, dataset_urn):
             dict: glossary terms 정보
         """
         variables = {"urn": dataset_urn}
-        return self.client.execute_graphql_query(GLOSSARY_TERMS_BY_URN_QUERY, variables)
+        return self.client.execute_graphql_query(GLOSSARY_TERMS_BY_URN_QUERY, variables)

data_utils/datahub_services/metadata_service.py

@@ -4,7 +4,11 @@
 테이블 메타데이터, 리니지, URN 관련 기능을 제공합니다.
 """
 
-from datahub.metadata.schema_classes import DatasetPropertiesClass, SchemaMetadataClass, UpstreamLineageClass
+from datahub.metadata.schema_classes import (
+    DatasetPropertiesClass,
+    SchemaMetadataClass,
+    UpstreamLineageClass,
+)
 from datahub.ingestion.graph.client import DatahubClientConfig, DataHubGraph
 from collections import defaultdict
 
@@ -13,11 +17,11 @@
 
 class MetadataService:
     """메타데이터 관련 서비스 클래스"""
-    
+
     def __init__(self, client: DataHubBaseClient):
         """
         메타데이터 서비스 초기화
-        
+
         Args:
             client (DataHubBaseClient): DataHub 기본 클라이언트
         """
@@ -309,4 +313,4 @@ def _print_urn_details(self, metadata):
                     print(
                         f"    {col['upstream_column']} → {col['downstream_column']} (신뢰도: {confidence})"
                     )
-            print()
+            print()

data_utils/datahub_services/query_service.py

@@ -13,11 +13,11 @@
 
 class QueryService:
     """쿼리 관련 서비스 클래스"""
-    
+
     def __init__(self, client: DataHubBaseClient):
         """
         쿼리 서비스 초기화
-        
+
         Args:
             client (DataHubBaseClient): DataHub 기본 클라이언트
         """
@@ -107,7 +107,7 @@ def get_query_data(self, start=0, count=10, query="*", filters=None):
     def get_queries_by_urn(self, dataset_urn):
         """
         특정 데이터셋 URN과 연관된 쿼리들을 조회하는 함수
-        
+
         전체 쿼리를 가져온 후 클라이언트 사이드에서 필터링하는 방식 사용
 
         Args:
@@ -117,31 +117,31 @@ def get_queries_by_urn(self, dataset_urn):
             dict: 연관된 쿼리 목록
         """
         # 먼저 전체 쿼리 목록을 가져옴
-        input_params = {
-            "start": 0,
-            "count": 1000,  # 충분히 큰 수로 설정
-            "query": "*"
-        }
+        input_params = {"start": 0, "count": 1000, "query": "*"}  # 충분히 큰 수로 설정
 
         variables = {"input": input_params}
         result = self.client.execute_graphql_query(QUERIES_BY_URN_QUERY, variables)
 
-        if "error" not in result and "data" in result and "listQueries" in result["data"]:
+        if (
+            "error" not in result
+            and "data" in result
+            and "listQueries" in result["data"]
+        ):
             # 클라이언트 사이드에서 특정 URN과 연관된 쿼리만 필터링
             all_queries = result["data"]["listQueries"]["queries"]
             filtered_queries = []
-            
+
             for query in all_queries:
                 subjects = query.get("subjects", [])
                 for subject in subjects:
                     if subject.get("dataset", {}).get("urn") == dataset_urn:
                         filtered_queries.append(query)
                         break
-            
+
             # 필터링된 결과로 응답 구조 재구성
             result["data"]["listQueries"]["queries"] = filtered_queries
             result["data"]["listQueries"]["count"] = len(filtered_queries)
-            
+
         return result
 
     def get_glossary_terms_by_urn(self, dataset_urn):
@@ -155,6 +155,6 @@ def get_glossary_terms_by_urn(self, dataset_urn):
             dict: glossary terms 정보
         """
         from data_utils.queries import GLOSSARY_TERMS_BY_URN_QUERY
-        
+
         variables = {"urn": dataset_urn}
-        return self.client.execute_graphql_query(GLOSSARY_TERMS_BY_URN_QUERY, variables)
+        return self.client.execute_graphql_query(GLOSSARY_TERMS_BY_URN_QUERY, variables)