ÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂ U S¤ðW[!ã@sddlZddlZddlmZddlmZzddlmZWn ek rXddlmZYnXz0ddlZddlm Z ddlmZddl mZWn ek rªdZ ZZYnXddlmZe d¡ZGdd „d eƒZGd d„deƒZdZd ZGdd„deƒZGdd„deƒZGdd„deƒZdS)éN©ÚEvent)ÚThread)ÚQueue)ÚGreenlet)ÚSqliteExtDatabasezpeewee.sqliteqc@seZdZdS)Ú ResultTimeoutN)Ú__name__Ú __module__Ú__qualname__©rrúD/opt/alt/python38/lib64/python3.8/site-packages/playhouse/sqliteq.pyrsrc@s|eZdZdZdd„Zddd„Zddd„Zd d „Zdd„ZeZ e d d„ƒZe dd„ƒZe dd„ƒZ dd„Zdd„Zdd„ZdS)ÚAsyncCursor) ÚsqlÚparamsÚcommitÚtimeoutÚ_eventÚ_cursorÚ_excÚ_idxÚ_rowscCs:||_||_||_||_||_d|_|_|_|_dS©N) rrrrrrrrr)ÚselfÚeventrrrrrrr Ú__init__ szAsyncCursor.__init__NcCs6||_||_d|_|dkr"| ¡ng|_|j ¡|S)Nr)rrrÚfetchallrrÚset)rÚcursorÚexcrrr Ú set_result(s zAsyncCursor.set_resultcCs@|dk r|n|j}|jj|ds,|r,tdƒ‚|jdk r<|j‚dS)N)rzresults not ready, timed out.)rrÚwaitrr)rrrrr Ú_wait0s zAsyncCursor._waitcCs| ¡|jdk r|j‚|Sr)r"rÚ_exec©rrrr Ú__iter__7s zAsyncCursor.__iter__cCs@z|j|j}Wntk r(t‚YnX|jd7_|SdS)Né)rrÚ IndexErrorÚ StopIteration)rÚobjrrr Únext=s zAsyncCursor.nextcCs| ¡|jjSr)r"rÚ lastrowidr$rrr r+GszAsyncCursor.lastrowidcCs| ¡|jjSr)r"rÚrowcountr$rrr r,LszAsyncCursor.rowcountcCs|jjSr)rÚdescriptionr$rrr r-QszAsyncCursor.descriptioncCs|j ¡dSr)rÚcloser$rrr r.UszAsyncCursor.closecCst|ƒSr)Úlistr$rrr rXszAsyncCursor.fetchallcCs.| ¡z t|ƒWStk r(YdSXdSr)r"r*r(r$rrr Úfetchone[s zAsyncCursor.fetchone)N)N)r r rÚ __slots__rr r"r%r*Ú__next__Úpropertyr+r,r-r.rr0rrrr rs rz×threadlocals cannot be set to True when using the Sqlite thread / queue database. All queries are serialized through a single connection, so allowing multiple threads to connect defeats the purpose of this database.z·SQLite 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.csreZdZd‡fdd„ Zdd„Zddd „Zd d„Zdd „Zdd„Zdd„Z ddd„Z dd„Zdd„Zdd„Z ‡ZS)ÚSqliteQueueDatabaseFr&Ncs¬| d¡rttƒ‚d|d<d|d<| | dd¡| dd¡¡} tt|ƒ} | j|_| j |f|žd| i|—Ž||_ ||_||_d|_ | |¡|ƒ|_| ¡|j r¨| ¡dS)NZthreadlocalsFZcheck_same_threadÚjournal_modeÚpragmasT)ÚgetÚ ValueErrorÚTHREADLOCAL_ERROR_MESSAGEÚ_validate_journal_modeÚpopÚsuperr4Úexecute_sqlÚ!_SqliteQueueDatabase__execute_sqlrZ _autostartÚ_results_timeoutÚ_num_readersÚ_is_stoppedÚget_thread_implÚ_thread_helperÚ_create_queues_and_workersÚstart)rZdatabaseÚ use_geventZ autostartZreadersÚqueue_max_sizeZresults_timeoutÚargsÚkwargsr6ZParent©Ú __class__rr ros& þ zSqliteQueueDatabase.__init__cCs|rtStSr)ÚGreenletHelperÚThreadHelper)rrFrrr rBsz#SqliteQueueDatabase.get_thread_implcCsh|r| ¡dkrttƒ‚|r^tdd„|Dƒƒ}| dd¡ ¡dkrJttƒ‚dd„|DƒdgSdgSdS)NÚwalcss|]\}}| ¡|fVqdSr)Úlower©Ú.0ÚkÚvrrr Ú ˜sz=SqliteQueueDatabase._validate_journal_mode..r5cSs g|]\}}|dkr||f‘qS)r5rrPrrr Ú œsÿz>SqliteQueueDatabase._validate_journal_mode..)r5rN)rOr8ÚWAL_MODE_ERROR_MESSAGEÚdictr7)rr5r6Zpdictrrr r:“sÿz*SqliteQueueDatabase._validate_journal_modecsPˆj ¡ˆ_ˆj ¡ˆ_ˆj‰ˆj ˆˆj¡ˆ_‡‡fdd„tˆjƒDƒˆ_ dS)Ncsg|]}ˆj ˆˆj¡‘qSr)rCÚthreadÚ_read_queue)rQÚ_©rÚtargetrr rU§sÿzBSqliteQueueDatabase._create_queues_and_workers..) rCÚqueueÚ_write_queuerYÚ_run_worker_looprXÚ_writerÚranger@Ú_readersr$rr[r rD¡sÿz.SqliteQueueDatabase._create_queues_and_workerscCs<| ¡}|tkrt d¡dSt d|j¡| |¡qdS)Nzworker shutting down.zreceived query %s)r7r(ÚloggerÚinfoÚdebugrÚ_process_execution)rr]Úasync_cursorrrr r_ªs z$SqliteQueueDatabase._run_worker_loopc CsPz| |j|j|j¡}Wn&tk r>}zd}W5d}~XYnXd}| ||¡Sr)r>rrrÚ Exceptionr )rrgrrrrr rf´sÿz&SqliteQueueDatabase._process_executioncCs|j ¡|j ¡fSr)r^ZqsizerYr$rrr Ú queue_size¾szSqliteQueueDatabase.queue_sizeTcCsDt|j ¡||||dkr|jn|d}|r0|jn|j}| |¡|S)N)rrrrr)rrCrr?r^rYÚput)rrrZrequire_commitrrr]rrr r=Ásû zSqliteQueueDatabase.execute_sqlc Csb|jR|jsW5QR£dS|j ¡|jD]}| ¡q,t d¡d|_W5QR£dSQRXdS)NFzworkers started.T)Ú _conn_lockrAr`rErbrcrd)rÚreaderrrr rEÌs zSqliteQueueDatabase.startc Cs€t d¡|jf|jr&W5QR£dS|j t¡|jD]}|j t¡q8|j ¡|jD]}| ¡qZW5QR£dSQRXdS)Nzenvironment stop requested.FT)rcrerkrAr^rjr(rbrYr`Újoin)rrZrlrrr Ústop×s zSqliteQueueDatabase.stopc Cs$|j|jW5QR£SQRXdSr)rkrAr$rrr Ú is_stoppedäszSqliteQueueDatabase.is_stopped)FFr&NN)NN)NTN)r r rrrBr:rDr_rfrir=rErnroÚ __classcell__rrrJr r4nsÿ! r4c@s4eZdZdZddd„Zdd„Zddd„Zd d „ZdS) rM©rGNcCs ||_dSrrq)rrGrrr rìszThreadHelper.__init__cCstƒSrrr$rrr rïózThreadHelper.eventcCs |dk r|n|j}t|pddS©Nr)Úmaxsize)rGr©rZmax_sizerrr r]ñszThreadHelper.queuecOst|||d}d|_|S)N)r\rHrIT)rÚdaemon)rÚfnrHrIrXrrr rXõszThreadHelper.thread)N)N)r r rr1rrr]rXrrrr rMés rMc@s*eZdZdZdd„Zd dd„Zdd„ZdS) rLrqcCstƒSr)ÚGEventr$rrr rþrrzGreenletHelper.eventNcCs |dk r|n|j}t|pddSrs)rGÚGQueuerurrr r]szGreenletHelper.queuecs‡fdd„}t|f|ž|ŽS)Ncst ¡ˆ||ŽSr)ÚgeventÚsleep)ÚarR©rwrr Úwrapsz#GreenletHelper.thread..wrap)ÚGThread)rrwrHrIr~rr}r rXszGreenletHelper.thread)N)r r rr1rr]rXrrrr rLûs rL)ZloggingÚweakrefÚ threadingrrrÚImportErrorr]rzrrZgevent.eventrxZgevent.queueryZplayhouse.sqlite_extrZ getLoggerrcrhrÚobjectrr9rVr4rMrLrrrr Ús. G{