diff options
author | 2016-11-11 02:15:30 +0000 | |
---|---|---|
committer | 2016-11-11 02:15:30 +0000 | |
commit | 17328cc3f88131cfdfb0fe0833336a583a191f97 (patch) | |
tree | 858fc8a56cfe0ce2e28e61dd8d56ef519ba8dcf9 /build | |
parent | 9dc55a9dc191d61e35d34042afd310626cb32538 (diff) | |
parent | 404fcf8f737f405da7fd2e61441dd20a57d63935 (diff) |
Merge "autopep8 make py files more standard"
Diffstat (limited to 'build')
-rw-r--r-- | build/gen_ins_pkg_script.py | 40 |
1 files changed, 25 insertions, 15 deletions
diff --git a/build/gen_ins_pkg_script.py b/build/gen_ins_pkg_script.py index ea6ccef5..e89b4285 100644 --- a/build/gen_ins_pkg_script.py +++ b/build/gen_ins_pkg_script.py @@ -1,12 +1,15 @@ -import yaml, os, sys +import yaml +import os +import sys from Cheetah.Template import Template + def get_file_list(root, arch): files = [] dirs = os.listdir(os.path.join(root, 'roles')) - for dir in dirs: + for dir in dirs: var_dir = os.path.join(root, 'roles', dir, 'vars') for name in ['main.yml', arch + r'.yml']: if os.path.exists(os.path.join(var_dir, name)): @@ -14,6 +17,7 @@ def get_file_list(root, arch): return files + def get_packages_name_list(file_list, special_packages): package_name_list = [] @@ -24,9 +28,10 @@ def get_packages_name_list(file_list, special_packages): for key, value in datas.items(): if key == "pip_packages": - continue + continue - if not key.endswith("packages") and not key.endswith("packages_noarch"): + if not key.endswith("packages") and not key.endswith( + "packages_noarch"): continue if not value: @@ -43,11 +48,17 @@ def get_packages_name_list(file_list, special_packages): return package_name_list + def generate_download_script(root="", arch="", tmpl="", docker_tmpl="", default_packages="", special_packages="", special_packages_script_dir="", special_packages_dir=""): - package_name_list = get_packages_name_list(get_file_list(root, arch), special_packages) if root else [] - - tmpl = Template(file=tmpl, searchList={'packages':package_name_list, 'default_packages':default_packages}) + package_name_list = get_packages_name_list( + get_file_list(root, arch), special_packages) if root else [] + + tmpl = Template( + file=tmpl, + searchList={ + 'packages': package_name_list, + 'default_packages': default_packages}) with open('work/repo/install_packages.sh', 'w') as f: f.write(tmpl.respond()) @@ -57,23 +68,22 @@ def generate_download_script(root="", arch="", tmpl="", docker_tmpl="", default_ if os.path.exists(os.path.join(special_packages_script_dir, name)): make_script.append(name) - searchList = {'scripts':make_script} + searchList = {'scripts': make_script} if os.path.exists(special_packages_dir): - special_packages_names=[] + special_packages_names = [] for parent, dirname, filenames in os.walk(special_packages_dir): for filename in filenames: - if os.path.isfile(os.path.join(parent, filename)): - special_packages_names.append(filename) - searchList.update({'spcial_packages':special_packages_names}) + if os.path.isfile(os.path.join(parent, filename)): + special_packages_names.append(filename) + searchList.update({'spcial_packages': special_packages_names}) - Dockerfile=os.path.basename(docker_tmpl).split('.')[0] + Dockerfile = os.path.basename(docker_tmpl).split('.')[0] tmpl = Template(file=docker_tmpl, searchList=searchList) with open(os.path.join('work/repo', Dockerfile), 'w') as f: f.write(tmpl.respond()) -if __name__=='__main__': +if __name__ == '__main__': # generate_download_script('ansible', 'Debian', 'Debian.tmpl') generate_download_script(sys.argv[1], sys.argv[2], sys.argv[3], sys.argv[4], sys.argv[5].split(' '), sys.argv[6].split(' '), sys.argv[7], sys.argv[8]) - |