File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/__pycache__/templates.cpython-310.pyc
o
�N�gN^ � @ s d Z ddlZddlZddlZddlZddlZddlZddlZddl Z ddl
m Z ddl Z ddl
Z ddlZ ddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlmZ ddl m Z m Z m Z ddl!m"Z" ddl#m$Z$ ddl%m&Z&m'Z'm(Z( dd l)m*Z* dd
l+m,Z, e�-e.�Z/ej0�1e d d �Z2d Z3e�4e3�Z5G d
d� d�Z6G dd� d�Z7dd� Z8dd� Z9dd� Z:dd� Z;dd� Z<d+dd �Z=d+d d �Z>d+d d �Z?d+d!d"�Z@d+d#d$�ZAd+d%d&�ZBd,d(d)�ZCe9e>�ZDe9e?�ZEe9e@�ZFe9eA�ZGe9eB�ZHeDeEeCeFeGeHd*�ZIdS )-z
Template render systems
� N)�Path)�__path__)�CommandExecutionError�SaltInvocationError�SaltRenderError)�NamedLoaderContext)�__file_client__)� JinjaFilter� JinjaGlobal� JinjaTest)� OrderedDict)�VersionZ templateszutf-8c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �
AliasedLoaderah
Light wrapper around the LazyLoader to redirect 'cmd.run' calls to
'cmd.shell', for easy use of shellisms during templating calls
Dotted aliases ('cmd.run') must resolve to another dotted alias
(e.g. 'cmd.shell')
Non-dotted aliases ('cmd') must resolve to a dictionary of function
aliases for that module (e.g. {'run': 'shell'})
c C s
|| _ d S �N��wrapped)�selfr � r �H/opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/templates.py�__init__< �
zAliasedLoader.__init__c C s
| j | S r r �r �namer r r � __getitem__? r zAliasedLoader.__getitem__c C � t | j|�S r ��getattrr r r r r � __getattr__B � zAliasedLoader.__getattr__c C s
|| j v S r r r r r r � __contains__E r zAliasedLoader.__contains__N)�__name__�
__module__� __qualname__�__doc__r r r r r r r r r 0 s r c @ s e Zd ZdZdd� Zdd� ZdS )�
AliasedModulea
Light wrapper around module objects returned by the LazyLoader's getattr
for the purposes of `salt.cmd.run()` syntax in templates
Allows for aliasing specific functions, such as `run` to `shell` for easy
use of shellisms during templating calls
c C s || _ || _d S r )�aliasesr )r r r% r r r r R s
zAliasedModule.__init__c C r r r r r r r r V r zAliasedModule.__getattr__N)r r! r" r# r r r r r r r$ I s r$ c
C s i }|� dd�}| rktt| ��� �}|s |�dd�d }n-|�|� d��r1|dt|� d� }n|�|� d��rE|d t|� d� }nt�d
� d|v rV|�dd�d nd }|� t
| ||ra|nd|� dd�d
�� |� t
||� dd�|� dd�|� dd�d�� |S )ao
Generate SLS/Template Context Items
Return values:
tplpath - full path to template on filesystem including filename
tplfile - relative path to template -- relative to file roots
tpldir - directory of the template relative to file roots. If none, "."
tpldot - tpldir using dots instead of slashes, if none, ""
slspath - directory containing current sls - (same as tpldir), if none, ""
sls_path - slspath with underscores separating parts, if none, ""
slsdotpath - slspath with dots separating parts, if none, ""
slscolonpath- slspath with colons separating parts, if none, ""
�.�/� ���z.sls� Nz /init.sls� z(Failed to determine proper template pathr � )ZtplpathZtplfileZtpldirZtpldot�:�_)�slspathZ
slsdotpathZ slscolonpathZsls_path) �replace�strr �as_posix�rsplit�endswith�len�log�warning�update�dict)�tmplpath�sls� sls_contextr/ �templater r r �generate_sls_contextZ s:
��
�� r> c s d� fdd� }� |_ |S )a
Each template processing function below, ``render_*_tmpl``, is wrapped by
``render_tmpl`` before being inserted into the ``TEMPLATE_REGISTRY``. Some
actions are taken here that are common to all renderers. Perhaps a
standard decorator construct would have been more legible.
:param function render_str: Template rendering function to be wrapped.
Each function is responsible for rendering the source data for its
repective template language.
:returns function render_tmpl: The wrapper function
FNc
sN |d u ri }d|v rt |d �|d< |�|� |}d|v s J �d|v s%J �d|v r5t||d �}|�|� t| t�r�|r?| }n^z(|d urKtj�|| �} t� | dt
�� }|� � }W d � n1 saw Y W n5 t t
tfy� } z tjj�| �r�td| d�W Y d } ~ S tjd| | tjd � � d } ~ ww | � � }| �� z0� |||�}
tjj�� r�d
} tjjj |
t
d �� d tj f�r�d} tj �|
� � �}
| r�|
tj 7 }
W n" t y� } zt�!d
� � d } ~ w t"y� td
t#�$� d� Y S w |r�td|
d�S t%j&dd
tjjj'd��} | �(tjjj)|
t
d �� W d � n 1 �sw Y td| j*d�S )N�salt�opts�saltenvr; �rT��result�dataz,Exception occurred while reading file %s: %s)Zexc_info_on_loglevelF��encoding�
z Rendering exception occurred�wb)�delete�prefix)+r r8 r> �
isinstancer1 �os�path�join�codecs�open� SLS_ENCODING�read�UnicodeDecodeError�
ValueError�OSErrorr? �utils�filesZ is_binaryr9 r6 �error�logging�DEBUG�close�platformZ
is_windows� stringutils�
to_unicoder4 �linesep�
splitlinesr � exception� Exception� traceback�
format_exc�tempfileZNamedTemporaryFileZTEMPFILE_PREFIX�write�to_bytesr )
ZtmplsrcZfrom_str�to_str�contextr: �kwsr< �tmplstrZ_tmplsrc�exc�output�newlineZoutf��
render_strr r � render_tmpl� s�
����� � �
�
� � �� � z#wrap_tmpl_func.<locals>.render_tmpl)FFNNrp )rq rr r rp r �wrap_tmpl_func� s �Nrs c C s* z
dd� | D �d W S t y Y dS w )�C
Return the line number where the template error was found
c S s g | ]
}|d dv r|�qS )� )ztop-level template coder= z<module>r )�.0�xr r r �
<listcomp>� s
�z)_get_jinja_error_slug.<locals>.<listcomp>r) N)�
IndexError��tb_datar r r �_get_jinja_error_slug� s �� �r| c C s* z
t | �}dj|� W S ty Y dS w )zB
Return an understandable message from jinja error output
z
{0}({1}):
{3}N)r| �formatry )r{ �liner r r �_get_jinja_error_message s �r c C s$ zt | �d W S ty Y dS w )rt r( N)r| ry rz r r r �_get_jinja_error_line s �r� c C s& |si }d}t | �}t| �}t| �}d}d}d|vr/|d dkr.tj�|d �r.|d }d}n |d �dd �}|d �|�sLtj�|d �rLd}|d }|r�|r�d
|�� d � d
�}t j
j � |��} t j
j
�| �� �}
W d � n1 suw Y |t j
j
j|
|d d �7 }nd
|� d
�}d}||fS )
z�
Return the error line and error message output from
a stacktrace.
If we are in a macro, also output inside the message the
exact location of the error in the macro
r, FNr; r z <unknown>Tr& r' rH z <======================)Zmarker)r| r� r rM rN �existsr0 r4 ra r? rW rX �fopenr^ r_ rS Z get_context) �tracerj �outrY r~ �msgZadd_logZ
template_pathZfilenZfp_Ztemplate_contentsr r r �_get_jinja_error s: � �
� r� c
s� |d }|d }d}d}| rt | t�s| �t�} | �tj�r"tj}n| �d�r)d}�z�|s9|r8t�tj �
|��}nt j j
j|||�dd�|�dt�� �d�}g |d �� ttjd
�r`� d �d � ttjd
�rm� d �d� ttjd�rz� d �d� � d �t j j
j� |�di �}|�di �}t |t�r�|ni }t |t�r�|ni }|�dd�r�t�d� t�d� d|d< d|d< |�dd�r�t�d� t�d� d|d< d|d< � fd d �} d |v r�|d d kr�| |d� n| |d� |�d d�r�tjjd7i � ��}
n tjjd7d!tj i� ��}
|
j �d"�} |
j � t j!� |
j � t"j#� t j j
j$t%d#�k�r,| |
j d"< |
j&� t'j(� t)|
j&d$< t j j
j*|
j&d%< t j j+j,|
j d&< i } |�-� D ]E\}
}t |t��skt |t.��re|�� | |
< n|| |
< �qNzt j j/j0|td'�| |
< W �qN t1�y� t�d(t� t j j+�|�| |
< Y �qNw |
j&� | � z|
�2| �}|j3d7i | ��}W n� tj4j5�y� } z!t6�7t8�9� d) �}t:|| d*�\}}|�s�d } t;d+|� |� �|| ��d}~w tj4j<tj4j=tj4j>f�y } z!t6�7t8�9� d) �}t:|| d*�\}}|�s d } t;d,|� |� �|| ��d}~w t?t@f�y= } z t6�7t8�9� d) �}t:|| d*�\}}|�s/d } t;d-�A||�|| ��d}~w tB�y� } zSt6�C� }t6�7t8�9� d) �}t:|| d*�\}}|�s`d } n| d|� �7 } t�d.� tjd/dd0� t�d1|� t�d2|� t�d3| � t�d4|� t;d5|� |� �|| |d6��d}~ww W |�r�t |t j j
j��r�|�D� n|�r�t |t j j
j��r�|�D� w w w |�r�||7 }|S )8aU
Render a Jinja template.
:param str tmplstr: A string containing the source to be rendered.
:param dict context: Any additional context data used by the renderer.
:param str tmplpath: Base path from which ``tmplstr`` may load additional
template files.
:returns str: The string rendered by the template.
r@ rA NFrH � _pillar_rendZ
fileclient)� pillar_rendZ _file_client)�
extensions�loaderZwith_r� zjinja2.ext.with_Zdoz
jinja2.ext.doZ loopcontrolszjinja2.ext.loopcontrols� jinja_envZ
jinja_sls_envZjinja_trim_blocksz Jinja2 trim_blocks is enabledzzjinja_trim_blocks is deprecated and will be removed in a future release, please use jinja_env and/or jinja_sls_env insteadTZ trim_blocksZjinja_lstrip_blocksz Jinja2 lstrip_blocks is enabledz|jinja_lstrip_blocks is deprecated and will be removed in a future release, please use jinja_env and/or jinja_sls_env insteadZ
lstrip_blocksc sT | � � D ]#\}}|�� }ttj|�� �r!t�d|||� |� |< qt�d|� qd S )Nz)Jinja2 environment %s was set to %s by %sz'Jinja2 environment %s is not recognized) �items�lower�hasattr�jinja2�defaults�upperr6 �debugr7 )r@ Zoptname�k�v�Zenv_argsr r �opt_jinja_env_helper� s �
�z/render_jinja_tmpl.<locals>.opt_jinja_env_helperr; r, Zallow_undefinedZ undefined�indentz2.11Zodict�show_full_context�listrF zDFailed to decode using default encoding (%s), trying system encodingru )rj zJinja variable zJinja syntax error: z5Problem running salt function in Jinja template: {}{}z Jinja Errorz
Exception:)�exc_infozOut: %szLine: %sz TmplStr: %sz TraceStr: %sz
Jinja error: )r� r )ErL r1 �decoderR r4 rM r` r� ZFileSystemLoaderrN �dirnamer? rW �jinjaZSaltCacheLoader�getr �valuer� �ext�appendZSerializerExtensionr9 r6 r� r7 ZsandboxZSandboxedEnvironmentZStrictUndefined�filtersZtestsr8 r Zsalt_jinja_testsr Zsalt_jinja_filtersZ
JINJA_VERSIONr
�globalsr
Zsalt_jinja_globalsr r� rE Zis_listr� r r^ r_ rT Z from_string�render�
exceptionsZUndefinedErrorrd �
extract_tb�sysr� r� r ZTemplateRuntimeErrorZTemplateSyntaxErrorZ
SecurityErrorr r r} rc re �destroy)rl rj r: r@ rA r� ro Z
opt_jinja_envZopt_jinja_sls_envr� r� Z
indent_filterZdecoded_context�keyr� r= rn rm r� r~ r� Ztracestrr r� r �render_jinja_tmplJ s*
�
�
�
�
�
��
���
������
����0��
�r� c C s ddl }ddlm} ddlm} |d }d}|s+|r*ddlm} |tj� |�gd�}n
||d ||�
d d
�d �}z:z)|| d d
|v rI|d
� dd�nd|d�j di |��W W |rat
||�rb|�� S S S tyr t|j�� � � ��w |rt
||�r�|�� w w w )aT
Render a Mako template.
:param str tmplstr: A string containing the source to be rendered.
:param dict context: Any additional context data used by the renderer.
:param str tmplpath: Base path from which ``tmplstr`` may load additional
template files.
:returns str: The string rendered by the template.
r N��Template)�SaltMakoTemplateLookuprA )�TemplateLookup)� directoriesr@ r� F)r� Tr; r& r' )Zstrict_undefinedZuri�lookupr )Zmako.exceptionsZ
mako.templater� Zsalt.utils.makor� Z mako.lookupr� rM rN r� r� r0 r� rL r� rc r r� Ztext_error_template) rl rj r: �makor� r� rA r� r� r r r �render_mako_tmpl sB
�����
� ��
�r� c C s ddl m} || �jdi |��S )a
Render a Wempy template.
:param str tmplstr: A string containing the source to be rendered.
:param dict context: Any additional context data used by the renderer.
:param str tmplpath: Unused.
:returns str: The string rendered by the template.
r )�TemplateParserNr )Zwemplate.wemplater� r� )rl rj r: r� r r r �render_wempy_tmpl< s r� c C sz |� dd�}|dks|dkrddlm} || �}n|dkr(ddlm} || �}n
dd lm} || �}|jd i |���|�S ) a�
Render a Genshi template. A method should be passed in as part of the
context. If no method is passed in, xml is assumed. Valid methods are:
.. code-block:
- xml
- xhtml
- html
- text
- newtext
- oldtext
Note that the ``text`` method will call ``NewTextTemplate``. If ``oldtext``
is desired, it must be called explicitly
�methodZxml�textZnewtextr )�NewTextTemplateZoldtext)�OldTextTemplate)�MarkupTemplateNr )r� Zgenshi.templater� r� r� Zgenerater� )rl rj r: r� r� Ztmplr� r� r r r �render_genshi_tmplM s
r� c C sj ddl m} |�| �}||gd�}t| t�rt|�}nt| t�r%t|�}n td�t| ����t j
j � |�S )z$
Render a Cheetah template.
r r� )Z
namespacesz>Unknown type {!s} for Cheetah template while trying to render.)
ZCheetah.Templater� �compilerL r1 �bytesr r} �typer? rW rE r� )rl rj r: r� ZtclassrE �resr r r �render_cheetah_tmplo s
��r� Fc K s| t j�| �si S t j�| �}|�d�d }tj�|| �}tjj || |d�}|du r,t
� �tj� |�}|j �
|� |tj|< d|vrdd|v rdt|d|d � g d�}|D ]} d| � d�}
t||
|| � qS|D ]
} t|| || � qfz:|�� } |r~d | d
�W S tjj�� }
tjj�|
d ��}|�tjj�| �� W d � n1 s�w Y d |
d
�W S ty� t� � }d |d
� Y S w )
z�
Render a template from a python source file
Returns::
{'result': bool,
'data': <Error data or rendered file path>}
r& r )r� NZ__env__rA )r? ZgrainsZpillarr@ �__TrC zw+F) rM rN �isfile�basename�split� importlib� machinery�SourceFileLoader�util�spec_from_file_location� ImportError�module_from_specr� � exec_moduler� �modules�setattr�runr? rW rX Zmkstempr� rg r^ ri rc rd re )Zsfn�string�kwargsZ
base_fnamer r� �spec�mod�builtinsZbuiltin�arg�kwargrE Ztgt�targetZtrbr r r �py� s@
� �r� )r� r� r� ZwempyZgenshiZcheetahr )F)Jr# rP Zimportlib.machineryr� �importlib.utilrZ rM r� rf rd �pathlibr r� Z
jinja2.extZjinja2.sandboxZsalt.modules.matchr? Zsalt.utils.dataZsalt.utils.dateutilsZsalt.utils.filesZsalt.utils.hashutilsZsalt.utils.httpZsalt.utils.jinjaZsalt.utils.networkZsalt.utils.platformZsalt.utils.stringutilsZsalt.utils.yamlencodingr ZsaltpathZsalt.exceptionsr r r Zsalt.loader.contextr Zsalt.loader.dunderr Zsalt.utils.decorators.jinjar r
r Zsalt.utils.odictr Zsalt.utils.versionsr
� getLoggerr r6 rN rO ZTEMPLATE_DIRNAMErR �
getencoderZ SLS_ENCODERr r$ r> rs r| r r� r� r� r� r� r� r� r� ZJINJAZMAKOZWEMPYZGENSHIZCHEETAHZTEMPLATE_REGISTRYr r r r �<module> s|
=`
.
G
,
"
/
�