build(setup.py): 超参数更新

zjykzj · zjykzj · commit cdfd8d067955 · 2020-08-15T11:26:56.000+08:00
diff --git a/setup.py b/setup.py
@@ -30,6 +30,10 @@
     "Operating System :: OS Independent",
     "License :: OSI Approved :: Apache Software License"
 ]
+SOURCE_FOLDER = 'python_setup'
+CONSOLE_SCRIPTS = 'print_hello = python_setup.tools.cli:main'
+
+
 # ---------------------- #
 
 
@@ -72,7 +76,7 @@ def run(self):
 
 
 def get_version():
-    init_py_path = os.path.join(os.path.abspath(os.path.dirname(__file__)), "python_setup", "__init__.py")
+    init_py_path = os.path.join(os.path.abspath(os.path.dirname(__file__)), SOURCE_FOLDER, "__init__.py")
     init_py = open(init_py_path, "r").readlines()
     version_line = [l.strip() for l in init_py if l.startswith("__version__")][0]
     version = version_line.split("=")[-1].strip().strip("'\"")
@@ -97,7 +101,7 @@ def get_version():
     python_requires=PYTHON_REQUIRES,
     entry_points={
         'console_scripts': [
-            'print_hello = python_setup.tools.cli:main'
+            CONSOLE_SCRIPTS
         ]
     },
     cmdclass={
