iy d dl mZ n
# e$ r eZY nw xY wd dl mZ d dlmZ d dlZd dlZd dl T d dl m
Z
d dl m Z d dl m Z d d l m
Z
d d
l mZ d d l mZ d d lmZ n # e$ r d d lmZ n
# e$ r dZY nw xY wY nw xY w d d
lmZ n
# e$ r dZY nw xY w d dlmZ n
# e$ r dZY nw xY w eg d Z G d de Z G d de Z G d de Z G d de Z G d de Z G d de Z G d d e Z ed d Z! G d d!e! Z" G d" d#e Z#d+d$Z$d+d%Z%d,d(Z&d) Z'd* Z(dS )- ) OrderedDict)
namedtuple)isclassN)*) _StringField)_query_val_transform)
CommaNodeList) SCOPE_VALUES)make_snake_case) text_type)
FIELD_TYPE) postgres_ext)CockroachDatabase) andasassertbreakclasscontinuedefdelelifelseexceptexecfinallyforfromglobalifimportinislambdanotorpassprintraisereturntrywhilewithyieldc e Zd ZdS ) UnknownFieldN)__name__
__module__ __qualname__ K/opt/imunify360/venv/lib64/python3.11/site-packages/playhouse/reflection.pyr0 r0 ) s Dr5 r0 c V e Zd ZdZeefZ d dZd Zd Z d Z
d Z d Z d
d
Z
d ZdS )Columnz1
Store metadata about a database column.
FNc || _ || _ || _ || _ || _ || _ || _ || _ | | _ |
| _ d | _
d | _ d | _ d S N)
name field_classraw_column_typenullable primary_key column_nameindexuniquedefaultextra_parameters rel_model related_nameto_field) selfr; r< r= r> r? r@ rA rB rC rD s r6 __init__zColumn.__init__3 sk & .
&&
0
r5 c j g d}d fd|D }d j d|dS )N)r< r= r> r? r@ , c 3 B K | ]}|d t | V dS =N)getattr).0attrrH s r6 z"Column.__repr__..M sM ! ! ttWT4 0 0 0 1! ! ! ! ! ! r5 zColumn())joinr; )rH attrs keyword_argss ` r6 __repr__zColumn.__repr__F sh yy ! ! ! ! ! ! ! $(999lll;;r5 c d i }| j | | j | j rd|d< | j t u s| j | j k r
d| j z |d< | j r t | j t sd|d< | j
!d| j
dd z |d < | r2| j
|d
< | j r
d| j z |d < | j r
d| j z |d < | s-| j rd
|d< n | j r| sd
|d< |S )NTnull'%s'r@ r? z[SQL("DEFAULT %s")]"z\" constraintsmodelfieldbackrefTruerB rA )rD updater> r< ForeignKeyFieldr; r@ r?
issubclass AutoFieldrC replaceis_foreign_keyrE rG rF is_primary_keyrB rA )rH paramss r6 get_field_parameterszColumn.get_field_parametersR sn , MM$/ 0 0 0 = " !F6N . .$)t?O2O2O$*T-=$=F= ! )Jt/?$K$K )$(F= ! < #$9L((e44%5F= ! ? "nF7O}
9"(4="8w
?$*T->$>y! ""$$ ){
)#)x
)D$7$7$9$9
)"(w
r5 c , | j t u p| j S r: )r< rd r? rH s r6 rg zColumn.is_primary_keys s 9,@0@@r5 c | j t u S r: )r< rb rk s r6 rf zColumn.is_foreign_keyv s ?22r5 c 4 | j t u o
| j dk S Nz'self')r< rb rE rk s r6 is_self_referential_fkz Column.is_self_referential_fky s! O3 +(* ,r5 c || _ t | _ |j |j k rd| _ n||j | _ |r|j pd | _ |pd | _ d S rn ) foreign_keyrb r<
dest_tabletablerE r; rG rF )rH rq model_namesdestrF s r6 set_foreign_keyzColumn.set_foreign_key} sa &* ![%6 6 6 %DNN ()? @DN *2d
( 0Dr5 c i }| D ]5\ }}t | r t |t r|j }|||< 6d d t | D }| j d| j j d|d}| j t u r |d| j }|S )NrK c 3 * K | ]\ }}|d |V dS rM r4 )rP kvs r6 rR z#Column.get_field.. sM H H"&!Q *+AA . H H H H H Hr5 z = (rS z # ) ri itemsr rc Fieldr1 rT sortedr; r< r0 r= )rH field_paramskeyvalue param_strr^ s r6 get_fieldzColumn.get_field s 3355;;== & &JCu~~
'*UE":":
' %L II H H*0 1C1C1E1E*F*F H H H H H
III % % % II
| + +"'%%)=)=>E r5 )FNFFNN)NN)r1 r2 r3 __doc__ IntegerFieldrd primary_key_typesrI rW ri rg rf ro rv r r4 r5 r6 r8 r8 - s &y1 =B>B &
<
<
< BA A A3 3 3, , , >B%) 1 1 1 1 r5 r8 c P e Zd Zi ZdZd Zd Zd dZd dZd Z d dZ
d d Z d d
Z dS ) Metadata c " || _ d| _ d S )NF)databaserequires_extension)rH r s r6 rI zMetadata.__init__ s
"' r5 c 8 | j || S r: )r execute_sql)rH sqlrh s r6 executezMetadata.execute s }((f555r5 Nc
r t d | j || D }| || \ }}| || }t | dk r;|d }|| t u r t ||< n|| t u r
t ||< t }|
D ]d\ } }
|| } | | |
j } t | | |
j |
j |
j | | | | || < e|S )Nc 3 ( K | ]
}|j |fV d S r: r; )rP metadatas r6 rR z'Metadata.get_columns.. sE F F ]H % F F F F F Fr5 r )r< r= r> r? r@ rC rD )r r get_columnsget_column_typesget_primary_keyslenr rd BigIntegerField BigAutoFieldr| _clean_defaultrC r8 data_typerY r? get)
rH rs schemar column_types extra_paramspk_namespkcolumnsr; column_datar< rC s
r6 r zMetadata.get_columns sc F F M 5 5eV D D F F F F F
&*%:%:5&%I%I" l ((77 x==A !BB <//#, R b!_44#/ R --!) !1!1 9 9 D+&t,K))+{7JKKG " ' + 5$) ' 3 !-!1!1$!7!7 9 9 9GDMM r5 c t r: )NotImplementedErrorrH rs r s r6 r zMetadata.get_column_types s !!r5 c |(|t t fv s| dk rd S t |t r/t |t r| d sd|z }|pdS )NrY 'rZ z'')rd r lowerrc r
isinstancer
startswith)rH r< rC s r6 r zMetadata._clean_default s| ?ki -F F F ==??f $ $ F k< 0 0 ' gy ) ) '292D2DS2I2I ' w&G$ r5 c 8 | j || S r: )r get_foreign_keysr s r6 r zMetadata.get_foreign_keys }--eV<<$$# $ #
?
?r5 c i }i }t Ft t j t j t j t j t j f n
t }|d|d}| d|d }| D ]T\ }} | j
| t ||< || |v rd| _ | | j
v rd| j
| i||< U||fS )Nz."r[ zcSELECT attname, atttypid FROM pg_catalog.pg_attribute WHERE attrelid = %s::regclass AND attnum > %sr Tr< )r setr r r r r r r r r r0 r r )
rH rs r r r extension_types
identifierr r; r s
r6 r z#PostgresqlMetadata.get_column_types s +7*B # ( " & $ & ' ' '
IL #)&&%%%0
<=GL L
** L LID#!% !4!4S,!G!GL D!_44*.'d&&&&3T5Ec5J%K T"\))r5 Nc ^ |pd}t t | || S Npublic)r r r rH rs r r s r6 r z PostgresqlMetadata.get_columns4 . #8'..::5&IIIr5 c ^ |pd}t t | || S r )r r r r s r6 r z#PostgresqlMetadata.get_foreign_keys8 . #8'..??vNNNr5 c ^ |pd}t t | || S r )r r r r s r6 r z#PostgresqlMetadata.get_primary_keys< r r5 c ^ |pd}t t | || S r )r r r r s r6 r z PostgresqlMetadata.get_indexes@ r r5 r: )r1 r2 r3 BooleanField BlobFieldr SmallIntegerFieldr TextField
FloatField DoubleField CharField DateField
DateTimeField TimeField DecimalField UUIDFieldr r r rI r r r r r
__classcell__r s @r6 r r s
L
I O
L I
Z [
i
i
i
m
m
i
i
l!"
i#J(
K >? ? ? ? ?,* * *4J J J J J JO O O O O OO O O O O OJ J J J J J J J J Jr5 r c e Zd Zej Zeed< ej Zeed< dZd Z dS )CockroachDBMetadatar r z#from playhouse.cockroachdb import *c 0 t | | d| _ t j| d }| }|D ] \ }}}|dk rt j | j |< !| j D ]}t j | j |< d S d S )NTr r )
r rI r r r r r r r r )rH r r r r r r s r6 rI z CockroachDBMetadata.__init__N s $)))"& #\\ #3 4 4F oo''G07
H
H,Wng%%+7+GDOC('
?
?'3'>$$ $ #
?
?r5 N)
r1 r2 r3 r r copyr r r rI r4 r5 r6 r r E sf $.3355J!JrN$05577K$K<? ? ? ? ?r5 r c r e Zd Zei Zni ej eej eej e
ej e ej
eej eej eej eej eej eej eej eej eej eej eej e ej e ej eej eej! eiZ fdZ"ddZ# xZ$S )
MySQLMetadataNc d|v r| d |d< t t | j |fi | d S )Npasswordpasswd)popr r rI )rH r kwargsr s r6 rI zMySQLMetadata.__init__{ sO %zz* 5 5F8 +mT""+H???????r5 c i }| d|z }|j D ]2}|d d \ }}| j |t ||< 3|i fS )NzSELECT * FROM `%s` LIMIT 1 )r descriptionr r r0 )rH rs r r r column_descriptionr; type_codes r6 r z MySQLMetadata.get_column_types ss :U BCC #)"4 N N 0! 4OD)!% !4!4Y !M!ML R r5 r: )%r1 r2 r3 r
r BLOBr CHARr DATEr DATETIMEr DECIMALr DOUBLEr FLOATINT24r LONG_BLOBLONGLONGLONGr MEDIUM_BLOB
NEWDECIMALSHORTSTRING TIMESTAMPTIMEr TINY_BLOBTINY
VAR_STRINGrI r r r s @r6 r r ` s~
OY
OY
OY
z
j
l
)
O\
"I
!<
l
y
-!
"
OY#
$
) O\ !9)
.@ @ @ @ @
r5 r c e Zd Zi dededededededede d e
d
e d e d e d
ede de
de de e e e
edZdZdZd ee ZdZd ZddZdS ) SqliteMetadatabigintblobboolbooleanchardatedatetimedecimalfloatintegerzinteger unsignedintlongnumericreal smallintegersmallint)zsmallint unsignedtexttimevarcharz
(?:["\[\(]+)?z
(?:["\]\)]+)?zi(?:FOREIGN KEY\s*)?{begin}(.+?){end}\s+(?:.+\s+)?references\s+{begin}(.+?){end}\s*\(["|\[]?(.+?)["|\]]?\))beginendz&^\s*(?:var)?char\s*\(\s*(\d+)\s*\)\s*$c ( | }|| j v r| j | }nft j | j | rt
}nDt j dd| }|dk rt }n | j |t }|S )Nz\(.+\)r )
r r research
re_varcharr sub BareFieldr r0 )rH column_typer= r< s r6 _map_colzSqliteMetadata._map_col s %++-- do - - //:KK
Yt
8
8 M#KK &B@@Kb ' "o 1 1+| L L r5 Nc i }| j | }|D ]$}| |j ||j < %|i fS r: )r r r0 r r; )rH rs r r r columns r6 r z SqliteMetadata.get_column_types sV -++E22 H HF(,
f6F(G(GL % %R r5 r: )r1 r2 r3 r r r r r r r r r r r r r' r( formatre_foreign_keyr, r0 r r4 r5 r6 r r sS / <
M < < L | <
!" L#$ * + J0
E
C & (.vEsv'C'C
;J r5 r _DatabaseMetadata)r primary_keys foreign_keysrt indexesc e Zd Zd Zd ZdS )DatabaseMetadatac g } j D ]J}t |j dk r0 fd|j D }| ||j f K|S )Nr c ^ g | ])}|j v j | j *S r4 )r r; )rP r2 rH rs s r6
z9DatabaseMetadata.multi_column_indexes.. sG A A A#)"(DL,?"?"? $|E 26 : ?"?"?"?r5 )r8 r r appendrB )rH rs accumrA field_namess `` r6 multi_column_indexesz%DatabaseMetadata.multi_column_indexes s \%( : :E5=!!A%% A A A A A-2] A A A k5< 8999 r5 c i }| j | D ]/}t |j dk r|j ||j d < 0|S )Nr r )r8 r r rB )rH rs r? rA s r6 column_indexesz DatabaseMetadata.column_indexes sK \%( 7 7E5=!!Q&&*/,emA&' r5 N)r1 r2 r3 rA rC r4 r5 r6 r: r: s2 r5 r: c ~ e Zd ZeegZddZd Zedd Z d Z
d Z d Z d Z
dd
Zdd Z dd
Z ddZdS ) IntrospectorNc " || _ || _ d S r: )r r )rH r r s r6 rI zIntrospector.__init__ s
r5 c d| j j z S )Nzr r rk s r6 rW zIntrospector.__repr__ s #dm&<< K | ]}| V d S r: )title)rP r- s r6 rR z/Introspector.make_model_name.. s* E ESSYY[[ E E E E E Er5 _r T)r r* r- rT splitisalpha)rH rs
snake_caser]
model_names r6 make_model_namez Introspector.make_model_name
s~ +#E**Ey"e,,WW E EEKK4D4D E E EEE
!}$$&& * z)Jr5 Fc P | }|rt | }| }|rt j dd| p|}t j dd| }|t
v r|dz
}t
| r |d rd|z }|S )Nz_id$r r_ rb r )stripr r r* r- RESERVED_WORDSr isdigit)rH r2 rf rf s r6 make_column_namez Introspector.make_column_name s -$V,,F : VFB//96F 3// ^ # # cMF v;; "6!9,,.. "6\F
r5 c
| j j | j }|rD| j j | j }| d |D fd|D }t
| }i }i } i }
i } i } |D ]-}
| j |
| j }| j |
| j } | j |
| j |
|
< H|
|
D ]?}|j
|vr4| |j
| |j
@n # t $ r}g |
|
< Y d }~nd }~ww xY w| |
| | |
< t
d |D }t
d |
|
D }| D ]\ }}|rt! j dd| }n| |||v | }| }| d r||v r| }||_ |D ]J}t- |j d k r0|j d
}||v r |j || _ d || _ K| j |
| j | |
< |||
< || |
< /i }d }|D ]}
t
}t7 |
|
|
D ]X} ||
|j }n# t: $ r Y #w xY w|j
} | |v r| d|j d||< C| | Y|D ]}}
|
|
D ]r}||j |j } ||j
|j } n# t: $ r d } Y nw xY w| || | |! | s~tE || |
| | S )NrJ c g | ] }|j
S r4 r )rP views r6 r= z+Introspector.introspect... s 77749777r5 c g | ]}|v | S r4 r4 )rP rs table_namess r6 r= z+Introspector.introspect..1 s# HHH5K3G3Ge3G3G3Gr5 c 3 > K | ]}| V d S r: )r )rP r@ s r6 rR z*Introspector.introspect..Y sJ "D "D&1 #."3"3"5"5 "D "D "D "D "D "Dr5 c 3 $ K | ] }|j V d S r: r2 )rP fk_cols r6 rR z*Introspector.introspect..[ s$ FFfmFFFFFFr5 r_ rb _idr r Tc | j S r: ru )rq s r6 z)Introspector.introspect..| s k&8 r5 )r _set)rq rt ru rF )#r r
get_tablesr get_viewsextendr r r r rr r> addrM rh r| r* r- rm r endswithr; r r rB rA r r~ r2 KeyErrorrs dest_columnrv r r: ) rH rr literal_column_names
include_viewsrf tablesviews table_setr r6 r7 rt r8 rs
table_indexes
table_columnsrq exclower_col_namesfkscol_namer2 new_name
lower_namerA
related_namessort_fnmodels_referencedrr srcru s ` r6
introspectzIntrospector.introspect( s '22$+2FF 9M*44DK4HHE MM77777 8 8 8 "HHHHHHHFKK 2 + 2 +E M 5 5eT[ I IM M 5 5eT[ I IM
B&*m&D&D4;'( '( U# *'3E': B B &1BB "MM+*@ A A A %MM+*@ A A A
)
)
)&( U######
) "&!5!5eZ!H!HK " "D "D5B"D "D "D D DOFF,u2EFFFFFC$1$7$7$9$9
'
' &' A !vih ? ?HH # 4 4Xx35? A AH
&^^ - -
&&u-- 0(o2M2M '~~ / /H & &
;
;u}%%** "]1 -F ..7<|
f-46:
f-3"&-"@"@t{#$ #$L *GEN *GENN
88 6 6E # %l5&9w G G G
6
6 $U^K,> ?FF H ) 3
!222"
-%M&)) &))*5555
6
9
9E +E 2
9
9 k /01CD ";#9:#/ 1DD DDD ## + + !.!2!23!7!7 $ 9 9 9 9
9 s< #E
E#E E#8L
LL6N N N c
||| i } G fddt t
fd
t j D ]\ }}||vr
|| |S )Nc . e Zd Z G fdd ZdS )/Introspector.generate_models..BaseModelc 6 e Zd Z j j Z j ZdS )4Introspector.generate_models..BaseModel.MetaN)r1 r2 r3 r r r rk s r6 Metar s =1 r5 r N)r1 r2 r3 r rk s r6 BaseModelr sG
%
%
%
%
%
%
%
%
%
%
%
%r5 r c j D ]>}|j }||vr1| k r+|v rt j d d| 2 || ?g j }| D ]&\ }}|j r |j '
} G fdd }d} t dk r d|vrd|_ n|
t dk r.t fd | D |_ d
} d|i}
| D ]Y\ }}|j } | t ur
rt" } n| t$ u rt" } ||j d } |j r| r| t( u rt* } d| d < n|j r| t( urd
| d < | r^| rd
| d< n(|j j }
|
|v r ||
| d< n t2 } |
| d< |j r
|j | d< d|d| d< |j t9 d|j z }|g| d< | s1||v r(|| rd
| d< n | sd
| d< nd| d< | di | |
|j < [ t= t? f|
| < n# t@ $ r s Y nw xY w v r! d S d S # v r! w w xY w)Nz'Possible reference cycle found between z and c e Zd Z ZZdS )AIntrospector.generate_models.._create_model..MetaN)r1 r2 r3 r8
table_name)rA rs s r6 r r s . "
r5 r Fr idr c , g | ]\ }}|v |j S r4 r )rP colr^ r6 s r6 r= zGIntrospector.generate_models.._create_model.. s4 2, 2, 2,#-3l** J***r5 T)r@ rY r? rH r] rel_model_namer^ rb _relr_ z
DEFAULT %sr\ rB rA r4 )"r~ r7 rr warningswarnr r| r? r> r; rA rC r keys CompositeKeyr< rb r. r0 r> rd r rf ro rq DeferredForeignKeyrG rC SQLrg rT strrM remove)rs modelsrq ru r r@ r2 rC r
composite_keyrU
FieldClassrh rr
constraintrA r6 r
_create_model bare_fieldsr pending skip_invalids ` @@r6 r z3Introspector.generate_models.._create_model s{ KK ' 4U ;
4
4 "-v%%$%--w
5:UUDD'B C C C C &
dF333 L &u-G'.}}
5
5# V% 5 '' 444#+#@#@#G#G % 4 4U ; ;N
#
#
#
#
#
#
#
#
#
#
#
"M< A%%w&&',D$$#*<<>>L< 1$$#/ 2, 2, 2, 2,18 2, 2, 2, $- !%
TNE'.}} -
: -
:# V # /
_444!*JJ <//!*J $/ "O- - % 1- 1!Y..%1
,1F=))' 1Ji,G,G,0F=)((** K 4466 B*0w%+%7%B
%//.4Z.@F7OO);J7AF#3 4 :*0/w 8=uukkk(JF9%>-!$\FN%B!C!CJ-7LF=) ,,.. 0"n44)+6 3/3F8 , ,!'!6!6!8!8 3.2F7O*/w%/Z%9%9&%9%9fk""
* $SZZ) u E Eu
#
G##NN5))))) $#5G##NN5)))) $s* "K7 6L' 7
LL' LL' ' M)r Modelr r~ rt r| )
rH r rr r r r r rs r] r r r r s
`` ` @@@@r6 generate_modelsz Introspector.generate_models s ??;0D#02 2 % % % % % % % % % %
%%^ * ^ * ^ * ^ * ^ * ^ * ^ * ^ * ^ * ^ *B #8#7#=#=#?#? @ @ - -LE5F""
eV,,,
r5 r: )T)FT)NFFT)FNFFF)r1 r2 r3 rd r
pk_classesrI rW classmethodrR rU rW rZ r] rh rm r r r4 r5 r6 rE rE s \*J = = = , , , [,", , ,/ / /5 5 5
$ AF37z z z zx ?C@E&+s s s s s sr5 rE c b t | | }| S )NrJ )rE rR r )r r introspectors r6 r r s, --hv-FFL " " $ $$r5 c T t | | } |j di |S )NrJ r4 )rE rR r )r r optionsr s r6 r r s3 --hv-FFL '< ' 2 2' 2 22r5 TFc t | j j | j j D ]}d|j d|j g}|j r| d n;|r9|j r| d n |j r| d t |t r/| d|j j d|j
j t d | |r| j }|sd S t d |D ]U}dg}| j j }| d
d 5 | t) |j |j r/| d
| |j | \ }} d d d n # 1 swxY w Y |t3 t5 t6 | z }
| |
dd |j r| d t d | Ud S d S )Nz z PKz UNIQUEz INDEXz FK: .r z
index(es)%sz"")paramquotez WHERE r[ ) r( _metar;
sorted_fields
field_typer? r> rB rA r rb rE r1 rel_fieldrT fields_to_indexr get_sql_context scope_valuesr r _expressions_whereliteralquerytuplemapr re _unique) r] r8 inline_indexesr^ parts
index_listrA ctxr rh cleans r6 print_modelr $ s %+
* #jjj%*:*:;< ' LL
'|
' Y''''
' X&&& e_ - - @ LLL )A)A)A). )=)=?
@
@
@
bggenn "[0022
F
m " "EFE+&6688C!!D!99
*
*
e&899:::< *KK ***GGEL))) !iikk V
*
*
*
*
*
*
*
*
*
*
*
*
*
*
* %$8& A ABBBE LL sB// 0 0 0}
( Y''' "''%.. ! ! ! !+" " " "s &A5G''G+ .G+ c . | j \ }}| j j j dk r%| | j j j d }t j d| }| \ }}}d
d | d D }d
|||f }|t t t | z S )Nr z^(.+?\()(.+)(\).*)z,
c 3 K | ] }d |z V
dS )z %sNr4 )rP r2 s r6 rR z get_table_sql..T s& LLf&6/LLLLLLr5 rK r\ )_schema
_create_tabler r r r re r* matchgroupsrT rd rj r r r ) r] r rh match_objcreater extraindentedr s r6
get_table_sqlr L s ---//5577KC {!T))kk%+.4d;; .44I & - - / /FGUzzLL
d8K8KLLLLLH IIvx/ 0 0 6 6 8 8E 5 16::;; ;;r5 c > t t | d S r: )r( r )r] s r6 print_table_sqlr Y s -
r5 r: )TF)) collectionsr ImportErrordictr inspectr r* r peeweer r r r
r r pymysql.constantsr
MySQLdb.constants playhouser playhouse.cockroachdbr r rk objectr0 r8 r r r r r r5 r: rE r r r r r r4 r5 r6 r s2 ''''''' KKK " " " " " " ' ' ' ' ' ' " " " " " " ,,,,,,, 0000000
&&&&&&& LLL 7777777 6 m m m m mV m m m`@8 @8 @8 @8 @8v @8 @8 @8FbJ bJ bJ bJ bJ bJ bJ bJJ? ? ? ? ?, ? ? ?6+ + + + + H + + + \7 7 7 7 7 X 7 7 7 t J 2 5 ( &t t t t t6 t t tn % % % %
3 3 3 3
%" %" %" %"P < < < sf A A7 A'&A7'A1.A70A11A76A7;B B B B B! B!