Viewing: sqlite_queue.cpython-36.pyc
3 �N�b� � @ s� d Z ddlZddlZddlZddlZddlZddlZddlm Z ej 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 �"/usr/lib/python3.6/sqlite_queue.py� __virtual__ s r c C sP t d }tjj|dj| ��}tjd|� tj|�}t |�}| |krLt || � |S )z% Return an sqlite connection �sqlite_queue_dirz{}.dbzConnecting to: %s) �__opts__�os�path�join�format�log�debug�sqlite3Zconnect� _list_tables� _create_table)�queue� queue_dirZdb�conZtablesr r r �_conn$ s r c C sH | �: | j � }d}tjd|� |j|� |j� }dd� |D �S Q R X d S )Nz3SELECT name FROM sqlite_master WHERE type = "table"z SQL Query: %sc S s g | ] }|d �qS )r r )�.0�xr r r � <listcomp>: s z _list_tables.<locals>.<listcomp>)�cursorr r �execute�fetchall)r �cur�cmd�resultr r r r 3 s r c C s< | �. | j � }dj|�}tjd|� |j|� W d Q R X dS )Nz9CREATE TABLE {}(id INTEGER PRIMARY KEY, name TEXT UNIQUE)z SQL Query: %sT)r r r r r )r r r r r r r r = s r c C sL t | �}|�6 |j� }dj| �}tjd|� |j|� |j� }W dQ R X |S )z6 Private function to list contents of a queue zSELECT name FROM {}z SQL Query: %sN)r r r r r r r )r r r r �contentsr r r � _list_itemsF s r c C s2 t d } tjj| d�}tj|�}dd� |D �}|S )z< Return a list of sqlite databases in the queue_dir r z*.dbc S s$ g | ] }t jjt jj|��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 r) c C s t | �}dd� |D �}|S )z" List contents of a queue c S s g | ] }|d �qS )r r )r r# r r r r m s z list_items.<locals>.<listcomp>)r )r Z itemstuple�itemsr r r � list_itemsh s r+ c C s t | �}t|�S )z0 Provide the number of items in a queue )r �len)r r* r r r � list_lengthq s r- c C s t jdt j�}|jd| �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 r3 c 3 C s� t | �}|��v |j� }t|t�rzt|�}dj| |�}tjd|� y|j|� W n( t j k rx } z dj|�S d}~X nX t|t ��rdd� |D �}dj| �}tjd|� g }x|D ]}|j |f� q�W y|j ||� W n* t j k �r } z dj|�S d}~X nX t|t��r�tjjj|�jd d �}t|�}dj| |�}tjd|� y|j|� W n* t j k �r~ } z dj|�S d}~X nX W dQ R X d S ) z) Add an item or items to a queue z!INSERT INTO {}(name) VALUES('{}')z SQL Query: %sz3Item already exists in this queue. sqlite error: {}Nc S s g | ] }t |��qS r )r3 )r �elr r r r � s zinsert.<locals>.<listcomp>z INSERT INTO {}(name) VALUES(?)z@One or more items already exists in this queue. sqlite error: {}�"r. T)r r � isinstance�strr3 r r r r r ZIntegrityError�list�append� executemany�dict�salt�utils�json�dumps�replace)r r* r r r �esc�newitemsr# r r r �insert� sB rC c C s t | �}|�� |j� }t|t�rNt|�}dj| |�}tjd|� |j|� dS t|t �r�dd� |D �}dj| �}tjd|� g }x|D ]}|j |f� q�W |j ||� t|t �r�t jjj|�jdd�}t|�}dj| |�}tjd|� |j|� dS dS Q R X d S ) z. Delete an item or items from a queue z DELETE FROM {} WHERE name = '{}'z SQL Query: %sTc S s g | ] }t |��qS r )r3 )r r4 r r r r � s zdelete.<locals>.<listcomp>z DELETE FROM {} WHERE name = ?r5 r. N)r r r6 r7 r3 r r r r r8 r9 r: r; r< r= r> r? r@ )r r* r r r rB r# r r r �delete� s2 rD � Fc C s dj | �}|dkrhy t|�}W n4 tk rR } zdj |�}t|��W Y dd}~X nX dj|dj |�g�}tjd|� t| �}g }|�r |j� }|j |�j � } t | �dkr�d d � | D �}d j|�} t | � d j | | �} tjd| � |j | � |j � W dQ R X |�rd d � | D �}tj|� |S )zB Pop one or more or all items from the queue return them. zSELECT name FROM {}�allz2Quantity must be an integer or "all". Error: "{}".N� z LIMIT {}z SQL Query: %sr c S s g | ] }|d �qS )r r )r r# r r r r � s zpop.<locals>.<listcomp>z","z#DELETE FROM {} WHERE name IN ("{}")c S s& g | ] }t jjj|d jdd���qS )r r. r5 )r< r= r> �loadsr@ )r r# r r r r � s )r �int� ValueErrorr r r r r r r r r, r3 Zcommit�info) r ZquantityZ is_runnerr �excZ error_txtr r* r r ZitemlistZdel_cmdr r r �pop� s6 rM )rE F)�__doc__r$ Zloggingr r/ r Zsalt.utils.jsonr< Zsalt.exceptionsr Z getLogger�__name__r r r r r r r r( r) r+ r- r3 rC rD rM r r r r �<module> s, *
Return