49 lines
2 KiB
Python
49 lines
2 KiB
Python
|
#!/usr/bin/python3
|
||
|
|
|
||
|
|
from os import listdir, makedirs
|
||
|
|
from os.path import isfile, isdir, join, dirname
|
||
|
|
from shutil import copy2, copytree, rmtree
|
||
|
|
|
||
|
|
from ansible.plugins.action import ActionBase
|
||
|
|
from risotto.utils import RISOTTO_CONFIG
|
||
|
|
|
||
|
|
|
||
|
|
class ActionModule(ActionBase):
|
||
|
|
def run(self, tmp=None, task_vars=None):
|
||
|
|
super(ActionModule, self).run(tmp, task_vars)
|
||
|
|
module_args = self._task.args.copy()
|
||
|
|
modules = module_args['modules']
|
||
|
|
dataset_directory = RISOTTO_CONFIG['directories']['dataset']
|
||
|
|
install_dir = join('/tmp/risotto/images')
|
||
|
|
if isdir(install_dir):
|
||
|
|
rmtree(install_dir)
|
||
|
|
for module_name, depends in modules.items():
|
||
|
|
for depend in depends:
|
||
|
|
manual = join(dataset_directory, depend, 'manual', 'image')
|
||
|
|
if not isdir(manual):
|
||
|
|
continue
|
||
|
|
for filename in listdir(manual):
|
||
|
|
src_file = join(manual, filename)
|
||
|
|
dst_file = join(install_dir, module_name, filename)
|
||
|
|
if isdir(src_file):
|
||
|
|
if not isdir(dst_file):
|
||
|
|
makedirs(dst_file)
|
||
|
|
for subfilename in listdir(src_file):
|
||
|
|
if not isfile(dst_file):
|
||
|
|
src = join(src_file, subfilename)
|
||
|
|
dst = join(dst_file, subfilename)
|
||
|
|
if isfile(src):
|
||
|
|
copy2(src, dst)
|
||
|
|
else:
|
||
|
|
copytree(src, dst)
|
||
|
|
elif not isfile(dst_file):
|
||
|
|
dst = dirname(dst_file)
|
||
|
|
if not isdir(dst):
|
||
|
|
makedirs(dst)
|
||
|
|
if isfile(src_file):
|
||
|
|
copy2(src_file, dst_file)
|
||
|
|
else:
|
||
|
|
copytree(src_file, dst_file)
|
||
|
|
return dict(ansible_facts=dict({}))
|
||
|
|
|