yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-07-24 17:59:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

MINOR: athena usage & lineage minor fix (#16156)

Browse Source 
* MINOR: athena usage & lineage minor fix

* py format
This commit is contained in:
Mayur Singal 2024-05-07 22:29:39 +05:30 committed by GitHub
parent 5cae2137d3
commit cf9b1a55ec
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
5 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ingestion/src/metadata/ingestion/source/database/athena/client.py
View File

@ -74,7 +74,7 @@ class AthenaLakeFormationClient:
return LFTags(**response)
except Exception as exc:
logger.debug(traceback.format_exc())
logger.warning(
logger.debug(
f"Unable to get lf tags for table resource [{table_name}] due to: {exc}"
)
return LFTags()

2
ingestion/src/metadata/ingestion/source/database/athena/lineage.py
View File

@ -15,6 +15,7 @@ from typing import Iterable, Optional
from metadata.generated.schema.type.tableQuery import TableQuery
from metadata.ingestion.source.database.athena.query_parser import (
QUERY_SUCCESS_STATUS,
AthenaQueryParserSource,
)
from metadata.ingestion.source.database.lineage_source import LineageSource
@ -37,6 +38,7 @@ class AthenaLineageSource(AthenaQueryParserSource, LineageSource):
if (
query.Status.SubmissionDateTime.date() >= self.start.date()
and self.is_not_dbt_or_om_query(query.Query)
and query.Status.State.upper() == QUERY_SUCCESS_STATUS
):
yield TableQuery(
query=query.Query,

2
ingestion/src/metadata/ingestion/source/database/athena/models.py
View File

@ -23,7 +23,7 @@ class QueryExecutionIdsResponse(BaseModel):
class Status(BaseModel):
State: Optional[str]
State: Optional[str] = "FAILED" # Default value
SubmissionDateTime: Optional[datetime]

2
ingestion/src/metadata/ingestion/source/database/athena/query_parser.py
View File

@ -41,6 +41,8 @@ ATHENA_QUERY_PAGINATOR_LIMIT = 50
ATHENA_ENABLED_WORK_GROUP_STATE = "ENABLED"
QUERY_SUCCESS_STATUS = "SUCCEEDED"
class AthenaQueryParserSource(QueryParserSource, ABC):
"""

2
ingestion/src/metadata/ingestion/source/database/athena/usage.py
View File

@ -15,6 +15,7 @@ from typing import Iterable
from metadata.generated.schema.type.tableQuery import TableQueries, TableQuery
from metadata.ingestion.source.database.athena.query_parser import (
QUERY_SUCCESS_STATUS,
AthenaQueryParserSource,
)
from metadata.ingestion.source.database.usage_source import UsageSource
@ -56,6 +57,7 @@ class AthenaUsageSource(AthenaQueryParserSource, UsageSource):
for query in query_list.QueryExecutions
if query.Status
and query.Query
and query.Status.State.upper() == QUERY_SUCCESS_STATUS
and query.Status.SubmissionDateTime.date() >= self.start.date()
and self.is_not_dbt_or_om_query(query.Query)
]