. d dl Z d dlZd dlmZ d dlmZ d dlmZ d dlmZ d dlmZ n# e $ r d dl
mZ Y nw xY w d dl Z d dl m Z
d dlmZ d dlmZ d dlmZ n# e $ r dxZ
xZZY nw xY wd dlmZ d d lmZ e j d
Z G d d e Z G d
de Z G d de Z G d de Z e Z! e Z" e Z# e Z$ G d de Z% G d de Z& G d de Z' G d de' Z(dS ) N)localEvent)Lock)Thread)Queue)Greenlet)__deprecated__)SqliteExtDatabasezpeewee.sqliteqc e Zd ZdS )
ResultTimeoutN__name__
__module__ __qualname__ q/builddir/build/BUILD/imunify360-venv-2.6.2/opt/imunify360/venv/lib/python3.11/site-packages/playhouse/sqliteq.pyr
r
Dr r
c e Zd ZdS ) WriterPausedNr r r r r r r r r c e Zd ZdS )ShutdownExceptionNr r r r r r " r r r c e Zd ZdZd ZddZddZd Zd ZeZ e
d Z e
d Z e
d
Z
d Zd Zd
ZdS ) AsyncCursor) sqlparamstimeout_event_cursor_exc_idx_rows_readyc || _ || _ || _ || _ d x| _ x| _ x| _ | _ d| _ d S NF) r r r r r r! r" r# r$ )selfeventr r r s r __init__zAsyncCursor.__init__* sE <@@ @ty@49tz r Nc || _ || _ d| _ || ng | _ | j | S )Nr )r r! r" fetchallr# r set)r' cursorexcs r
set_resultzAsyncCursor.set_result2 sH *-+V__&&&2
r c ||n| j }| j | s|rt d | j | j d| _ d S )N)r z results not ready, timed out.T)r r waitr
r! r$ )r' r s r _waitzAsyncCursor._wait: s_ $ 0''dl{ 00 AW A ?@@ @ 9 )O r c X | j s| | j | j | S N)r$ r2 r! r' s r __iter__zAsyncCursor.__iter__B s- { JJLLL 9 )O r c | j s| | j | j }| xj dz
c_ |S # t $ r t
w xY w)N )r$ r2 r# r"
IndexError
StopIteration)r' objs r nextzAsyncCursor.nextI se { JJLLL *TY'C
IINIIJ s A Ac P | j s| | j j S r4 )r$ r2 r lastrowidr5 s r r> zAsyncCursor.lastrowidU s$ { JJLLL|%%r c P | j s| | j j S r4 )r$ r2 r rowcountr5 s r r@ zAsyncCursor.rowcount[ s$ { JJLLL|$$r c | j j S r4 )r descriptionr5 s r rB zAsyncCursor.descriptiona s
|''r c 8 | j d S r4 )r closer5 s r rD zAsyncCursor.closee s r c t | S r4 )listr5 s r r+ zAsyncCursor.fetchallh s Dzzr c z | j s| t | S # t $ r Y d S w xY wr4 )r$ r2 r< r: r5 s r fetchonezAsyncCursor.fetchonek sL { JJLLL :: 44 s ,
::r4 )r r r __slots__r) r/ r2 r6 r< __next__propertyr> r@ rB rD r+ rH r r r r r & s II H
& & X&
% % X%
( ( X( r r c 0 e Zd ZdZd Zd Zd Zd Zd ZdS )Writerdatabasequeuec " || _ || _ d S r4 rN )r' rO rP s r r) zWriter.__init__| s
r c | j } |.| r| j }n| | }ng# t $ rZ t
d Y |:| j | | j j d S d S w xY w# |9| j | | j j w w xY w)NTz*writer received shutdown request, exiting.)
rO
connection wait_unpauseloopr loggerinfo_close_statereset)r' conns r runz
Writer.run s }'')) -
| ,,.. >#'=#;#;#=#=D #yy( KK LMMM
$$T***
$**,,,,,
$$T***
$**,,,, s/ C AA# "C #$CC CC =Dc | j \ }}|t u r0t d | dS |t u rt |t u r0t d | d S |
d t t d| d S )Nz+writer unpaused - reconnecting to database.Tz-writer received pause, but is already paused.z writer paused, not handling %s)
rP getUNPAUSErV rW r, SHUTDOWNr PAUSEerrorr/ r warningr' opr; s r rT zWriter.wait_unpause s *..""C
== KKE F F F GGIII4
8^^#%% %
5[[ LLH I I I GGIIIII NN4 0 0 0 NN;S A A A A Ar c 2 | j \ }}|t u r| | n|t u rht
d | j | | j j
| d S |t u r/t
d | n2|t u rt t
d| |S )Nz,writer paused - closing database connection.z0writer received unpause, but is already running.z&writer received unsupported object: %s)rP r^ QUERYexecutera rV rW rO rX rY rZ r, r_ rb r` r )r' r[ re r; s r rU z Writer.loop s *..""C
;; LL
5[[ KKF G G G M & & & M & & ( ( ( GGIII F
7]] LLK L L L GGIIII
8^^#%% % LLA3 G G G r c t d|j | j |j |j }d }n# t $ r}d }|}Y d }~nd }~ww xY w| || S )Nzreceived query %s)rV debugr rO _executer Exceptionr/ )r' r; r- r. execute_errs r rh zWriter.execute s (#'222 ]++CGSZ@@F
CC F CCCCCC
~~fc***s %A
A"A A"N) r r r rI r) r\ rT rU rh r r r rM rM y sg %I - - -$
B
B
B & + + + + +r rM c z e Zd ZdZ d fd Zd ZddZd Zd Zdd
Z d Z
d Z d
Z d Z
d Zd ZexZxZZ xZS )SqliteQueueDatabasezSQLite must be configured to use the WAL journal mode when using this feature. WAL mode allows one or more readers to continue reading while another connection writes to the database.FTNc d|d< t | _ | | dd }t t
| } | j | _ | j |g|R d|i| || _ || _
d| _ | | | | _
| | j r| d S d S )NFcheck_same_threadpragmasT)r _qlock_validate_journal_modepopsuperro execute_sqlrk r)
_autostart_results_timeout _is_stoppedget_thread_impl_thread_helper_create_write_queuestart) r' rO
use_gevent autostartqueue_max_sizeresults_timeoutargskwargsrr Parent __class__s r r) z SqliteQueueDatabase.__init__ s &+"# ff --fjjD.I.IJJ
*D11 *
CDCCC'CFCCC# / ?d 2 2: > >~ N N
""" ? JJLLLLL r c " |rt nt S r4 )GreenletHelper ThreadHelper)r' r s r r{ z#SqliteQueueDatabase.get_thread_impl s !+=~~ =r c |sddiS t |t st d |D }| dd dk rt | j d|d< |S )N journal_modewalc 3 H K | ] \ }}| |fV d S r4 )lower).0kvs r z=SqliteQueueDatabase._validate_journal_mode.. s2 @@fq!AGGIIq>@@@@@@r )
isinstancedictr^ r
ValueErrorWAL_MODE_ERROR_MESSAGE)r' rr s r rt z*SqliteQueueDatabase._validate_journal_mode s +"E* *'4(( A@@@@@@@G ;;~u - - 3 3 5 5 > > T 899 9"' r c B | j | _ d S r4 )r| rP _write_queuer5 s r r} z'SqliteQueueDatabase._create_write_queue s / 5 5 7 7r c 4 | j S r4 )r qsizer5 s r
queue_sizez SqliteQueueDatabase.queue_size s &&(((r c J |t d | d r| || S t | j |||| j n| }| j t |f |S )Nz,"commit" has been deprecated and is a no-op.select)r( r r r ) r
r
startswithrk r r| r( ry r putrg )r' r r commitr r- s r rw z SqliteQueueDatabase.execute_sql s I J J J 99;; ! !( + + .==f-- - %++---4_D))' K K K
ufo...
r c j 5 j s d d d dS fd} j | _ j d _ d d d dS # 1 swxY w Y d S )NFc Z t j } | d S r4 )rM r r\ )writerr' s r r\ z&SqliteQueueDatabase.start..run s& d&788
r T)rs rz r| thread_writerr~ )r' r\ s ` r r~ zSqliteQueueDatabase.start
s
[
#
.55c::DL L $D
s A1AA11A58A5c t t d | j 5 | j r d d d dS | j t d f | j | j s| j
\ }}|t k s |t k r|
n-|t k r"| d t! | j d| _ d d d dS # 1 swxY w Y d S )Nzenvironment stop requested.FT)rV rj rs rz r r r` r joinemptyr^ ra r_ r, rg r/ r rd s r stopzSqliteQueueDatabase.stop s 2333
[
! !8T"2 3 3 3 L '--//
> +//11C;;"--GGIIII5[[NN4):)<)<=== '--//
> $D ! s D-C(D--D14D1c R | j 5 | j cd d d S # 1 swxY w Y d S r4 )rs rz r5 s r
is_stoppedz SqliteQueueDatabase.is_stopped+ st
[ $ $# $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $s c
| j 5 | j r d d d dS | j }| j t |f d d d n # 1 swxY w Y | d S r& )rs rz r| r( r r ra r1 r' evts r pausezSqliteQueueDatabase.pause/ s
[ 0 0
0 0 0 0 0 0 0 0 %++--C ! !5#, / / / 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0
A$:A$$A(+A(c
| j 5 | j r d d d dS | j }| j t |f d d d n # 1 swxY w Y | d S r& )rs rz r| r( r r r_ r1 r s r unpausezSqliteQueueDatabase.unpause9 s
[ 2 2
2 2 2 2 2 2 2 2 %++--C ! !7C. 1 1 1 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2
r c @ t dt | z )Nz#This method is not supported by %r.)r type)r' r r s r __unsupported__z#SqliteQueueDatabase.__unsupported__C s >dKLLLr )FTNNr4 )NNN)r r r r r) r{ rt r} r rw r~ r r r r r atomic transaction savepoint
__classcell__)r s @r ro ro s * >B6: @> > >
8 8 8) ) ) ($ $ $ M M M'66F6[99999r ro c . e Zd ZdZddZd ZddZd ZdS )r r Nc || _ d S r4 r )r' r s r r) zThreadHelper.__init__K s ,r c t S r4 r r5 s r r( zThreadHelper.eventN s
EGG^r c < ||n| j }t |pd S Nr )maxsize)r r r' max_sizes r rP zThreadHelper.queueP s( ' 3889LX]++++r c 8 t ||| }d|_ |S )N)targetr r T)r daemon)r' fnr r r s r r zThreadHelper.threadT s" rV<<<
r r4 )r r r rI r) r( rP r r r r r r H s[ #I- - - - $##, , , , r r c & e Zd ZdZd ZddZd ZdS )r r c t S r4 )GEventr5 s r r( zGreenletHelper.event] s
FHH_r Nc < ||n| j }t |pd S r )r GQueuer s r rP zGreenletHelper.queue_ s( ' 3889Lhm!,,,,r c . fd}t |g|R i |S )Nc : t j | i |S r4 )geventsleep)ar r s r wrapz#GreenletHelper.thread..wrapd s LNNN2q;A;; r )GThread)r' r r r r s ` r r zGreenletHelper.threadc s< t-d---f---r r4 )r r r rI r( rP r r r r r r Z sE I$$$- - - -. . . . .r r ))loggingweakref threadingr thread_localr r r r ImportErrorrP r r r gevent.eventr gevent.localgreenlet_local gevent.queuer peeweer
playhouse.sqlite_extr getLoggerrV rl r
r r objectr r: r` rg ra r_ rM ro r r r r r r s + + + + + + %MMM******,,,,,,444444,,,,,,, % % % $$G$fvvv% " ! ! ! ! ! 2 2 2 2 2 2
+ , , I 9 K K K K K& K K KZ
&((D+ D+ D+ D+ D+V D+ D+ D+NE7 E7 E7 E7 E7+ E7 E7 E7P 6 $
.
.
.
.
.\
.
.
.
.
.s ) 77 A A&%A&