valitydev/salt
14
0
Fork 0
mirror of https://github.com/valitydev/salt.git synced 2024-11-08 01:18:58 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #23530 from dr4Ke/fix_salt-ssh_to_include_pkg_sources

Browse Source 
salt-ssh state: fix including all salt:// references
This commit is contained in:
Thomas S Hatch 2015-05-12 09:13:43 -06:00
commit a664a3c6fd
1 changed files with 11 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
salt/client/ssh/state.py
View File

@ -99,20 +99,22 @@ def lowstate_file_refs(chunks, extras=''):
return refs
def salt_refs(data):
def salt_refs(data, ret=None):
'''
Pull salt file references out of the states
'''
proto = 'salt://'
ret = []
if ret is None:
ret = []
if isinstance(data, str):
if data.startswith(proto):
return [data]
if data.startswith(proto) and data not in ret:
ret.append(data)
if isinstance(data, list):
for comp in data:
if isinstance(comp, str):
if comp.startswith(proto):
ret.append(comp)
salt_refs(comp, ret)
if isinstance(data, dict):
for comp in data:
salt_refs(data[comp], ret)
return ret
@ -154,7 +156,7 @@ def prep_trans_tar(file_client, chunks, file_refs, pillar=None):
if not os.path.isdir(tgt_dir):
os.makedirs(tgt_dir)
shutil.copy(path, tgt)
break
continue
files = file_client.cache_dir(name, saltenv)
if files:
for filename in files:
@ -170,7 +172,7 @@ def prep_trans_tar(file_client, chunks, file_refs, pillar=None):
if not os.path.isdir(tgt_dir):
os.makedirs(tgt_dir)
shutil.copy(filename, tgt)
break
continue
cwd = os.getcwd()
os.chdir(gendir)
with closing(tarfile.open(trans_tar, 'w:gz')) as tfp: