diff --git a/csvkit/utilities/csvsql.py b/csvkit/utilities/csvsql.py index 55bc32ee..9ace9b97 100644 --- a/csvkit/utilities/csvsql.py +++ b/csvkit/utilities/csvsql.py @@ -8,6 +8,13 @@ from csvkit.cli import CSVKitUtility, isatty from sqlalchemy import create_engine, dialects +try: + import importlib_metadata +except ImportError: + import importlib.metadata as importlib_metadata + +DIALECTS = dialects.__all__ + tuple(e.name for e in importlib_metadata.entry_points(group='sqlalchemy.dialects')) + def parse_list(pairs): options = {} @@ -20,14 +27,6 @@ def parse_list(pairs): return options -try: - import importlib_metadata -except ImportError: - import importlib.metadata as importlib_metadata - -DIALECTS = dialects.__all__ + tuple(e.name for e in importlib_metadata.entry_points(group='sqlalchemy.dialects')) - - class CSVSQL(CSVKitUtility): description = 'Generate SQL statements for one or more CSV files, or execute those statements directly on a ' \ 'database, and execute one or more SQL queries.'