B>qjsddlZddlZddlddlmZddlmZddlmZddlmZddlmZddlmZdd lm Z dd lm Z dd lm Z dd l m Zd ZdZdZdZGddeZGddeZGddeZGddeZGddeZGddeZGddeZGdd eZGd!d"eZGd#d$e Z!Gd%d&e"Z#Gd'd(e$Z%Gd)d*e%Z&Gd+d,e&Z'd-Z(e)d.e)e(ze)e(jUze)e+d/fzZ,e)d0e-d1DZ.[([,ej^d2Z0Gd3d4e&Z1d5e _2d6Z3y)7N)*) ColumnBase)EnclosedNodeList)Entity) Expression)Node)NodeList)OP) merge_dict)sqlite3) fts_parserpcxpcnalx) rc.eZdZdZdxZxZZfdZxZS) RowIDFieldTrowidc||jk7r$tt|d|jdtt|||g|y)Nz must be named "z".) required_name ValueErrortypesuperrbind)selfmodelnameargs __class__s S/opt/rentech/trading_bot/.venv/lib/python3.12/site-packages/playhouse/sqlite_ext.pyrzRowIDField.bindsH 4%% %"4j$*<*<>? ? j$$UD848) __name__ __module__ __qualname__auto_increment column_namerrr __classcell__r s@r!rrs!N)00K0$99r"rceZdZdxZxZZy) DocIDFielddocidN)r#r$r%r'rrr"r!r+r+"s)00K0$r"r+ceZdZfdZxZS)AutoIncrementFieldcVtt| |}t|t dfS)N AUTOINCREMENT)rr/ddlr SQL)rctx node_listr s r!r2zAutoIncrementField.ddl's*,d7< C$89::r")r#r$r%r2r(r)s@r!r/r/&s ;;r"r/ceZdZdZdZy) TDecimalFieldTEXTcyNr-rs r! get_modifierszTDecimalField.get_modifiers.sr"N)r#r$r% field_typer<r-r"r!r7r7,s J!r"r7ceZdZgdZdZy)ISODateTimeField)z%Y-%m-%dT%H:%M:%S.%f%zz%Y-%m-%dT%H:%M:%S%zz%Y-%m-%dT%H:%M:%S.%fz%Y-%m-%dT%H:%M:%Sz%Y-%m-%dct|tr|j|}|r t|ts|jS|Sr:) isinstancestradapt isoformatrvalues r!db_valuezISODateTimeField.db_value:s8 eS !JJu%E E3/??$ $ r"N)r#r$r%formatsrGr-r"r!r?r?1sGr"r?ceZdZdfd ZdZedZdZddZddZ ddZ ddZ dd Z d Z d Zd Zd ZdZdZdZxZS)JSONPathcNtt| ||_|xsd|_yNr-)rrJ__init___field_path)rfieldpathr s r!rMzJSONPath.__init__Ds" h&( ZR r"c8|jj|Sr:)rN python_valuerEs r! _converterzJSONPath._converterIs{{''..r"cPtddj|jzS)Nz$%s)ValuejoinrOr;s r!rQz JSONPath.pathLsURWWTZZ0011r"ct|ts|dk(rd|z}nd|jddz}t||j|j |fzS)N#z[%s]z."%s""z"")rAintreplacerrNrO)ridxitems r! __getitem__zJSONPath.__getitem__PsR c3 3#:C>r"c|st|ttfr.tj |j j |}||j |j|Sr:)rArcrdrerfrNrgrQrfuncrFrjs r!_json_operationzJSONPath._json_operation\sF jt 5GGDKK33E:;EDKKE22r"cD|jtj||Sr:)rore json_insertris r!insertzJSONPath.insertas##BNNE7CCr"cD|jtj||Sr:)rorerhris r!setz JSONPath.setds##BKK@@r"cD|jtj||Sr:)rore json_replaceris r!r]zJSONPath.replaceg##BOOUGDDr"c~|jtj||jj |Sr:)rtre json_patchrNrgrEs r!updatezJSONPath.updatejs+xx dDKK,C,CE,JKLLr"cVtj|j|jSr:)re json_removerNrQr;s r!removezJSONPath.removems~~dkk49955r"cVtj|j|jSr:)re json_typerNrQr;s r!rzJSONPath.json_typep||DKK33r"cVtj|j|jSr:)rejson_array_lengthrNrQr;s r!lengthzJSONPath.lengthss##DKK;;r"cVtj|j|jSr:)re json_eachrNrQr;s r!childrenzJSONPath.childrenvrr"cVtj|j|jSr:)re json_treerNrQr;s r!treez JSONPath.treeyrr"c|j|jr.tj|j|j S|jSr:)sqlrOre json_extractrNrQrr4s r!__sql__zJSONPath.__sql__|sCww**t{{DII>7 7*.++7 7r"r:)r#r$r%rMrTpropertyrQr`rkrorrrtr]rzr}rrrrrr(r)s@r!rJrJCsf /22=? 3 DAEM64<447r"rJcFeZdZd dZd dZd dZd dZd dZdZdZ d Z y) JSONBPathNc|st|ttfr.tj |j j |}tj|j |dj|Srb) rArcrdrejsonbrNrg jsonb_setrQris r!rkzJSONBPath.appendsN jt 5HHT[[44U;   ''.. z*   s **ct|6t|ts$tj|j |}|Syr:)rArrerfrgrEs r!rGzJSONField.db_values5  eT* 0 0 78L r"cfd}|S)Ncvt|ttfrt||j}t ||Sr:)rArcrdAsIsrGr)rrhsops r!innerzJSONField._e..inners0#d|,3 .dB, ,r"r-)rrs` r!_ez JSONField._es - r"c*|j||Sr:)Path)rr_s r!r`zJSONField.__getitem__syyt$$r"cl|Dcgc]}t|d}}tj|g|Scc}wNF converter)rWrerrpathsps r!extractzJSONField.extracts549:qqE*::t,e,,;1c2t|dt|dS)Nz->FrrrWrrQs r! extract_jsonzJSONField.extract_jsons$eDE&BCCr"c2t|dt|dS)Nz->>Frrrs r! extract_textzJSONField.extract_texts$uTU'CDDr"cD|j|j||Sr:)rrkris r!rkzJSONField.appendyy%%eW55r"cD|j|j||Sr:)rrrris r!rrzJSONField.insertrr"cD|j|j||Sr:)rrtris r!rtz JSONField.setsyy""5'22r"cD|j|j||Sr:)rr]ris r!r]zJSONField.replacesyy&&ug66r"cB|j|j|Sr:)rrz)rdatas r!rzzJSONField.updatesyy%%d++r"cp|s|j|jStj|g|Sr:)rr}rer|rrs r!r}zJSONField.removes099T?))+ +~~d+U++r"c,tj|Sr:)rerr;s r!rzJSONField.json_type||D!!r"c8|r||fn|f}tj|Sr:)rer)rrQrs r!rzJSONField.lengths"#d|$##T**r"c,tj|S)a? Schema of `json_each` and `json_tree`: key, value, type TEXT (object, array, string, etc), atom (value for primitive/scalar types, NULL for array and object) id INTEGER (unique identifier for element) parent INTEGER (unique identifier of parent element or NULL) fullkey TEXT (full path describing element) path TEXT (path to the container of the current element) json JSON hidden (1st input parameter to function) root TEXT hidden (2nd input parameter, path at which to start) )rerr;s r!rzJSONField.childrens||D!!r"c,tj|Sr:)rerr;s r!rzJSONField.treerr")NNr:)(r#r$r%r=rJrrMrSrGrr EQ__eq__NE__ne__GT__gt__GTE__ge__LT__lt__LTE__le__Field__hash__r`rrrrkrrrtr]rzr}rrrrr(r)s@r!rrsJ D2   YF YF YF ZF YF ZF~~H%-DE6637,, "+"""r"rc,eZdZdZeZdZdZdZdZ y) JSONBFieldJSONBct|6t|ts$tj|j |}|Syr:)rArrerrgrEs r!rGzJSONBField.db_values5  eT*!1!1%!89L r"c,tj|Sr:)rerfr;s r!rfzJSONBField.jsonswwt}r"cl|Dcgc]}t|d}}tj|g|Scc}wr)rWrerrs r!rzJSONBField.extract s749:qqE*::-u--;rcp|s|j|jStj|g|Sr:)rr}rerrs r!r}zJSONBField.removes099T?))+ +t,e,,r"N) r#r$r%r=rrrGrfrr}r-r"r!rrs J D .-r"rcDeZdZdfd ZdZedZdZddZxZ S) SearchFieldc d|rtdt|ztt|||dy)Nz8SearchField does not accept these keyword arguments: %s.T) unindexedr'null)rsortedrrrM)rrr'kr s r!rMzSearchField.__init__s@ .06q :; ; k4)I6A * Nr"ct||Sr:)match)rterms r!rzSearchField.matchsT4  r"c t|dsg|jjjDcgc]}t |t r |j  }}|j|j |_|jScc}w)N_fts_column_index) hasattrr_meta sorted_fieldsrArrindexr)rf search_fieldss r!fts_column_indexzSearchField.fts_column_index sot01-1ZZ-=-=-K-K< *1k :VV>   dnn=c!f= >    T::7C Dww' 233!E>s I  I%c t|jtr|j|fi|St t ||fi|Sr:) issubclassr VirtualModelrrr _create_table)rrr r s r!rz'VirtualTableSchemaManager._create_table`sI djj, /-4--d>g> >.C  r"T)r#r$r%rrr(r)s@r!rr4s)4Vr"rc0eZdZGddZedZy)rc eZdZdZdZdZdZeZy)VirtualModel.MetaNF) r#r$r%rrr primary_keyrschema_manager_classr-r"r!Metar#is  8r"r&c|Sr:r-)clsr s r!r zVirtualModel.clean_optionspsr"N)r#r$r%r& classmethodr r-r"r!rrhs 99r"rceZdZedZy) BaseFTSModelc*|jd}|jd}|jd}t|tr |dk(rd|d<nt|tr t d|rqt|t t fr)dj|Dcgc] }t|c}}nt|tr t|}d|jd z|d<|r/|jjjd k(rd |z|d<|Scc}w) NcontentprefixtokenizerVz''z[The content option must be a Model, a table-name string, or the empty string (contentless).,z'%s'z' fts5z"%s") getrArBrImproperlyConfiguredrctuplerXr\striprrlower)r(r r-r.r/is r!r zBaseFTSModel.clean_optionsvs++i(X&;;z* gs #2 !%GI   '&(?@ @ &4-06":a3q6":;FC(V &d); ;GH   2288:fD#)8"3GJ #;sDN)r#r$r%r)r r-r"r!r+r+usr"r+cNeZdZdZeZGddZedZedZ edZ edZ eddZ edd Z ed Zed Zed Zed ZedZedZe ddZe ddZe ddZe ddZy)FTSModelz VirtualModel class for creating tables that use the FTS4 search extension. To use FTS3, set ``Meta.extension_module = 'FTS3'``. ceZdZdezZy) FTSModel.MetazFTS%sN)r#r$r% FTS_VERSIONrr-r"r!r&r;s "[0r"r&c |jj}|jjjd|d|d|d}|j S)Nz INSERT INTO (z ) VALUES('z');)r table_namedatabase execute_sqlfetchone)r(cmdtblress r!_fts_cmdzFTSModel._fts_cmdsBii""ii  ,,25sC @B||~r"c$|jdSNoptimizerFr(s r!rIzFTSModel.optimize||J''r"c$|jdSNrebuildrJrKs r!rOzFTSModel.rebuild||I&&r"c$|jdS)Nintegrity-checkrJrKs r!integrity_checkzFTSModel.integrity_checks||-..r"c0|jd|d|S)Nzmerge=r0rJ)r(blockssegmentss r!mergezFTSModel.merges||VX>??r"c:|jd|xrdxsdzS)Nz automerge=%s10rJ)r(states r! automergezFTSModel.automerges||Nem.BsCDDr"cBt|jj|SzU Generate a `MATCH` expression appropriate for searching this table. rrrr(rs r!rzFTSModel.match SYY%%t,,r"ctj|jjt}tj |g|Sr:)re matchinforrFTS3_MATCHINFOfts_rank)r(weightsrcs r!rankz FTSModel.ranks0LL!1!1>B {{9/w//r"ctj|jjt}tj |g|Sr:)rercrrFTS4_MATCHINFOfts_bm25r(rf match_infos r!bm25z FTSModel.bm25s0\\#))"2"2NC {{:000r"ctj|jjt}tj |g|Sr:)rercrrri fts_bm25frks r!bm25fzFTSModel.bm25fs0\\#))"2"2NC ||J111r"ctj|jjt}tj |g|Sr:)rercrrri fts_lucenerks r!lucenezFTSModel.lucenes0\\#))"2"2NC }}Z2'22r"c|s|}nt|trqg}|jjD]P} t| tr|j | |j | j d} |j| R||}n||}d} |} |r||j|f} |r |s t|} |j| j|j|j| S)N?r-)rArdrrrr2rrkaliasr3selectwhererorder_by) r(rrf with_score score_aliasscore_fnexplicit_orderingrg weight_argsrP field_weight selectionrys r!_searchzFTSModel._searchs:D  &K00 1eZ0 '{{5'++ejj#2NO ""<0 1[)DW%D  djj56I /;'H$syy'(# %r"NcB|j|||||j|S'Full-text search using selected `term`.)rrgr(rrfrzr{r}s r!searchzFTSModel.search+{{     HH   r"cB|j|||||j|Sz:Full-text search for selected `term` using BM25 algorithm.)rrmrs r! search_bm25zFTSModel.search_bm25rr"cB|j|||||j|Sr)rrprs r! search_bm25fzFTSModel.search_bm25f s+{{     II   r"cB|j|||||j|Sr)rrsrs r! search_lucenezFTSModel.search_lucenes+{{     JJ   r")r NFscoreF)r#r$r%__doc__rrr&r)rFrIrOrSrWr\rrgrmrprsrrrrrr-r"r!r9r9sv LE11 ((''//@@EE-- 00112233%%<FM!&  8=;@  9>r)sK3q63JSVKs))z [^\s"]+|"[^"\\]*(?:\\.[^"\\]*)*"ceZdZdZeZGddZddddZedZ ed Z e d Z e e d fd Zed ZedZedZedZe d dZe d dZedZedZedZedZedZedZedZedZedZedZed!dZed"dZ y)# FTS5Modela Requires SQLite >= 3.9.0. Table options: content: table name of external content, or empty string for "contentless" content_rowid: column name of external content primary key prefix: integer(s). Ex: '2' or '2 3 4' tokenize: porter, unicode61, ascii. Ex: 'porter unicode61' The unicode tokenizer supports the following parameters: * remove_diacritics (1 or 0, default is 1) * tokenchars (string of characters, e.g. '-_' * separators (string of characters) Parameters are passed as alternating parameter name and value, so: {'tokenize': "unicode61 remove_diacritics 0 tokenchars '-_'"} Content-less tables: If you don't need the full-text content in its original form, you can specify a content-less table. Searches and auxiliary functions will work as usual, but the only values returned when SELECT-ing can be rowid. Also content-less tables do not support UPDATE or DELETE, unless created with the contentless_delete=1 option (sqlite >= 3.43). External content tables: You can set up triggers to sync these, e.g. -- Create a table. And an external content fts5 table to index it. CREATE TABLE tbl(a INTEGER PRIMARY KEY, b); CREATE VIRTUAL TABLE ft USING fts5(b, content='tbl', content_rowid='a'); -- Triggers to keep the FTS index up to date. CREATE TRIGGER tbl_ai AFTER INSERT ON tbl BEGIN INSERT INTO ft(rowid, b) VALUES (new.a, new.b); END; CREATE TRIGGER tbl_ad AFTER DELETE ON tbl BEGIN INSERT INTO ft(ft, rowid, b) VALUES('delete', old.a, old.b); END; CREATE TRIGGER tbl_au AFTER UPDATE ON tbl BEGIN INSERT INTO ft(ft, rowid, b) VALUES('delete', old.a, old.b); INSERT INTO ft(rowid, b) VALUES (new.a, new.b); END; Note: with sqlite's default recursive_triggers=off, INSERT OR REPLACE on the content table skips the delete trigger and corrupts the index. Enable PRAGMA recursive_triggers if REPLACE may be used. Built-in auxiliary functions: * bm25(tbl[, weight_0, ... weight_n]) * highlight(tbl, col_idx, prefix, suffix) * snippet(tbl, col_idx, prefix, suffix, ?, max_tokens) ceZdZdZy)FTS5Model.Metar1N)r#r$r%rr-r"r!r&rns!r"r&zQBesides the implicit `rowid` column, all columns must be instances of SearchFieldz3Secondary indexes are not supported for FTS5 modelsz4FTS5 models must use the default `rowid` primary key)r=rpkc|jjjdk7rt|jd|jj j D]0}t|ttfrt|jd|jjrt|jdy)Nrrr=r) rr$rr3_error_messagesfieldsvaluesrArrindexes)r(rPs r!validate_modelzFTS5Model.validate_modelxs 99 % % 0&s':':4'@A AYY%%,,. NEek:%>?*3+>+>|+LMM N 99  &s':':7'CD D r"ctjddtkrytjd} |j d|jy# |j d|j d|jjj dn#YY|jyxYwYvxYw#|jwxYw)NrFz:memory:z0CREATE VIRTUAL TABLE fts5test USING fts5 (data);Tr1) r sqlite_version_infoFTS5_MIN_SQLITE_VERSIONconnectexecuteenable_load_extensionload_extensionrr@close)r(tmp_dbs r!fts5_installedzFTS5Model.fts5_installeds  & &r *-D D,  NNM N LLN : :,,T2%%f- ""11&9  LLN: LLNs;AB9"B9&B9B4!B9"B<4B97B<<Cctj|}|D]9}|jdr|jdr&t |t zs9yy)z Simple helper function to indicate whether a search query is a valid FTS5 query. Note: this simply looks at the characters being used, and is not guaranteed to catch all problematic queries. r[FT) _quote_refindall startswithendswithrt_invalid_ascii)querytokenstokens r!validate_queryzFTS5Model.validate_querysS""5) E$)<5zN*   r"rcXg}d}tj|}|D]x}|jdr#|jdr|j |7t |}|t z}|rd}|D]}|j||}|j |z|rdj|S|S)z2 Clean a query of invalid tokens. Fr[Tr ) rrrrrkrtrr]rX) rr]accum any_invalidrr token_setinvalid_for_tokencs r! clean_queryzFTS5Model.clean_querys  ""5) E$)< U#E I )N :  " *6A!MM!W5E6 LL   88E? " r"c|jjDcgc]*}t|tr|js |j ,}}t j||Scc}w)z Translate a search-box query (quoted phrases, AND/OR/NOT, -exclusion, column:filters, parentheses) into an FTS5 query. )rrrArrr' _fts_parserparse)r(rrcolumnss r! web_queryzFTS5Model.web_querysZ +.))*A*AFQ K0==FF  00Fs/A cBt|jj|Sr^r_r`s r!rzFTS5Model.matchrar"c:|r|j|StdS)Nrg)rmr3)r(rs r!rgzFTS5Model.ranks"&xsxx7CK7r"cVtj|jjg|Sr:)rermrr)r(rfs r!rmzFTS5Model.bm25s wwsyy''2'22r"NcR|jtj|||||Sr)rrrrs r!rzFTS5Model.searchs/  ! !$ '      r"c |s td}nt|trg}|jjD]N}t|t s|j |j||j|jdPtj|jjg|}n*tj|jjg|}d} |} |r||j|f} |r |s t|} |j| j|jt j#|j%| S)rrgrur-)r3rArdrrrrkr2rrermrrvrwrxrrrry) r(rrfrzr{r}rgr~rPrrys r!rzFTS5Model.search_bm25sv;D  &K00 Je[1&& E7;;uzz3+GHJ  J 77399++:k:D77399++6g6D  djj56I /;'H$syy!6!6t!<=>(# %r"c P|jj}|g}|g}|jD]0\}}|jt ||j|2t t d|jjt|t dt|fS)Nz INSERT INTOVALUES)rritemsrkrr r3r)r(rC extra_paramsrDrrkeyrFs r! _fts_cmd_sqlzFTS5Model._fts_cmd_sqlsii%&,,. !JC NN6#; ' MM%  !   II   W % M V $ &' 'r"c r|j|fi|}|jjj|Sr:)rrr@r)r(rCrrs r!rFzFTS5Model._fts_cmds4   5 5yy!!))%00r"cld|cxkrdkstdtd|jd|S)Nr@zlevel must be between 0 and 64r\rg)rrF)r(levels r!r\zFTS5Model.automerges?U b => >!=> >||Ke|44r"c(|jd|S)NrWrrJ)r(npagess r!rWzFTS5Model.merges||G&|11r"c$|jdSrHrJrKs r!rIzFTS5Model.optimize!rLr"c$|jdSrNrJrKs r!rOzFTS5Model.rebuild%rPr"c(|jd|S)NpgszrrJ)r(rs r!set_pgszzFTS5Model.set_pgsz)s||F|..r"c(|jd|S)NrgrrJ)r(rank_expressions r!set_rankzFTS5Model.set_rank-s||F|99r"c$|jdS)Nz delete-allrJrKs r! delete_allzFTS5Model.delete_all1s||L))r"c i}|jjD]}t|ts|j|vr)|j |j||j <K|j |vr)|j |j ||j <|jrtd|jz|r&tddjt|z|jdd|i|S)a Remove a row using the fts5 'delete' command, passing the values exactly as they were indexed. This is how rows are removed from external-content and contentless tables, which cannot look up the old values themselves. zIdelete_command() requires the value of "%s" as it was originally indexed.z4delete_command() received unrecognized column(s): %sz, r)delete) rrrArrpopr'rrrXrrF)r(rrrrPs r!delete_commandzFTS5Model.delete_command5sYY,, OEe[1zzV#+1::ejj+Ae''(""f,+1::e6G6G+He''(__!"ACH::"NOO O "$(IIfVn$=>? ?s||;E;U;;r"c(|jd|S)NrRrrJ)r(rgs r!rSzFTS5Model.integrity_checkOs||-D|99r"c:dvr tddk(rdndzGfdd}dk(rd}n d k(rd }nd }|Dcic]}||d vr tn t}}t|d <||d<t dj zt f|Scc}w)N)rowcolinstancez5table_type must be either "row", "col" or "instance".r_vz_v_%sceZdZWjjZWxsWjj WzZejWjje WZ y)"FTS5Model.VocabModel..MetaN) r#r$r%rr@r?re fts5vocabrr3r)r(suffixtable table_typesr!r&r]sNyy))H?#))"6"6"?J!||   J ! r"r&)rdoccntr)rrrr)rrroffset)rrrz%sVocab)r TextField IntegerFieldrrr#r)r(rrr&rrattrsrs``` @r! VocabModelzFTS5Model.VocabModelSs 7 7+, , $u,'J2F ! !  ,G 5 3G6G"#1#7IK\^K###gf I ,|ouEE#s Br)r)rN)!r#r$r%rrrr&rr)rr staticmethodrrrrrrgrmrrrrFr\rWrIrOrrrrrSrr-r"r!rr/s9x LE""5FD OEE*  #&r7.11-- 8833FM!&8=;@%%: ' '1155 22((''//::**<<2::FFr"rMATCHc8t|tj|Sr:)rr r)lhsrs r!rrws c288S ))r")4rfrepeeweerrrrrr r r r playhouser rrdrir<r AutoFieldrr+r/ DecimalFieldr7 DateTimeFieldr?rJrrrrrr SchemaManagerrModelrr+r9 _alphabetrtupperrrrangercompilerrrrr-r"r!r s #/ #9911;; "L" }$;7z;7z77>\" \"~--.3%3>1 1h 5 <>M|M` ) * + ^ " #$#b'_ KU3ZKK  BJJ: ; CF CFL *r"