Current File : //proc/227033/root/lib/python2.7/site-packages/salt/queues/sqlite_queue.pyo
�
���^c @@ s d Z d d l m Z m Z m Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l
Z d d l m
Z
d d l m Z e j e � Z d Z d � Z d � Z d � Z d
� Z d � Z d � Z d
� Z d � Z d � Z d � Z d � Z d � Z d e d � Z! d S( u{
.. 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
i ( t absolute_importt print_functiont unicode_literalsN( t SaltInvocationError( t sixu sqlitec C@ s t S( N( t __virtualname__( ( ( s< /usr/lib/python2.7/site-packages/salt/queues/sqlite_queue.pyt __virtual__" s c C@ ss t d } t j j | d j | � � } t j d | � t j | � } t | � } | | k ro t
| | � n | S( u%
Return an sqlite connection
u sqlite_queue_diru {0}.dbu Connecting to: %s( t __opts__t ost patht joint formatt logt debugt sqlite3t connectt _list_tablest
_create_table( t queuet queue_dirt dbt cont tables( ( s< /usr/lib/python2.7/site-packages/salt/queues/sqlite_queue.pyt _conn( s
c C@ sg | �[ | j � } d } t j d | � | j | � | j � } g | D] } | d ^ qI SWd QXd S( Nu3 SELECT name FROM sqlite_master WHERE type = "table"u
SQL Query: %si ( t cursorR R
t executet fetchall( R t curt cmdt resultt x( ( s< /usr/lib/python2.7/site-packages/salt/queues/sqlite_queue.pyR 7 s
c C@ sI | �= | j � } d j | � } t j d | � | j | � Wd QXt S( Nu: CREATE TABLE {0}(id INTEGER PRIMARY KEY, name TEXT UNIQUE)u
SQL Query: %s( R R R R
R t True( R R R R ( ( s< /usr/lib/python2.7/site-packages/salt/queues/sqlite_queue.pyR A s c C@ sa t | � } | �I | j � } d j | � } t j d | � | j | � | j � } Wd QX| S( u6
Private function to list contents of a queue
u SELECT name FROM {0}u
SQL Query: %sN( R R R R R
R R ( R R R R t contents( ( s< /usr/lib/python2.7/site-packages/salt/queues/sqlite_queue.pyt _list_itemsK s
c C@ sg t d } t j j | d � } t j | � } g | D]( } t j j t j j | � � d ^ q5 } | S( u<
Return a list of sqlite databases in the queue_dir
u sqlite_queue_diru *.dbi ( R R R R
t globt splitextt basename( R t filest pathst itemt queues( ( s< /usr/lib/python2.7/site-packages/salt/queues/sqlite_queue.pyt _list_queuesY s
5c C@ s
t � } | S( u9
Return a list of Salt Queues on the Salt Master
( R) ( R( ( ( s< /usr/lib/python2.7/site-packages/salt/queues/sqlite_queue.pyt list_queuese s c C@ s- t | � } g | D] } | d ^ q } | S( u"
List contents of a queue
i ( R! ( R t
itemstupleR' t items( ( s< /usr/lib/python2.7/site-packages/salt/queues/sqlite_queue.pyt
list_itemsm s c C@ s t | � } t | � S( u0
Provide the number of items in a queue
( R! t len( R R, ( ( s< /usr/lib/python2.7/site-packages/salt/queues/sqlite_queue.pyt list_lengthv s c C@ s% t j d t j � } | j d | � S( ua
Make sure single quotes are escaped properly in sqlite3 fashion.
e.g.: ' becomes ''
u 'u ''( t ret compilet Mt sub( R' t rex_sqlquote( ( s< /usr/lib/python2.7/site-packages/salt/queues/sqlite_queue.pyt
_quote_escape~ s c C@ s� t | � } | ��| j � } t | t j � r� t | � } d j | | � } t j d | � y | j | � Wq� t
j k
r� } d j | � SXn t | t � rDg | D] } t | � ^ q� } d j | � } t j d | � g } x | D] } | j
| f � q� Wy | j | | � WqDt
j k
r@} d j | � SXn t | t � r�t j j j | � j d d � } t | � } t d � j | | � } t j d | � y | j | � Wq�t
j k
r�} d j | � SXn Wd QXt S( u)
Add an item or items to a queue
u# INSERT INTO {0}(name) VALUES('{1}')u
SQL Query: %su4 Item already exists in this queue. sqlite error: {0}u INSERT INTO {0}(name) VALUES(?)uA One or more items already exists in this queue. sqlite error: {0}u "u 'N( R R t
isinstanceR t string_typesR5 R R R
R R t IntegrityErrort listt appendt executemanyt dictt saltt utilst jsont dumpst replacet strR ( R R, R R R t esct elt newitemsR' ( ( s< /usr/lib/python2.7/site-packages/salt/queues/sqlite_queue.pyt insert� sF
! c C@ st t | � } | �\| j � } t | t j � rp t | � } d j | | � } t j d | � | j | � t
St | t � r� g | D] } t | � ^ q� } d j | � } t j d | � g } x | D] } | j | f � q� W| j
| | � n t | t � rft j j j | � j d d � } t | � } d j | | � } t j d | � | j | � t
St
SWd QXd S( u.
Delete an item or items from a queue
u" DELETE FROM {0} WHERE name = '{1}'u
SQL Query: %su DELETE FROM {0} WHERE name = ?u "u 'N( R R R6 R R7 R5 R R R
R R R9 R: R; R< R= R> R? R@ RA ( R R, R R R RD RE R' ( ( s< /usr/lib/python2.7/site-packages/salt/queues/sqlite_queue.pyt delete� s2
!
i c
C@ s� d j | � } | d k r} y t | � } Wn. t k
r[ } d j | � } t | � � n Xd j | d j | � g � } n t j d | � t | � } g } | �� | j � } | j | � j
� } t | � d k rAg | D] }
|
d ^ q� } d j | � } t | � d j | | � } t j d | � | j | � n | j
� Wd
QX| r�g | D]+ }
t j j j |
d j d d � � ^ q^} n t j | � | S(
uB
Pop one or more or all items from the queue return them.
u SELECT name FROM {0}u allu3 Quantity must be an integer or "all".
Error: "{0}".u u
LIMIT {0}u
SQL Query: %si u ","u% DELETE FROM {0} WHERE name IN ("{1}")Nu 'u "( R t intt
ValueErrorR R
R R
R R R R R. R5 t commitR= R> R? t loadsRA t info(
R t quantityt is_runnerR t exct error_txtR R, R R R' t itemlistt del_cmd( ( s< /usr/lib/python2.7/site-packages/salt/queues/sqlite_queue.pyt pop� s8 !
;
(" t __doc__t
__future__R R R R" t loggingR R0 R t salt.utils.jsonR= t salt.exceptionsR t salt.extR t getLoggert __name__R R R R R R R! R) R* R- R/ R5 RF RG t FalseRS ( ( ( s< /usr/lib/python2.7/site-packages/salt/queues/sqlite_queue.pyt <module> s0
*
Mini Shell