-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsetup.py
62 lines (54 loc) · 1.9 KB
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
from setuptools import setup, find_packages
from setuptools.command.install_scripts import install_scripts
from setuptools.command.install import install
from setuptools.command.develop import develop
from setuptools.command.egg_info import egg_info
from setuptools.command.build_ext import build_ext
import subprocess
import os
import glob
# From https://stackoverflow.com/questions/5932804/set-file-permissions-in-setup-py-file
# https://blog.niteo.co/setuptools-run-custom-code-in-setup-py/
def customize(command) :
command_name = str(command.mro()[1].__name__).strip()
original_run = command.run
def run(self) :
# Run the rest of the installer first
original_run(self)
# Create a new subprocess to run the included shell script
print("Running " + command_name + " commands...")
current_dir_path = os.path.dirname(os.path.realpath(__file__))
create_service_script_path = os.path.join(current_dir_path, 'setup.sh')
# stdout and stderr are combined in shell output
output=subprocess.run([create_service_script_path,command_name],stdout=subprocess.PIPE,stderr=subprocess.STDOUT).stdout
print(output.decode('UTF-8'))
command.run = run
return command
@customize
class CustomInstallCommand(install) :
pass
@customize
class CustomDevelopCommand(develop) :
pass
@customize
class CustomEggInfoCommand(egg_info) :
pass
@customize
class CustomBuildExtCommand(build_ext) :
pass
files = glob.glob('python-tool/plugins/*.py')
setup(name='python-tool',
version='1.0.0',
url='',
license='Apache2',
author='Daniel Dayley',
author_email='[email protected]',
description='',
packages=find_packages(exclude=['tests']),
package_data={"": ['plugins/*.py']},
install_requires=['pyyaml',],
scripts=['bin/python-tool'],
long_description=open('README.md').read(),
cmdclass={'install':CustomInstallCommand,'develop':CustomDevelopCommand,'egg_info':CustomEggInfoCommand,'build_ext':CustomBuildExtCommand},
zip_safe=True
)