]jb/ U d Z ddlmZ ddlmZmZmZmZmZm Z m
Z
m Z m Z m
Z
mZ ddlZddlmZ erddlmZ ddlmZmZ d Zd
ed < d ez Zd
ed
< ddddddddZ d
ed< dZ d
ed< dKdZ dLdZ ej d ej! d ej" e ej# e iZ$d ed < G d d e ZdMd$Z% dNd%d&dOd,Z&dPd/Z'dPd0Z(dPd1Z)dQd4Z*dQd5Z+d6d7dRd:Z,dSd>Z-dTdAZ.dUdDZ/dVdGZ0dWdHZ1dXdJZ2dS )Yz Automatically create migrations. ) annotations)
TYPE_CHECKINGAnyCallableDictFinalIterableListOptionalTypeUnioncastN)Column )Migrator)
TModelTypeTParamsz r INDENT
NEWLINEpw_pext)
ArrayFieldBinaryJSONFieldDateTimeTZField HStoreField
IntervalField JSONField
TSVectorFieldFIELD_MODULES_MAP)zplayhouse.postgres_extzplayhouse.fieldspeewee
PW_MODULESfieldpw.ForeignKeyFieldreturnr c ^ i }| j d| j d|d< | j d| j d|d< |S )z Get params from the given fk.N' on_delete on_update)r' r( r" paramss J/opt/imunify360/venv/lib64/python3.11/site-packages/peewee_migrate/auto.py fk_to_paramsr, ' sJ
F " 4%/ 4 4 4{ " 4%/ 4 4 4{ M pw.DateTimeFieldc R i }t | j t s
| j |d< |S )z)Get params from the given datetime field.formats)
isinstancer0 listr) s r+
dtf_to_paramsr3 3 s,
F emT * * * !My Mr- c d| j iS )N
max_length)r5 fs r+ <lambda>r8 = s
\1< 8 r- c 8 | j | j | j | j dS )N
auto_rounddecimal_places
max_digitsroundingr: r6 s r+ r8 r8 > s# l*lJ r- z.Dict[Type[pw.Field], Callable[[Any], TParams]]FIELD_TO_PARAMSc P e Zd ZU dZded< d fd Zdd fd
Zd d
d fdZ xZS )r z!Get field's migration parameters.Type[pw.Field] field_classr" pw.Fieldc 0 t t | j |j t | |j |j f|j |j |j |j
d| || _ | j t v r@| j i | _ | j t | j | d | _ d | _ t% |t& j r>|j j | _ |j |j k rdnd|j j j z | _ d S d S )N) primary_key column_nameindexuniquez'self'zmigrator.orm['%s'])superr __init__namefind_field_type
field_typenullrE rF rG rH r" rB r? extra_parametersupdate rel_modelto_fieldr1 pwForeignKeyField rel_fieldmodel_meta
table_name)selfr" kwargs __class__s r+ rJ zColumn.__init__N s& $fd$ J E " " J
) )+<
. .$,(*% ! ( ( 9I)J5)Q)Q R R R
eR / 0 0 !O 0DM ?ek11 )EO,A,LL
NNN r- spacestrr$ c t t | }t | j j d }d | d D \ }}}d |||| S )z.Generate the field definition for this column.rS c : g | ]}|o| S )strip).0ss r+
<listcomp>z$Column.get_field.<locals>.<listcomp>q s$ HHHa!/ HHHr- =z%{name}{space}={space}{module}.{field})rK r" r] module) rI r get_fieldr getrB __name__ partitionformat)rY r] r" rg rK _r[ s r+ rh zColumn.get_fieldm s fd##--//"&&t'7'@$GGHH5??33G3GHHHa6==U% >
r- Fchanger c L t t | }| dd |rt | dd |d< t | dd p|d |d< | dd | dd | j |d< | j j Mt | j j s4t | j
| j j |d < |S )
z#Generate parameters for self field.backrefNrH FrG r' r( rN default) rI r get_field_parameterspopboolnullabler" rr callablereprdb_value)rY ro r* r[ s r+ rs zColumn.get_field_parametersv s vt$$99;;
9d### + #FJJx$?$? @ @F8 "6::gu#=#= > > R&BRF7O JJ{D ) ) ) JJ{D ) ) ) !]F6N : )(4:;M2N2N ) $TZ%8%89K%L%L M MF9
r- )r" rC )r\ )r] r^ r$ r^ )r$ r ) rj
__module__ __qualname____doc____annotations__rJ rh rs
__classcell__)r[ s @r+ r r I s ++ >
.3 r- r model1r model2 List[str]c g } j |j }}|j |j }t t | z
}|r1fd|D }| t g|R i | t | t z
} | r | t g| R g }
g } g } t |z
| z
D ]}
|
||
}}t
|| }| dd }| dd }| dd }|r|
||f || |
|f ||| |
||f |
r | t g|
R | D ]*\ }
}| t |
| +| D ]\ }
}}|du s|du rc||
j
s
||
j r#| t |
| t |
| q| t |
|j }|j }t | t | z
}| fd |D t | t | z
}| fd
|D |S ) z,Find difference between given peewee models.c g | ]
}| S ra ra )rc rK field_names1s r+ re z diff_one.<locals>.<listcomp> s 888,t$888r- rN NrG rH )rN T)rH c 3 K | ]V}t |d t t f r2t |d dk =t |d V WdS )r r )rK N)r1 r2 tuplelen
drop_indexrc rG r s r+ <genexpr>zdiff_one.<locals>.<genexpr> sq eAhu
. . 47uQx==13D3D 6a)))3D3D3D3D r- c 3 K | ]]}t |d t t f r9t |d dk =t |d |d V ^dS )r r )rK rH N)r1 r2 r r add_indexr s r+ r zdiff_one.<locals>.<genexpr> sw eAhu
. . 47uQx==13D3D &uQxa9993D3D3D3D r- )rW fieldssetappend
create_fields drop_fieldscompare_fieldsrt
change_fieldschange_not_nullrH rG r r indexesextend)r r rZ changesmeta1meta2 field_names2names1r names2fields_to_change fields_nullsfields_indexesrK field1field2diffrN rG rH indexes1indexes2indexes_to_dropindexes_to_addr s ` @r+ diff_oner s G< 5E<L<L
\!2!2
2F
A8888888}V?f?????@@@
\!2!2
2F
5{63F333444 LNL!!F*V3 9 9%d+\$-? ff--xx%%$''(D)) 4 # #VTN 3 3 3 t - - - 6#5 ! !4"7 8 8 8 A}V?.>???@@@" A A
dvt$???@@@@ - 5 5eV TMMv~~D!(
9L,>,D
9z&$ 7 7888 NN9VT&AAA B B B B NN:fd33 4 4 4 4 }H}H (mmc(mm3O NN $ ]]S]]2N NN # Nr- F)reversemodels1List[TModelType]models2migratorOptional[Migrator]c t t d t j | } t t d t j | }|r8t t | } t t | }d | D d |D g } D ]5\ }}|vr
| t || 6| fdfdD D | fdfdD D |S ) z9Calculate changes for migrations from models2 to models1.r c N i | ]"}t t |j j |#S ra r r^ rW rX rc ms r+
<dictcomp>z diff_many.<locals>.<dictcomp> ) EEE4QW /00!EEEr- c N i | ]"}t t |j j |#S ra r r s r+ r z diff_many.<locals>.<dictcomp> r r- r c 3 F K | ]}t | V dS )r N) create_model)rc rK r models_map1s r+ r z diff_many.<locals>.<genexpr> sI [&::: r- c g | ]}|v| S ra ra )rc r models_map2s r+ re z diff_many.<locals>.<listcomp> s# DDD1q /C/CQ/C/C/Cr- c 3 B K | ]}t | V d S N) remove_model)rc rK r s r+ r z diff_many.<locals>.<genexpr> sA ,0 [&'' r- c g | ]}|v| S ra ra )rc r r s r+ re z diff_many.<locals>.<listcomp> s$ 4d4d4d1qXcOcOcQOcOcOcr- ) r r
rS sort_modelsr2 reverseditemsr r ) r r r r r rK r r r s ` @@r+ diff_manyr s 4 %r~g'>'>??G4 %r~g'>'>??G *x(())x(())EEWEEEKEEWEEEKG#))++ O O f { " " x D(9HMMMNNNN NN DDDD DDD NN 4d4d4d4d 4d4d4d Nr-
model_typer^ c
d}| j }t t fd|j D z }t t t ddt d|j d|j rt d|j dndt |j t j rt d |j j nd|j
rt d
|j
ndg z }| | j || S ) z(Generate migrations for the given model.z-class {classname}(pw.Model):
{fields}
{meta}c t g | ]4}t |t j r |j d k 't |fi 5S )id)r1 rS PrimaryKeyFieldrK
field_to_coderc r" rZ s r+ re z!model_to_code.<locals>.<listcomp> sX
ub&899
?DjD>P>P
% * *6 * *>P>P>Pr- Nz class Meta:ztable_name = ""z
schema = " z primary_key = pw.CompositeKeyz
indexes = ) classnamer meta)rW r r join
sorted_fieldsfilterrX schemar1 rE rS CompositeKey field_namesr rl rj )r rZ templater r s ` r+
model_to_coder s0 AH D
gll
+
F GLL ;; ;;;7;{J633T[3333 "$"2BODD v\\D<L<X\\\9= M655T\5552
D" ??Z%8d? S SSr- c " dt | fi |z S )z$Generate migrations to create model.z@migrator.create_model
)r )r rZ s r+ r r s %
j(K(KF(K(K KKr- c $ | j }d|j z S )z$Generate migrations to remove model.zmigrator.remove_model('%s'))rW rX )r rZ r s r+ r r s D (4? ::r- r rC c
| j }dt d|j dt dt z fd|D z dS )z"Generate migrations to add fields.zmigrator.add_fields(r& ', ,c . g | ]}t |fd diS )r] Fr r s r+ re z!create_fields.<locals>.<listcomp>+ s/ ` ` `PU
e K K5 KF K K ` ` `r- ))rW r rX r r r rZ r s ` r+ r r $ sg DD =
` ` ` `Y_ ` ` `
a
a b b b r- c x | j }d|j dd t t | dS )z%Generate migrations to remove fields.zmigrator.remove_fields('r z, r )rW rX r maprx r s r+ r r / s> DD #dF##$$$$ r- Tr] r] ru c R t | }| |rdnd S )zGenerate field description.r\ r )r rh )r" r] rZ cols r+ r r 8 s(
--C ==-2 . ..r- r r r c t | t | }}||k rddiS t | | j | j d t ||j |j d }}| d | d }}t t
| t
| z
S )z$Find diffs between the given fields.typeT)rG rH )rO rn )rL r rG rH rs dictr r )r r ftype1ftype2col1col2params1params2s r+ r r > s $V,,of.E.EFF
~ v&,&-(X(XYYYv&,&-(X(XYYY D
!!!.. !!!.. G GMMOO$$s7==??';';; < <<r- model_clspw.Tuple[pw.Field, Dict]c v | j }d|j ddt z d |D dS )z%Generate migrations to change fields.zmigrator.change_fields('r r c 6 g | ]\ }}t |d S )Fr r )rc r7 r s r+ re z!change_fields.<locals>.<listcomp>T s) R R R4mAU ; ; ; R R Rr- r )rW rX r r )r r r s r+ r r O sH ?DD w R R6 R R RSSSS r- rK rN c X | j }|rdnd}d|d|j dt | dS )Generate migrations.
drop_not_null add_not_nullz migrator.z('r r )rW rX rx )r rK rN r operations r+ r r X s; D#';^II&/ii $t**** MMr- Union[str, Iterable[str]]rH c v | j }t | d }d|j d| d| dS )r ()[]zmigrator.add_index('r z , unique=r rW rx rb rX )r rK rH r columnss r+ r r _ sF D4jj v&&G Q$/ Q Qg Q Q Q Q QQr- c p | j }t | d }d|j d| dS )r r zmigrator.drop_index('r r r )r rK r r s r+ r r f s> D4jj v&&G A4? A Aw A A AAr- rA c t | }|j t vr)| D ]}|j t v r|c S |S r )r rz r! mro)r" ftypeclss r+ rL rL m sT KKE z))99;; C~++
, Lr- )r" r# r$ r )r" r. r$ r )r r r r r$ r r )r r r r r r r$ r )r r r$ r^ )r r r rC r$ r^ )r" rC r] ru r$ r^ )r rC r rC r$ r )r r r r r$ r^ )r r rK r^ rN ru r$ r^ )r r rK r rH ru r$ r^ )r r rK r r$ r^ )r" rC r$ rA )3r|
__future__r typingr r r r r r r
r r r
r r rS playhouse.reflectionr VanilaColumnr r typesr r r r} r r r! r, r3 CharField DecimalFieldrT
DateTimeFieldr? r r r r r r r r r r r r r rL ra r- r+ <module>r se & & & " " " " " " 7 7 7 7 7 7 +""""""******** K
J J J J L88O m
C
; ; ; ; ;\ ; ; ;|I I I I^ $($
$ $ $ $ $ $N T T T T>L L L L
; ; ; ; 59 / / / / / / = = = =" N N N NR R R RB B B B r-