File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/pillar/__pycache__/hg_pillar.cpython-310.pyc
o
�N�g� � @ s� d Z ddlZddlZddlZddlZddlZddlZddlm Z zddl
Z
W n e y1 dZ
Y nw e� e
�ZdZdd� Zdd� Zdd
d �Zd d
� ZG dd� d�ZdS )a�
Use remote Mercurial repository as a Pillar source.
.. versionadded:: 2015.8.0
The module depends on the ``hglib`` python module being available.
This is the same requirement as for hgfs\_ so should not pose any extra
hurdles.
This external Pillar source can be configured in the master config file as such:
.. code-block:: yaml
ext_pillar:
- hg: ssh://hg@example.co/user/repo
� N)�DEFAULT_HASH_TYPE�hgc C sB dd� t �dg �D �} tdd� | D ��sdS ts t�d� dS tS )z*
Only load if hglib is available.
c S s g | ]}|�qS � r ��.0�xr r �I/opt/saltstack/salt/lib/python3.10/site-packages/salt/pillar/hg_pillar.py�
<listcomp>* s z __virtual__.<locals>.<listcomp>�
ext_pillarc S s g | ]}d |v �qS )r r r r r r r + s Fzhglib not present)�__opts__�get�any�hglib�log�error�__virtualname__)Zext_pillar_sourcesr r r � __virtual__& s
r c C s dS )zX
Initialise
This is called every time a minion calls this external pillar.
Nr )r r r r �__init__3 s r �defaultc C s� t |��
}|�|� W d � n1 sw Y |dkr dn|}|r0tj�tj�|j|��}n|j}t�t �}|g|d |< t
j � |t
| |�}|jdd�S )z.
Extract pillar from an hg repository
Nr �baseZ pillar_rootsF)�ext)�Repo�update�os�path�normpath�join� working_dir�copy�deepcopyr �salt�pillarZPillarZ
__grains__Zcompile_pillar) Z minion_idr! �repo�branch�rootZenvnamer �optsZpilr r r r
; s
�
r
c C s6 t | ��
}|�� W d � dS 1 sw Y dS )z-
Execute an hg pull on all the repos
N)r �pull)�repo_urir" r r r r M s
"�r c @ sB e Zd ZdZdd� Zdd� Zddd�Zd d
� Zd d � Zd
d� Z dS )r z?
Deal with remote hg (mercurial) repository for Pillar
c C s� || _ tj�td d�}ttt�dt��}|t j
j � |���
� }tj�||�| _tj�| j�s>t�|| j�| _| j�� dS t�| j�| _dS )z6Initialize a hg repo (or open it if it already exists)�cachedirZ hg_pillar� hash_typeN)r' r r r r �getattr�hashlibr r r �utilsZ stringutils�to_bytes� hexdigestr �isdirr �cloner" �open)�selfr' r( r) Z repo_hashr r r r Z s z
Repo.__init__c C s t �d� | j�� d S )N�-Updating hg repo from hg_pillar module (pull))r �debugr" r& �r2 r r r r& g s
z Repo.pullr c C s2 t �d� | j�� t �d� | jj|dd� dS )zN
Ensure we are using the latest revision in the hg repository
r3 z/Updating hg repo from hg_pillar module (update)T)ZcleanN)r r4 r" r& r )r2 r# r r r r k s
z Repo.updatec C s | j �� dS )z2
Cleanup mercurial command server
N)r" �closer5 r r r r6 t s z
Repo.closec C s | S �Nr r5 r r r � __enter__z s zRepo.__enter__c C s | � � d S r7 )r6 )r2 �exc_type� exc_value� tracebackr r r �__exit__} s z
Repo.__exit__N)r )
�__name__�
__module__� __qualname__�__doc__r r& r r6 r8 r< r r r r r U s
r )r N)r@ r r+ �loggingr Z salt.pillarr Zsalt.utils.stringutilsZ salt.configr r � ImportError� getLoggerr= r r r r r
r r r r r r �<module> s( �