B>qj #PddlZddlZddlddlmZddlmZddlmZddlmZddlmZddlm Z  dd l m Z dd l m Z dd l m Z e Zd Zd ZGddeZGddeZGddeZGddeZGddeZ ddZGdde eZy#e$r dxZ xZ xZ ZY]wxYw)N)*)_atomic)_manual)EnclosedNodeList)Entity)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__T/opt/rentech/trading_bot/.venv/lib/python3.12/site-packages/playhouse/cockroachdb.pyrrsrrc"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@rrrsN< ?} j$($9&9r)rrr field_typer"r)r*s@rr,r,*sJ::rr,ceZdZejj Zej ddifdZfdZddZ dfd Z dZ fdZ d Z d Zdfd Zdfd Zfd Z ddZ ddZxZS)CockroachDatabaseBLOBBYTEScd|vr7|r5|jds$|jdd|jddtt||g|i|y)Ndsnz postgresql://userrootportif) startswithr r!r1r")r$databaser%r&r's rr"zCockroachDatabase.__init__:sW  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..Is(LAQ(Ls) cursorexecutefetchonerematchtuplegroupsr>server_versionr!r1_set_server_version)r$conncursraw match_objcleanr's rrJz%CockroachDatabase._set_server_versionCs{{{} '(}}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_sqlrD)r$tableschemaqueryrBrows r_get_pk_constraintz$CockroachDatabase._get_pk_constraintOsM,!!%%1C8*7*9:oo~s1v%%rctt| ||}|j||}|Dcgc]}|r|j|k7s|c}Scc}wr=)r!r1 get_indexesrWname)r$rSrTindexespkcidxr's rrYzCockroachDatabase.get_indexesYsK)4 timestamptz)cast)r$rzs 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!r1beginrBrCrgrr$ system_timepriorityr's rrzCockroachDatabase.beginsy ,.  " KKM ! !"H#.. 2  ~~'H88 !GHH KKM ! !"?("J K rc|jr|j|jtkr t |||St t |||Sr=) is_closedconnectrINESTED_TX_MIN_VERSION _crdb_atomicr!r1atomicrs rrzCockroachDatabase.atomicsH >> T\\^   !6 6k8< <&4[(KKrc|jr|j|jtkrt t t t|#Sr=) rrrIrNotImplementedError TXN_ERR_MSGr!r1 savepointr$r's rrzCockroachDatabase.savepoints@ >> T\\^   !6 6%k2 2&799rcfd}|S)NcLtjfd}|S)Nc"tSr=run_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|||||Sr=r)r$rrrrs rrz!CockroachDatabase.run_transactionstR{HMMrr=)NNNNN)rrrPostgresqlDatabase field_typescopyrer"rJrWrYrjrrr{rrrrrrr)r*s@rr1r14s$00557KK E&J ,I(1: LL : @D#'BF!%Nrr1ceZdZfdZxZS)rc|jjdkDr7t|jjtst t tt|'S)Nr) dbtransaction_depthrotop_transactionrrrr!r __enter__rs rrz_crdb_atomic.__enter__sJ 77 $ $ & *dgg557A)+66\4244r)rrrrr)r*s@rrrs 55rrc|xsd}|j||5}|jd|dk7r% ||}|jd|cdddS dddt dd #t$rV}t|jddxst|jdd}|d k(r|d z}|jd Yd}~d}~wwxYw#1swYwxYw) 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_restartNpgcodesqlstate40001z'ROLLBACK TO SAVEPOINT cockroach_restartzunable to commit transaction)rrROperationalErrorgetattrorigr) rcallbackrrrtxnresultexccodes rrrs %2L {X ># 45a !"DE  d$B CC$ (D9<*d;7? A%LNN#LM s5C A++ C 4A C?C CC  C  Cc eZdZy)PooledCockroachDatabaseNrrrrrrsrrr)rrEpeeweerrrrrplayhouse.poolr playhouse.postgres_extr r r JSONField ImportErrorrrrr UUIDFieldr AutoFieldr,rr1rrrrrrrs #4D164I + 2*1<9<::vN*vNr575BF!DD 79J OD?CCJCC=9DsB B%$B%