Merge pull request #136 from treasure-data/no-pkg-resources

Use importlib.metadata instead of pkg_resources since deprecated

pytd/__init__.py

@@ -1,10 +1,9 @@
+import importlib.metadata
 import logging
 
-import pkg_resources
-
 from .client import Client
 
-__version__ = pkg_resources.get_distribution("pytd").version
+__version__ = importlib.metadata.version("pytd")
 logger = logging.getLogger(__name__)
 logger.addHandler(logging.StreamHandler())
 logger.setLevel(logging.INFO)

pytd/query_engine.py

@@ -1,13 +1,13 @@
 import abc
+import importlib.metadata
 import logging
 import os
 from urllib.parse import urlparse
 
-import pkg_resources
 import prestodb
 import tdclient
 
-__version__ = pkg_resources.get_distribution("pytd").version
+__version__ = importlib.metadata.version("pytd")
 
 logger = logging.getLogger(__name__)
 
@@ -95,7 +95,8 @@ def execute(self, query, **kwargs):
         cur = self.cursor(**kwargs)
         self.executed = cur.execute(query)
         rows = cur.fetchall()
-        # cur.description is None for CREATE and DROP statements in recent version of Trino
+        # cur.description is None for CREATE and DROP statements in recent version of
+        # Trino
         columns = [desc[0] for desc in cur.description] if cur.description else None
         return {"data": rows, "columns": columns}