]j# d dl Z d dlZd dlZd dlT d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d d lm
Z
d d
lm Z d d l m
Z
d d lmZ d d
lmZ d dlmZ eZn# e$ r dxZxZxZZY nw xY wej d dk reZdZdZ G d de Z G d de Z G d de Z G d de Z G d de Z! d d Z" G d d e
e Z#dS ) N)*)_atomic)_manual)ColumnMetadata)EnclosedNodeList)Entity)ForeignKeyMetadata)
IndexMetadata)NodeList)_PooledPostgresqlDatabase)
ArrayField)BinaryJSONField)
IntervalField i
a CockroachDB 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 e Zd ZdS )ExceededMaxAttemptsN__name__
__module__ __qualname__ L/opt/imunify360/venv/lib64/python3.11/site-packages/playhouse/cockroachdb.pyr r # s r r c " e Zd ZdZ fdZ xZS ) UUIDKeyFieldTc | d r t dt | z t d g|d< | dd t t | j |i | d S )N constraints %s cannot specify constraints.zDEFAULT gen_random_uuid() primary_keyT)get
ValueErrortypeSQL
setdefaultsuperr __init__selfargskwargs __class__s r r& zUUIDKeyField.__init__) s ::m $ $ L =T
JKK K!$%@!A!A B} -...*lD!!*D;F;;;;;r )r r r auto_incrementr&
__classcell__r+ s @r r r & s= N< < < < < < < < <r r c " e Zd ZdZ fdZ xZS )
RowIDFieldINTc | d r t dt | z t d g|d< t t
| j |i | d S )Nr r zDEFAULT unique_rowid())r r! r" r# r% r0 r& r' s r r& zRowIDField.__init__4 sp ::m $ $ L =T
JKK K!$%=!>!> ?} (j$ ($9&99999r )r r r
field_typer& r- r. s @r r0 r0 1 s= J: : : : : : : : :r r0 c e Zd Zej Ze ddi dZ fdZ fdZ ddZ
d fd Z d Z fd
Z
d Zd Zd fd
Zd fd Z fdZ ddZ ddZ xZS )CockroachDatabaseBLOBBYTESTc d|vrC|rA| d s,| dd | dd t t | j |g|R i | d S )Ndsnz
postgresql://userrootportif )
startswithr$ r% r5 r& )r( databaser) r* r+ s r r& zCockroachDatabase.__init__C s H (0(;(;O(L(L ff - - - fe , , ,/ &&/J4JJJ6JJJJJr c | }| d | \ }t j d| }|Ddt d | D z }t | | _ d S t t | | d S )Nzselect version()z#^CockroachDB.+?v(\d+)\.(\d+)\.(\d+)z
%d%02d%02dc 3 4 K | ]}t | V d S N)int).0is r <genexpr>z8CockroachDatabase._set_server_version.<locals>.<genexpr>R s( (L(LAQ(L(L(L(L(L(Lr ) cursorexecutefetchonerematchtuplegroupsrB server_versionr% r5 _set_server_version)r( conncursraw match_objcleanr+ s r rN z%CockroachDatabase._set_server_versionL s {{}} '(((}}H CSII 5(L(L9I9I9K9K(L(L(L#L#LLE"%e**D
#T * * > >t D D D D Dr Nc | d}| |||pddf }| }|r|d pd S )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( tableschemaqueryrF rows r _get_pk_constraintz$CockroachDatabase._get_pk_constraintX sU , !!%%1C8*7*9 : :oo ~s1v%%r c t t | || }| || fd|D S )Nc . g | ]}r |j k |S r )name)rC idxpkcs r
<listcomp>z1CockroachDatabase.get_indexes.<locals>.<listcomp>g s% IIIsICr )r% r5 get_indexesr[ )r( rW rX indexesr` r+ s @r rb z CockroachDatabase.get_indexesb sU )400<<UFKK%%eV44IIIIwIIIIr c |j sd S |j }|dv rt d S |dvrt d d S )NreplaceupsertUPSERT)ignorenothingupdatezfUn-supported action for conflict resolution. CockroachDB supports REPLACE (UPSERT), IGNORE and UPDATE.)_actionlowerr# r! )r( on_conflictrY actions r conflict_statementz$CockroachDatabase.conflict_statementi sf "*FF $**,, * * *x==
:
:
: + , ,
, ;
:r c |j r|j nd}|dv ryt d g}|j r1| t d |j D | t d t
| S |dv rd S |j rt d t t | || S )N )ri rj z ON CONFLICTc Z g | ](}t |t rt | n|)S r )
isinstance
basestringr )rC cols r ra z5CockroachDatabase.conflict_update.<locals>.<listcomp>y sA /4 /4 /4 $.c:#>#>GF3KKKC/4 /4 /4r z
DO NOTHINGre zWCockroachDB does not support the usage of a constraint name. Use the column(s) instead.) rl rm r# _conflict_targetappendr r _conflict_constraintr! r% r5 conflict_update)r( ocrY ro partsr+ s r rz z!CockroachDatabase.conflict_updatet s ')z9!!###r * * * ''(E"
6 - /4 /4 ! 2/4 /4 /4 5 5 6 6 6
LL\** + + +E?? "
,
,
,
F
$ L K L L
L &--==b%HHHr c 8 t || S rA )fnextract)r( date_part
date_fields r extract_datez CockroachDatabase.extract_date s zz)Z000r c R | d d S )NrB timestamptz)cast)r( r s r from_timestampz CockroachDatabase.from_timestamp s$ u%%**=999r c V t t | |)| d|f |S| }|dvrt
d | d|z d S d S )Nz$SET TRANSACTION AS OF SYSTEM TIME %s)lownormalhighz$priority must be low, normal or highzSET TRANSACTION PRIORITY %s)r% r5 beginrF rG rm r! r( system_timepriorityr+ s r r zCockroachDatabase.begin s
&&,,... " KKMM ! !"H#..
2
2
2 ~~''H 888 !GHHH KKMM ! !"?("J K K K K K r c | r| | j t k rt | || S t t | || S rA ) is_closedconnectrM NESTED_TX_MIN_VERSION _crdb_atomicr% r5 atomicr s r r zCockroachDatabase.atomic sa >> +T\\^^^ !6 6 6 k8<< <&--44[(KKKr c | r| | j t k rt t
t
t | S rA ) r r rM r NotImplementedError TXN_ERR_MSGr% r5 savepointr( r+ s r r zCockroachDatabase.savepoint sY >> +T\\^^^ !6 6 6%k22 2&--77999r c fd}|S )Nc P t j fd }|S )Nc * t S rA run_transaction)cb max_attemptsr r( r s r new_fnzACockroachDatabase.retry_transaction.<locals>.deco.<locals>.new_fn s &tR {'/1 1 1r ) functoolswraps)r r r r r( r s ` r decoz1CockroachDatabase.retry_transaction.<locals>.deco sM
_R
1
1
1
1
1
1
1
1 !
1 Mr r )r( r r r r s ```` r retry_transactionz#CockroachDatabase.retry_transaction s5 r c ( t | |||| S rA r )r( r r r r s r r z!CockroachDatabase.run_transaction s tR {HMMMr rA )NNNNN)r r r PostgresqlDatabase field_typescopyrk release_after_rollbackr& rN r[ rb rp rz r r r r r r r r- r. s @r r5 r5 ; s $05577K "K K K K K
E
E
E
E
E& & & &J J J J J J , , ,I I I I I(1 1 1: : :
L L L L L LL L L L L L : : : : : @D#' BF!%N N N N N N N Nr r5 c e Zd Z fdZ xZS )r c | j dk r@t | j t st t t t | S )Nr )
dbtransaction_depthrt top_transactionr r r r% r __enter__r s r r z_crdb_atomic.__enter__ sf 7 $ $ & & * * dg 5 5 7 7AA
7)+666\4((22444r )r r r r r- r. s @r r r s8 5 5 5 5 5 5 5 5 5r r c |pd}| || 5 }| d |dk rq || }| d |cddd S # t $ r5}|j j dk r |dz }| d Y d}~n d}~ww xY w ddd n # 1 swxY w Y t dd
) 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.
)r r zSAVEPOINT cockroach_restartr z#RELEASE SAVEPOINT cockroach_restartN40001 z'ROLLBACK TO SAVEPOINT cockroach_restartz unable to commit transaction)r rV OperationalErrororigpgcoder )r callbackr r r txnresultexcs r r r sW %2L {X > > #
4555a
!" DEEE
$
8?g-- A%LNN#LMMM HHHH
d$B
C
CCs: B6!A''
B&1*B!B6 B!!B&&B66B:=B:c e Zd ZdS )PooledCockroachDatabaseNr r r r r r s Dr r r )$r rI syspeeweer r r r r r r
r playhouse.poolr playhouse.postgres_extr
r r JSONField ImportError version_infostrru r r r r UUIDFieldr AutoFieldr0 r r5 r r r r r r <module>r s
! ! ! ! ! ! # # # # # # % % % % % % 4 4 4 4 4 4D111111666666444444 II D D D?CCJC C=999D AJ + 2 1 1 1 1* 1 1 1< < < < <9 < < <: : : : : : : :xN xN xN xN xN* xN xN xNv5 5 5 5 57 5 5 5 BF ! D D D D@ 79J s A
A-,A-