yujunjun/datahub
2
0
Fork 0
mirror of https://github.com/datahub-project/datahub.git synced 2025-12-04 06:33:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(ingest/okta): fix event_loop RuntimeError with nested asyncio (#8637)

Browse Source
This commit is contained in:
skrydal 2023-08-16 07:02:57 +02:00 committed by GitHub
parent 526e626146
commit baae3d261d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
metadata-ingestion/setup.py
View File

@ -358,7 +358,7 @@ plugins: Dict[str, Set[str]] = {
"mysql": mysql,
# mariadb should have same dependency as mysql
"mariadb": sql_common | {"pymysql>=1.0.2"},
"okta": {"okta~=1.7.0"},
"okta": {"okta~=1.7.0", "nest-asyncio"},
"oracle": sql_common | {"cx_Oracle"},
"postgres": sql_common | {"psycopg2-binary", "GeoAlchemy2"},
"presto": sql_common | pyhive_common | trino,

2
metadata-ingestion/src/datahub/ingestion/source/identity/okta.py
View File

@ -7,6 +7,7 @@ from dataclasses import dataclass, field
from time import sleep
from typing import Dict, Iterable, List, Optional, Union
import nest_asyncio
from okta.client import Client as OktaClient
from okta.exceptions import OktaAPIException
from okta.models import Group, GroupProfile, User, UserProfile, UserStatus
@ -51,6 +52,7 @@ from datahub.metadata.schema_classes import (
)
logger = logging.getLogger(__name__)
nest_asyncio.apply()
class OktaConfig(StatefulIngestionConfigBase, ConfigModel):