h8'h#ddlZddlZddlZddlddlmZddlmZddlmZddlmZddlmZddlm Z dd lm Z dd lm Z dd l m Z  dd lmZdd lmZddlmZeZej(ddkDreZdZdZGddeZGddeZGddeZGddeZ GddeZ! ddZ"Gdde e Z#y#e$r dxZxZxZZYqwxYw) N)*)_atomic)_manual)ColumnMetadata)EnclosedNodeList)Entity)ForeignKeyMetadata) IndexMetadata)NodeList)_PooledPostgresqlDatabase) ArrayField)BinaryJSONField) IntervalFieldi aCockroachDB does not support nested transactions. You may alternatively use the @transaction context-manager/decorator, which only wraps the outer-most block in transactional logic. To run a transaction with automatic retries, use the run_transaction() helper.c eZdZy)ExceededMaxAttemptsN__name__ __module__ __qualname__Q/var/www/html/trade_iq/venv/lib/python3.12/site-packages/playhouse/cockroachdb.pyrr#srrc"eZdZdZfdZxZS) UUIDKeyFieldTc|jdrtdt|ztdg|d<|j ddt t ||i|y)N constraints%s cannot specify constraints.zDEFAULT gen_random_uuid() primary_keyT)get ValueErrortypeSQL setdefaultsuperr__init__selfargskwargs __class__s rr&zUUIDKeyField.__init__)s[ ::m $=T JK K!$%@!A B}-. lD*D;F;r)rrrauto_incrementr& __classcell__r+s@rrr&sN< ?} j$($9&9r)rrr field_typer&r-r.s@rr0r01sJ::rr0ceZdZejj Zej ddidZfdZfdZ ddZ dfd Z dZ fd Z d Zd Zdfd Zdfd ZfdZ ddZ ddZxZS)CockroachDatabaseBLOBBYTESTcd|vr7|r5|jds$|jdd|jddtt||g|i|y)Ndsnz postgresql://userrootportif) startswithr$r%r5r&)r(databaser)r*r+s rr&zCockroachDatabase.__init__CsW  H(0(;(;O(L   ff -   fe , /J4J6Jrc*|j}|jd|j\}tjd|}|4dt d|j Dz}t||_ytt|/|y)Nzselect version()z#^CockroachDB.+?v(\d+)\.(\d+)\.(\d+)z %d%02d%02dc32K|]}t|ywN)int).0is r z8CockroachDatabase._set_server_version..Rs(LAQ(Ls) cursorexecutefetchonerematchtuplegroupsrBserver_versionr%r5_set_server_version)r(conncursraw match_objcleanr+s rrNz%CockroachDatabase._set_server_versionLs{{{} '(}}HHCSI   5(L9I9I9K(L#LLE"%e*D  #T >t Drcrd}|j|||xsddf}|j}|xr|dxsdS)NzSELECT constraint_name FROM information_schema.table_constraints WHERE table_name = %s AND table_schema = %s AND constraint_type = %spublicz PRIMARY KEYr) execute_sqlrH)r(tableschemaqueryrFrows r_get_pk_constraintz$CockroachDatabase._get_pk_constraintXsM,!!%%1C8*7*9:oo~s1v%%rctt| ||}|j||}|Dcgc]}|r|j|k7s|c}Scc}wrA)r%r5 get_indexesr[name)r(rWrXindexespkcidxr+s rr]zCockroachDatabase.get_indexesbsK)4F3KCG/456 LL\* +E? " , ,   $ $KL L&=b%HH/4s$$C( c.tj||SrA)fnextract)r( date_part date_fields r extract_datezCockroachDatabase.extract_dateszz)Z00rcB|jdjdS)NrB timestamptz)cast)r(r~s rfrom_timestampz CockroachDatabase.from_timestampsu%**=99rctt| |!|jj d|f|B|j }|dvr t d|jj d|zyy)Nz$SET TRANSACTION AS OF SYSTEM TIME %s)lownormalhighz$priority must be low, normal or highzSET TRANSACTION PRIORITY %s)r%r5beginrFrGrkr!r( system_timepriorityr+s rrzCockroachDatabase.beginsy ,.  " KKM ! !"H#.. 2  ~~'H88 !GHH KKM ! !"?("J K rc|jr|j|jtkr t |||St t |||SrA) is_closedconnectrMNESTED_TX_MIN_VERSION _crdb_atomicr%r5atomicrs rrzCockroachDatabase.atomicsH >> T\\^   !6 6k8< <&4[(KKrc|jr|j|jtkrt t t t|#SrA) rrrMrNotImplementedError TXN_ERR_MSGr%r5 savepointr(r+s rrzCockroachDatabase.savepoints@ >> T\\^   !6 6%k2 2&799rcfd}|S)NcLtjfd}|S)Nc"tSrArun_transaction)cb max_attemptsrr(rsrnew_fnzACockroachDatabase.retry_transaction..deco..new_fns&tR{'/11r) functoolswraps)rrrrr(rs` rdecoz1CockroachDatabase.retry_transaction..decos% __R  1! 1Mrr)r(rrrrs```` rretry_transactionz#CockroachDatabase.retry_transactions   rc t|||||SrAr)r(rrrrs rrz!CockroachDatabase.run_transactionstR{HMMrrA)NNNNN)rrrPostgresqlDatabase field_typescopyrirelease_after_rollbackr&rNr[r]rnrvrrrrrrrr-r.s@rr5r5;s$00557K"K E&J ,I(1: LL : @D#'BF!%Nrr5ceZdZfdZxZS)rc|jjdkDr7t|jjtst t tt|'S)Nr) dbtransaction_depthrstop_transactionrrrr%r __enter__rs rrz_crdb_atomic.__enter__sJ 77 $ $ & *dgg557A)+66\4244r)rrrrr-r.s@rrrs 55rrcz|xsd}|j||5}|jd|dk7r% ||}|jd|cdddS dddt dd #t$r:}|jjdk(r|dz}|jd Yd}~|d}~wwxYw#1swY[xYw) a Run transactional SQL in a transaction with automatic retries. User-provided `callback`: * Must accept one parameter, the `db` instance representing the connection the transaction is running under. * Must not attempt to commit, rollback or otherwise manage transactions. * May be called more than once. * Should ideally only contain SQL operations. Additionally, the database must not have any open transaction at the time this function is called, as CRDB does not support nested transactions. )rrzSAVEPOINT cockroach_restartrz#RELEASE SAVEPOINT cockroach_restartN40001z'ROLLBACK TO SAVEPOINT cockroach_restartzunable to commit transaction)rrVOperationalErrororigpgcoder)rcallbackrrrtxnresultexcs rrrs %2L {X > # 45a !"DE    d$B CC $ 88??g- A%LNN#LM    s4B1A++ B.4/B)#B1(B))B..B11B:c eZdZy)PooledCockroachDatabaseNrrrrrrsrrr)$rrIsyspeeweerrrrrr r r playhouse.poolr playhouse.postgres_extr rr JSONField ImportError version_infostrrtrrrr UUIDFieldr AutoFieldr0rr5rrrrrrrs !#% 4D164IAJ+ 2*1<9<::xN*xNv575BF!D@ 79J UD?CCJCC=9Ds B?? CC