Skip to content

Commit

Permalink
Fix entry_points
Browse files Browse the repository at this point in the history
  • Loading branch information
amotl committed Nov 12, 2023
1 parent 9aa6960 commit 320ecec
Show file tree
Hide file tree
Showing 3 changed files with 18 additions and 12 deletions.
13 changes: 13 additions & 0 deletions influxio/util/compat.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
import sys
import typing

if typing.TYPE_CHECKING:
from importlib.metadata import EntryPoints


def entry_points(**params) -> "EntryPoints":
if sys.version_info <= (3, 8):
from importlib_metadata import entry_points
else:
from importlib.metadata import entry_points
return entry_points(**params)
12 changes: 3 additions & 9 deletions influxio/util/db.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,17 +4,11 @@ def get_sqlalchemy_dialects():
TODO: Synchronize with influxio.util.report.
"""
from importlib.metadata import entry_points

import sqlalchemy.dialects

dialects = list(sqlalchemy.dialects.__all__)
try:
eps = entry_points(group="sqlalchemy.dialects")
except TypeError as ex:
if "unexpected keyword argument" in str(ex):
from importlib_metadata import entry_points
from influxio.util.compat import entry_points

eps = entry_points(group="sqlalchemy.dialects")
dialects = list(sqlalchemy.dialects.__all__)
eps = entry_points(group="sqlalchemy.dialects")
dialects += [dialect.name for dialect in eps]
return sorted(set(dialects))
5 changes: 2 additions & 3 deletions influxio/util/report.py
Original file line number Diff line number Diff line change
Expand Up @@ -77,11 +77,10 @@ def platform():
print()

# SQLAlchemy
# TODO: Synchronize with influxio.util.db.
from importlib.metadata import entry_points

import sqlalchemy.dialects

from influxio.util.compat import entry_points

subsection("SQLAlchemy")
print(bullet_item(list(sqlalchemy.dialects.__all__), label="Dialects built-in"))
eps = entry_points(group="sqlalchemy.dialects")
Expand Down

0 comments on commit 320ecec

Please sign in to comment.