golden hour
/opt/saltstack/salt/lib/python3.10/site-packages/salt/queues/__pycache__
⬆️ Go Up
Upload
File/Folder
Size
Actions
__init__.cpython-310.pyc
195 B
Del
OK
pgjsonb_queue.cpython-310.pyc
7.28 KB
Del
OK
sqlite_queue.cpython-310.pyc
6.78 KB
Del
OK
Edit: sqlite_queue.cpython-310.pyc
o �xe� � @ s� d Z ddlZddlZddlZddlZddlZddlZddlm Z e� e �Z dZ dd� Zdd� Zd d � Zd d � Zd d� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd!d d �ZdS )"a{ .. versionadded:: 2014.7.0 This is the default local master event queue built on sqlite. By default, an sqlite3 database file is created in the `sqlite_queue_dir` which is found at:: /var/cache/salt/master/queues It's possible to store the sqlite3 database files by setting `sqlite_queue_dir` to another location:: sqlite_queue_dir: /home/myuser/salt/master/queues � N)�SaltInvocationErrorZsqlitec C s t S )N)�__virtualname__� r r �L/opt/saltstack/salt/lib/python3.10/site-packages/salt/queues/sqlite_queue.py� __virtual__ s r c C sP t d }tj�|d�| ��}t�d|� t�|�}t |�}| |vr&t || � |S )z% Return an sqlite connection �sqlite_queue_dirz{}.dbzConnecting to: %s) �__opts__�os�path�join�format�log�debug�sqlite3�connect� _list_tables� _create_table)�queue� queue_dirZdb�conZtablesr r r �_conn$ s r c C s` | �$ | � � }d}t�d|� |�|� |�� }dd� |D �W d � S 1 s)w Y d S )Nz3SELECT name FROM sqlite_master WHERE type = "table"� SQL Query: %sc S � g | ]}|d �qS �r r )�.0�xr r r � <listcomp>: � z _list_tables.<locals>.<listcomp>)�cursorr r �execute�fetchall)r �cur�cmd�resultr r r r 3 s $�r c C sR | � | � � }d�|�}t�d|� |�|� W d � dS 1 s"w Y dS )Nz9CREATE TABLE {}(id INTEGER PRIMARY KEY, name TEXT UNIQUE)r T)r r r r r )r r r! r" r r r r = s ��r c C sb t | �}|�! |�� }d�| �}t�d|� |�|� |�� }W d � |S 1 s*w Y |S )z6 Private function to list contents of a queue �SELECT name FROM {}r N)r r r r r r r )r r r! r" �contentsr r r � _list_itemsF s ��r& c C s2 t d } tj�| d�}t�|�}dd� |D �}|S )z< Return a list of sqlite databases in the queue_dir r z*.dbc S s$ g | ]}t j�t j�|��d �qS r )r r �splitext�basename�r �itemr r r r [ s $ z _list_queues.<locals>.<listcomp>)r r r r �glob)r �files�paths�queuesr r r � _list_queuesT s r/ c C s t � } | S )z9 Return a list of Salt Queues on the Salt Master )r/ )r. r r r � list_queues` s r0 c C s t | �}dd� |D �}|S )z" List contents of a queue c S r r r r) r r r r m r z list_items.<locals>.<listcomp>)r& )r Z itemstuple�itemsr r r � list_itemsh s r2 c C s t | �}t|�S )z0 Provide the number of items in a queue )r& �len)r r1 r r r � list_lengthq s r4 c C s t �dt j�}|�d| �S )za Make sure single quotes are escaped properly in sqlite3 fashion. e.g.: ' becomes '' �'z'')�re�compile�M�sub)r* Z rex_sqlquoter r r � _quote_escapey s r: c C s� t | �}|�� |�� }t|t�rGt|�}d�| |�}t�d|� z|�|� W n t j yF } zd�|�W Y d}~W d � S d}~ww t|t �r�dd� |D �}d�| �}t�d|� g }|D ]}|� |f� qbz|� ||� W n t j y� } zd�|�W Y d}~W d � S d}~ww t|t�r�tjj�|��d d �}t|�}d�| |�}t�d|� z|�|� W n' t j y� } zd�|�W Y d}~W d � S d}~ww W d � d S W d � d S 1 s�w Y d S ) z) Add an item or items to a queue z!INSERT INTO {}(name) VALUES('{}')r z3Item already exists in this queue. sqlite error: {}Nc S � g | ]}t |��qS r �r: �r �elr r r r � r zinsert.<locals>.<listcomp>z INSERT INTO {}(name) VALUES(?)z@One or more items already exists in this queue. sqlite error: {}�"r5 T)r r � isinstance�strr: r r r r r ZIntegrityError�list�append� executemany�dict�salt�utils�json�dumps�replace)r r1 r r! r" �esc�newitemsr* r r r �insert� sd �� ��� �� � �"� �"�"rM c C s2 t | �}|�� |�� }t|t�r.t|�}d�| |�}t�d|� |�|� W d � dS t|t �rXdd� |D �}d�| �}t�d|� g }|D ]}|� |f� qI|� ||� t|t �r�t jj�|��dd �}t|�}d�| |�}t�d|� |�|� W d � dS W d � dS 1 s�w Y dS ) z. Delete an item or items from a queue z DELETE FROM {} WHERE name = '{}'r NTc S r; r r< r= r r r r � r zdelete.<locals>.<listcomp>z DELETE FROM {} WHERE name = ?r? r5 )r r r@ rA r: r r r r rB rC rD rE rF rG rH rI rJ )r r1 r r! r" rL r* r r r �delete� s8 � �$�rN � Fc C s& d� | �}|dkr/zt|�}W n ty$ } z d� |�}t|��d}~ww d�|d� |�g�}t�d|� t| �}g }|�> |�� }|� |�� � } t | �dkrpd d � | D �}d �|�} t | � d � | | �} t�d| � |� | � |� � W d � n1 s~w Y |r�d d � | D �}t�|� |S )zB Pop one or more or all items from the queue return them. r$ �allz2Quantity must be an integer or "all". Error: "{}".N� z LIMIT {}r r c S r r r r) r r r r � r zpop.<locals>.<listcomp>z","z#DELETE FROM {} WHERE name IN ("{}")c S s& g | ]}t jj�|d �dd���qS )r r5 r? )rF rG rH �loadsrJ r) r r r r � s & )r �int� ValueErrorr r r r r r r r r3 r: Zcommit�info) r ZquantityZ is_runnerr" �excZ error_txtr r1 r! r# ZitemlistZdel_cmdr r r �pop� s> ��� � rW )rO F)�__doc__r+ Zloggingr r6 r Zsalt.utils.jsonrF Zsalt.exceptionsr Z getLogger�__name__r r r r r r r& r/ r0 r2 r4 r: rM rN rW r r r r �<module> s. *
Save