refactor: pkg_resources -> importlib.metadata (#24514)

This commit is contained in:
cwegener
2023-06-28 22:25:50 +00:00
committed by GitHub
parent 1e7ee8c4ad
commit f1b003f109
4 changed files with 7 additions and 5 deletions

View File

@@ -37,7 +37,7 @@ from typing import Any, Optional
import sqlalchemy.databases
import sqlalchemy.dialects
from pkg_resources import iter_entry_points
from importlib_metadata import entry_points
from sqlalchemy.engine.default import DefaultDialect
from sqlalchemy.engine.url import URL
@@ -74,7 +74,7 @@ def load_engine_specs() -> list[type[BaseEngineSpec]]:
if is_engine_spec(getattr(module, attr))
)
# load additional engines from external modules
for ep in iter_entry_points("superset.db_engine_specs"):
for ep in entry_points(group="superset.db_engine_specs"):
try:
engine_spec = ep.load()
except Exception: # pylint: disable=broad-except
@@ -150,7 +150,7 @@ def get_available_engine_specs() -> dict[type[BaseEngineSpec], set[str]]:
drivers[attr].add(attribute.dialect.driver)
# installed 3rd-party dialects
for ep in iter_entry_points("sqlalchemy.dialects"):
for ep in entry_points(group="sqlalchemy.dialects"):
try:
dialect = ep.load()
except Exception as ex: # pylint: disable=broad-except