U
/X2 @ s d dl mZmZ d dlZd dlmZ dZde Z dd Z
ej dd ej d
d ej
e
iZG d d d eZd
d Zd'ddZdd Zdd Zdd Zdd Zdd Zd(d d Zd d Zd!d" Zd#d$ Zd%d& ZdS )) )Hashable OrderedDictN)Columnz
c C s0 i }| j d k r| j |d< | jd k r,| j|d< |S )N on_delete on_update)r r )fieldparams r
/auto.py fk_to_params s
r c C s
d| j iS )N
max_length)r
fr
r
r <lambda> r c C s | j | j| j| jdS )NZ
max_digitsZdecimal_placesZ
auto_roundZroundingr r r
r
r r s
c s0 e Zd Zd ddZ fddZd
ddZ ZS ) r Nc C s |j | _ t|| _|j| _|j| _|j| _|j| _|j| _i | _ |j
d k r`t |j
s`|j
| j d< | jt kr| j
t | j | d | _d | _d | _t|tjr|jj | _|r|jjj |jkrd|jjj | _n
|jj| _d S )Ndefaultzmigrator.orm['%s'])nametype field_classnullZnullableZ primary_keyZ db_columnindexuniquer r callableFIELD_TO_PARAMSupdateZ rel_modelZ related_nameZto_field
isinstancepwForeignKeyField_metaZorm__name__)selfr migratorr
r
r __init__ s(
zColumn.__init__c s, t t| }|dd | j D |S )Nc S s i | ]\}}|t |qS r
)repr.0kvr
r
r
<dictcomp>; s z/Column.get_field_parameters.<locals>.<dictcomp>)superr get_field_parametersr r items)r" r __class__r
r r, 9 s zColumn.get_field_parameters c C s< | }ddd t| D }dj| j|| jj|dS )N, c s s | ]\}}d ||f V qdS )z%s=%sNr
r& r
r
r <genexpr>A s z#Column.get_field.<locals>.<genexpr>z-{name}{space}={space}pw.{classname}({params}))r space classnamer )r, joinsortedr- formatr r r! )r" r3 Z field_paramsZ param_strr
r
r get_field> s
zColumn.get_field)N)r0 )r!
__module__ __qualname__r$ r, r8
__classcell__r
r
r. r r s
r c s" g }| j j |j j}t t| }|rP fdd|D }|t| f|| t|t }|rx|t| f| g }g } t | | D ]P}
|
||
} } t| | }
|
dd}|
r|| |dk r| |
|f q|r|t| f|| | D ] \}
}|t | |
| q |S )z&Find difference between Peewee models.c s g | ] } | qS r
r
)r' r Zfields1r
r
<listcomp>Q s z diff_one.<locals>.<listcomp>r N)
r fieldssetappend
create_fields drop_fieldscompare_fieldspop
change_fieldschange_not_null)model1Zmodel2kwargschangesZfields2Znames1r> Znames2Zfields_Znulls_r field1field2Zdiffr r
r< r diff_oneG s2
rL Fc s t t |r(t ttdd D tdd D g }fdd D D ]}|t | |d qb fddD D ]}|t| q D ](\}}|krq|t|| |d7 }q|S )Nc S s g | ]}|j j|fqS r
r r r' mr
r
r r= w s z diff_many.<locals>.<listcomp>c S s g | ]}|j j|fqS r
rM rN r
r
r r= x s c s g | ]}| kr|qS r
r
rN )models2r
r r= } s )r# c s g | ]}| kr|qS r
r
rN )models1r
r r= s ) r Zsort_models_topologicallyreversedr r@ create_model remove_modelr- rL )rQ rP r# reverserI r rG r
)rQ rP r diff_manyo s"
rV c s4 d}t t fdd| jjD }|j| j|dS )Nz&class {classname}(pw.Model):
{fields}
c s. g | ]&}t |tjr |jd kst|f qS )id)r r ZPrimaryKeyFieldr
field_to_coder' r rH r
r r= s
z!model_to_code.<locals>.<listcomp>)r4 r> )INDENTNEWLINEr5 r Z
sorted_fieldsr7 r! )ModelrH templater> r
rZ r
model_to_code s
r_ c K s dt | f| S )Nz@migrator.create_model
)r_ r] rH r
r
r rS s rS c K s d| j j S )Nzmigrator.remove_model('%s'))r db_tabler` r
r
r rT s rT c s. dt | jjt dt fdd|D f S )Nz migrator.add_fields(%s'%s', %s),c s g | ]}t |d f qS FrX rY rZ r
r r= s z!create_fields.<locals>.<listcomp>)r\ r ra r5 r] r> rH r
rZ r rA s
rA c O s d| j jdtt|f S )Nz migrator.remove_fields('%s', %s)r1 )r ra r5 mapr% re r
r
r rB s rB Tc K s t | f|}||rdndS )Nr0 )r r8 )r r3 rH colr
r
r rX s rX c K sb t | t | }}||kr"ddiS t| }| j|d< t|}|j|d< tt| t| S )NclsTr )r field_to_paramsr dictr? r- )rJ rK rH Z
field_cls1Z
field_cls2Zparams1Zparams2r
r
r rC s
rC c K sF t t| dd | }| jd k rBt| jsBt| jtrB| j|d< |S )Nc S s i S )Nr
r r
r
r r r z!field_to_params.<locals>.<lambda>r )r getr r r r r )r rH r r
r
r rj s
rj c O s$ d| j jdt dd |D f S )Nz migrator.change_fields('%s', %s)rb c S s g | ]}t |d qS rc rd )r' r r
r
r r= s z!change_fields.<locals>.<listcomp>)r ra r\ r5 re r
r
r rE s rE c C s" |rdnd}d|| j jt|f S )NZ
drop_not_nullZ add_not_nullzmigrator.%s('%s', %s))r ra r% )r] r r Z operationr
r
r rF s rF )NF)T) collectionsr r Zpeeweer Zplayhouse.reflectionr Z VanilaColumnr[ r\ r Z CharFieldZ DecimalFieldr r rL rV r_ rS rT rA rB rX rC rj rE rF r
r
r
r <module> s2
*(