File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/proxy/__pycache__/dummy.cpython-310.pyc
o
�N�g � @ s( d Z ddlZddlZddlZddlZddlmZ ddlZddl Zddl
m Z m Z dgZ
e�e�Zdd� Zdd � Zd
d � Zed d
� �Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Z d d � Z d d!� Z d"d#� Z d$d%� Z d&d'� Z!d(d)� Z"d*d+� Z#d,d-� Z$d.d/� Z%d0d1� Z&d2d3� Z'd4d5� Z(dS )6zS
This is the a dummy proxy-minion designed for testing the proxy minion subsystem.
� N)�contextmanager)�CommandExecutionError� MinionErrorZdummyc C � t �d� dS )z6
Only return if all the modules are available
z#dummy proxy __virtual__() called...T��log�debug� r r �D/opt/saltstack/salt/lib/python3.10/site-packages/salt/proxy/dummy.py� __virtual__ �
r c C s� t d d }tj�| d d|� d��}tjj�|d��}|�tjj j
|dd�� W d � n1 s2w Y t � d |t
�|�� d S )
N� dummy_proxy�id�cachedir� dummy-proxy-�.cache�wbT)Z use_bin_typez Dummy Proxy Saved State(%s):
%s)� __context__�os�path�join�salt�utils�files�fopen�write�msgpackZpackbr �warning�pprint�pformat)�opts�details�_id� cachefile�pckr r r
� _save_state s �r% c
C s� t d d }tj�| d d|� d��}z%tjj�|d��}tjjj |�
� dd�}W d � n1 s2w Y W n2 t yI t � }t
| |� Y n" tyj } ztjd |d
d � t � }t
| |� W Y d }~nd }~ww t�d |t�|�� |S )
Nr
r r r r �rbF)�rawzFailed to load state: %sT)�exc_infoz Dummy Proxy Loaded State(%s):
%s)r r r r r r r r r Zunpackb�read�FileNotFoundError�_initial_stater% � Exceptionr � exceptionr r r )r r" r# r$ �state�excr r r
� _load_state' s$ �� ��r0 c c sN � t | �}t�|�}z|V W ||krt| |� d S d S ||kr&t| |� w w )N)r0 �copy�deepcopyr% )r r. Zoriginalr r r
�
_loaded_state8 s �
� �r3 c C s dddd�ddddd�d �S )
N�running�stopped)�apacheZntpZsamba�1.0z2.4z1.4z999.99)Z coreutilsr6 ZtincZredbull)�services�packagesr r r r r
r+ C s
��r+ c C s� | d � d�r tdd��| d � d�rtdd��d| d itd< t�d � t| ��
}d
|d < W d � d S 1 s9w Y d S )
zH
Required.
Can be used to initialize the server connection.
�proxyZraise_minion_errorzRaising A MinionError.)�messageZraise_commandexec_errorz Raising A CommandExecutionError.r r
z dummy proxy init() called...T� initializedN)�getr r r r r r3 �r r. r r r
�initT s
"�r? c C s8 t t��} | �dd�W d � S 1 sw Y dS )z�
Since grains are loaded in many different places and some of those
places occur before the proxy can be initialized, return whether
our init() function has been called
r< FN)r3 �__opts__r= �r. r r r
r< e s
$�r< c C sD t t��} dddd�| d< | d W d � S 1 sw Y dS )z
Make up some grains
ZoneZtwoZthree)Z
dummy_grain_1Z
dummy_grain_2Z
dummy_grain_3� grains_cacheN�r3 r@ rA r r r
�grainso s
�$�rD c C sV t t�� } d| v r| �d� W d � t� S W d � t� S 1 s#w Y t� S )z
Refresh the grains
rB N)r3 r@ �poprD rA r r r
�grains_refresh} s
��
��rF c C s ddiS )z)
Method called by grains module.
r! zbThis key is here because a function in grains/rest_sample.py called fns() here in the proxymodule.r r r r r
�fns� s �rG c C �: t t��}d|d | < W d � dS 1 sw Y dS )z/
Start a "service" on the dummy server
r4 r8 NrC ��namer. r r r
�
service_start� �
��rK c C rH )z.
Stop a "service" on the dummy server
r5 r8 NrC rI r r r
� service_stop� rL rM c C s dS )z0
Restart a "service" on the REST server
Tr )rJ r r r
�service_restart� s rN c C s8 t t��} t| d �W d � S 1 sw Y dS )z,
List "services" on the REST server
r8 N)r3 r@ �listrA r r r
� service_list� s
$�rP c C sZ t t�� }|d | dkrddiW d � S ddiW d � S 1 s&w Y dS )z:
Check if a service is running on the REST server
r8 r4 ZcommentNr5 rC rI r r r
�service_status� s
�$�rQ c C �4 t t�� } | d W d � S 1 sw Y dS )z6
List "packages" installed on the REST server
r9 NrC rA r r r
� package_list� �
$�rS c K s\ |� dd�r |d }nd}tt��}||d | < W d � | |iS 1 s%w Y | |iS )z0
Install a "package" on the REST server
�versionFr7 r9 N)r= r3 r@ )rJ �kwargsrU r. r r r
�package_install� s
��rW c C sj t t��'} | d D ]}t| d | �}|d }t|�| d |< q | d W d � S 1 s.w Y dS )z
"Upgrade" packages
r9 g �?N)r3 r@ �float�str)r. �pZ
version_floatr r r
�upgrade� s
$�r[ c C rR )zW
Call the REST endpoint to see if the packages on the "server" are up to date.
r9 NrC rA r r r
�uptodate� rT r\ c C sN dt d d< tt��}|d �| � |d W d � S 1 s w Y dS )z/
Remove a "package" on the REST server
Zbarr
Zfoor9 N)r r3 r@ rE rI r r r
�package_remove� s
$�r] c C sX t t�� }| |d v r| |d | iW d � S W d � dS 1 s%w Y dS )zG
Check the installation status of a package on the REST server
r9 NrC rI r r r
�package_status� s
�"�r^ c C r )z
Degenerate ping
zdummy proxy returning pingTr r r r r
�ping� r r_ c C s^ t �d� tt�� }d|v r t�|d � W d � dS W d � dS 1 s(w Y dS )z,
For this proxy shutdown is a no-op
z dummy proxy shutdown() called...�filenameN)r r r3 r@ r �unlinkr> r r r
�shutdown s
�"�rb c C r )zN
Test function so we have something to call from a state
:return:
ztest_from_state calledZ testvaluer r r r r
�test_from_state
s
rc ))�__doc__r1 �loggingr r �
contextlibr Zsalt.utils.filesr Zsalt.utils.msgpackZsalt.exceptionsr r Z__proxyenabled__� getLogger�__file__r r r% r0 r3 r+ r? r< rD rF rG rK rM rN rP rQ rS rW r[ r\ r] r^ r_ rb rc r r r r
�<module> sH