some improvements

saleweaver · Oct 3, 2021 · c0854c5 · c0854c5
1 parent ef244f5
commit c0854c5
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 6 deletions.
diff --git a/setup.py b/setup.py
@@ -1,8 +1,9 @@
 from setuptools import setup
+from sp_api.base import __version__
 
 setup(
     name='python-amazon-sp-api',
-    version='0.6.4',
+    version=__version__,
     install_requires=[
         "requests",
         "six>=1.15,<2",

diff --git a/sp_api/base/__init__.py b/sp_api/base/__init__.py
@@ -22,6 +22,8 @@
 from .reportTypes import ReportType
 from sp_api.auth import AccessTokenClient
 
+__version__ = '0.6.5'
+
 __all__ = [
     'AccessTokenClient',
     'ReportType',
@@ -51,6 +53,6 @@
     'encrypt_aes',
     'NotificationType',
     'CredentialProvider',
-    'MissingCredentials'
-
+    'MissingCredentials',
+    '__version__'
 ]
diff --git a/sp_api/base/base_client.py b/sp_api/base/base_client.py
@@ -9,9 +9,8 @@ class BaseClient:
 
     def __init__(self, account='default', credentials=None):
         try:
-            import pkg_resources
-            version = pkg_resources.require("python-amazon-sp-api")[0].version
-            self.user_agent += f'-{version}'
+            from sp_api.base import __version__
+            self.user_agent += f'-{__version__}'
         except:
             pass
         self.credentials = CredentialProvider(account, credentials).credentials