o

�N�g4�@sPdZd
dl
Z
d
dlZd
dlZddd�
Zddd�
Zdd	�Zdd
d�
Zddd
�
Z	dS)zB
Read in files from the file_root and save files to the file root
�N�basec	Cs�g}|
td
v
r
|Std
|
D]H}t
j�||�}tjjj||dd�s$qt
j�|�
rXtjj	�
|d��}tjj	�|�
rB|�|di
�

n|�|di
�

Wd�
n1sSw



Y
q|S)zP
    Return a dict of the files located with the given path and environment
    �
file_rootsT�
Zsubdir�rb�txt�binN)
�__opts__�os�path�join�salt�utils�verify�
clean_path�isfile�files�fopenZis_text�append)r
�saltenv�ret�root�full�fp_�r�I/opt/saltstack/salt/lib/python3.10/site-packages/salt/wheel/file_roots.py�finds 








���rc

Cs�i}
|td
v
r
|
Std
|D]\}i|
|<t
jj�|�
D]N\}}}|
|}||krY|}g}tj�||�sPtj�|�
}	|	rC|�d|	�
tj�	|�
}tj�||�r5|D]}
||
}qR|D]}i||<q[|D]}d||<qdqq|
S)z>
    Return all of the file paths found in an environment
    rr
�
f)
rrr
r
Zos_walkr	�samefile�basename�insert�dirname)
rrZf_rootr�dirsr�subZsrootZaboverZarootZdir_�fn_rrr�list_env!s2











�







��r$cCs0i}td
D]}
g||
<||
�
t|
�
�

q|S)zE
    Return all of the files names in all available environments
    r)rrr$)rrrrr�
list_roots?s




r%c	Cs�g}t||
�}|D]8}t
t|���
�
}||}|d
krAtjj�|d��}|�|tjj	�
|���
i
�

Wd�
n1s<w



Y
q	|S)zP
    Read the contents of a text file, if the file is binary then ignore it
    rrN)r�next�iter�keysrr
rrr�stringutilsZ
to_unicode�read)r
rrrr#rZformrrrrr*Js







��r*cCs�|td
v
rd|�d�St
td
|�
|k
rd�||�Stj�|
�
r(d�|
|�Std
||}tj�||
�}tjj	j
||dd�sFd|
��Stj�|�
}tj�|�
sWt�
|�

tjj�|d	��}|�tjj�|�
�

Wd
�
n1stw



Y
d|��S)z�
    Write the named file, by default the first file found is written, but the
    index of the file can be specified to write to a lower priority file root
    rzNamed environment z is not presentz3Specified index {} in environment {} is not presentz;The path passed in {} is not relative to the environment {}TrzInvalid path: zw+NzWrote data to file )r�len�formatr	r
�isabsrrr
rrr �isdir�makedirsrr�writer)Zto_str)�datar
r�indexr�destZdest_dirrrrrr0Zs*



�

�









�
r0)
r)rr
)
�__doc__r	Zsalt.utils.filesrZsalt.utils.pathrr$r%r*r0rrrr�<module>
s