mirror of
https://github.com/ARM-software/workload-automation.git
synced 2025-01-18 12:06:08 +00:00
103 lines
2.9 KiB
Plaintext
103 lines
2.9 KiB
Plaintext
import os
|
|
import sys
|
|
import warnings
|
|
from multiprocessing import Process
|
|
|
|
try:
|
|
from setuptools.command.install import install as orig_install
|
|
from setuptools import setup
|
|
except ImportError:
|
|
from distutils.command.install import install as orig_install
|
|
from distutils.core import setup
|
|
|
|
try:
|
|
import pwd
|
|
except ImportError:
|
|
pwd = None
|
|
|
|
warnings.filterwarnings('ignore', "Unknown distribution option: 'install_requires'")
|
|
|
|
try:
|
|
os.remove('MANIFEST')
|
|
except OSError:
|
|
pass
|
|
|
|
|
|
packages = []
|
|
data_files = {}
|
|
source_dir = os.path.dirname(__file__)
|
|
for root, dirs, files in os.walk('$package_name'):
|
|
rel_dir = os.path.relpath(root, source_dir)
|
|
data = []
|
|
if '__init__.py' in files:
|
|
for f in files:
|
|
if os.path.splitext(f)[1] not in ['.py', '.pyc', '.pyo']:
|
|
data.append(f)
|
|
package_name = rel_dir.replace(os.sep, '.')
|
|
package_dir = root
|
|
packages.append(package_name)
|
|
data_files[package_name] = data
|
|
else:
|
|
# use previous package name
|
|
filepaths = [os.path.join(root, f) for f in files]
|
|
data_files[package_name].extend([os.path.relpath(f, package_dir) for f in filepaths])
|
|
|
|
params = dict(
|
|
name='$package_name',
|
|
version='0.0.1',
|
|
packages=packages,
|
|
package_data=data_files,
|
|
url='N/A',
|
|
maintainer='$user',
|
|
maintainer_email='$user@example.com',
|
|
install_requires=[
|
|
'wa',
|
|
],
|
|
# https://pypi.python.org/pypi?%3Aaction=list_classifiers
|
|
classifiers=[
|
|
'Development Status :: 3 - Alpha',
|
|
'Environment :: Console',
|
|
'License :: Other/Proprietary License',
|
|
'Operating System :: Unix',
|
|
'Programming Language :: Python :: 3',
|
|
],
|
|
)
|
|
|
|
|
|
def update_wa_packages():
|
|
sudo_user = os.getenv('SUDO_USER')
|
|
if sudo_user:
|
|
user_entry = pwd.getpwnam(sudo_user)
|
|
os.setgid(user_entry.pw_gid)
|
|
os.setuid(user_entry.pw_uid)
|
|
env_root = os.getenv('WA_USER_DIRECTORY', os.path.join(os.path.expanduser('~'), '.workload_automation'))
|
|
if not os.path.isdir(env_root):
|
|
os.makedirs(env_root)
|
|
wa_packages_file = os.path.join(env_root, 'packages')
|
|
if os.path.isfile(wa_packages_file):
|
|
with open(wa_packages_file, 'r') as wfh:
|
|
package_list = wfh.read().split()
|
|
if params['name'] not in package_list:
|
|
package_list.append(params['name'])
|
|
else: # no existing package file
|
|
package_list = [params['name']]
|
|
with open(wa_packages_file, 'w') as wfh:
|
|
wfh.write('\n'.join(package_list))
|
|
|
|
|
|
class install(orig_install):
|
|
|
|
def run(self):
|
|
orig_install.run(self)
|
|
# Must be done in a separate process because will drop privileges if
|
|
# sudo, and won't be able to reacquire them.
|
|
p = Process(target=update_wa_packages)
|
|
p.start()
|
|
p.join()
|
|
|
|
|
|
params['cmdclass'] = {'install': install}
|
|
|
|
|
|
setup(**params)
|