Move version to __init__,py

owq-1777 · Mar 21, 2023 · 255f803 · 255f803
1 parent dcf311b
commit 255f803
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 4 deletions.
diff --git a/setup.py b/setup.py
@@ -1,5 +1,11 @@
+import importlib
+
 from setuptools import setup
 
+spec = importlib.util.spec_from_file_location("init", "tronpy/__init__.py")
+init_module = importlib.util.module_from_spec(spec)
+spec.loader.exec_module(init_module)
+
 packages = ["tronpy", "tronpy.keys", "tronpy.providers"]
 
 package_data = {"": ["*"]}
@@ -15,7 +21,7 @@
 
 setup_kwargs = {
     "name": "tronpy",
-    "version": "0.3.0",
+    "version": init_module.VERSION,
     "description": "TRON Python client library",
     "long_description": open("README.md").read(),
     "long_description_content_type": "text/markdown",

diff --git a/tronpy/__init__.py b/tronpy/__init__.py
@@ -5,5 +5,6 @@
 
 TRX = 1_000_000
 SUN = 1
+VERSION = "0.3.0"
 
-__all__ = ["Tron", "AsyncTron", "Contract", "AsyncContract", "ShieldedTRC20", "TRX", "SUN"]
+__all__ = ["Tron", "AsyncTron", "Contract", "AsyncContract", "ShieldedTRC20", "TRX", "SUN", "VERSION"]
diff --git a/tronpy/providers/async_http.py b/tronpy/providers/async_http.py
@@ -5,6 +5,8 @@
 import httpx
 from httpx import Timeout
 
+from tronpy import VERSION
+
 DEFAULT_TIMEOUT = 10.0
 DEFAULT_API_KEY = "f92221d5-7056-4366-b96f-65d3662ec2d9"
 
@@ -36,7 +38,7 @@ def __init__(
         else:
             raise TypeError(f"unknown endpoint uri {endpoint_uri}")
 
-        headers = {"User-Agent": "Tronpy/0.2", "Tron-Pro-Api-Key": api_key}
+        headers = {"User-Agent": f"Tronpy/{VERSION}", "Tron-Pro-Api-Key": api_key}
         if jw_token is not None:
             headers["Authorization"] = f"Bearer {jw_token}"
         if client is None:

diff --git a/tronpy/providers/http.py b/tronpy/providers/http.py
@@ -7,6 +7,8 @@
 
 import requests
 
+from tronpy import VERSION
+
 DEFAULT_TIMEOUT = 10.0
 DEFAULT_API_KEYS = [
     "f92221d5-7056-4366-b96f-65d3662ec2d9",
@@ -61,7 +63,7 @@ def __init__(
         self.jw_token = jw_token
 
         self.sess = requests.session()
-        self.sess.headers["User-Agent"] = "Tronpy/0.2"
+        self.sess.headers["User-Agent"] = f"Tronpy/{VERSION}"
 
         self.timeout = timeout
         """Request timeout in second."""