/X2 @ s. d d l m Z m Z d d l Z d d l m Z d Z d e Z d d Z
e j d d e j d
d e j
e
i Z Gd d d e Z d
d Z d d d d Z d d Z d d Z d d Z d d Z d d Z d d d Z d d Z d! d" Z d# d$ Z d% d& Z d S)' )Hashable OrderedDictN)Columnz
c C sB i } | j d k r" | j | d <| j d 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 i S)N
max_length)r
)fr
r
r <lambda> s r c C s( d | j d | j d | j d | j i S)N
max_digitsdecimal_places
auto_roundrounding)r r r r )r r
r
r r s c s@ e Z d Z d d d Z f d d Z d d d Z S) r Nc C sA | j | _ t | | _ | j | _ | j | _ | j | _ | j | _ | j | _ i | _ | j
d k r t | j
r | j
| j d <| j t k r | j j
t | j | d | _ d | _ d | _ t | t j r=| j j | _ | r.| j j j | j k r.d | j j j | _ n | j j | _ 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 | j } | j d d | j j D | S)Nc S s% i | ] \ } } t | | q Sr
)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_ | j } d j d d t | j D } d j d | j d | d | j j d | S) Nz, c s s% | ] \ } } d | | f Vq d S)z%s=%sNr
)r' r( 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# r1 Z field_paramsZ param_strr
r
r get_field> s
zColumn.get_field)r"
__module__ __qualname__r% r, r6 r
r
)r. r r s r c s g } | j j | j j } t t | } | rl f d d | D } | j t | | | t | t } | r | j t | | g } g } x t | | D]n }
|
| |
} } t | | }
|
j d d } |
r| j | | d k r | j |
| f q W| rP| j t | | | x- | D]% \ }
} | j t | |
| qWW| S)z&Find difference between Peewee models.c s g | ] } | q Sr
r
)r' r )fields1r
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
)r9 r diff_oneG s2
rI Fc s= t j t j | r< t t t d d D t d d D g } x> f d d D D]# } | j t | d | q Wx8 f d d D D] } | j t | q WxF j D]8 \ } } | k rq | t | | d | 7} q W| S)Nc S s" g | ] } | j j | f q Sr
)r! r )r' mr
r
r r: w s z diff_many.<locals>.<listcomp>c S s" g | ] } | j j | f q Sr
)r! r )r' rJ r
r
r r: x s c s" g | ] } | k r | q Sr
r
)r' rJ )models2r
r r: } s r$ c s" g | ] } | k r | q Sr
r
)r' rJ )models1r
r r: s ) r Zsort_models_topologicallyreversedr r= create_model remove_modelr- rI )rL rK r$ reverserF r rD r
)rL rK r diff_manyo s" ! !rQ c sK d } t t j f d d | j j D } | j d | j d | S)Nz&class {classname}(pw.Model):
{fields}
c s@ g | ]6 } t | t j o* | j d k s t | q S)id)r r ZPrimaryKeyFieldr
field_to_code)r' r )rE r
r r: s z!model_to_code.<locals>.<listcomp>r2 r; )INDENTNEWLINEr3 r! Z
sorted_fieldsr5 r" )ModelrE templater; r
)rE r
model_to_code s # rX c K s d t | | S)Nz@migrator.create_model
)rX )rV rE r
r
r rN s rN c K s d | j j S)Nzmigrator.remove_model('%s'))r! db_table)rV rE r
r
r rO s rO c s; d t | j j t d t j f d d | D f S)Nz migrator.add_fields(%s'%s', %s),c s" g | ] } t | d q S)F)rS )r' r )rE r
r r: s z!create_fields.<locals>.<listcomp>)rU r! rY r3 )rV r; rE r
)rE r r> s r> c O s& d | j j d j t t | f S)Nz migrator.remove_fields('%s', %s)z, )r! rY r3 mapr& )rV r; rE r
r
r r? s r? Tc K s( t | | } | j | r! d n d S)Nr/ )r r6 )r r1 rE colr
r
r rS s rS c K s t | t | } } | | k r/ d d i St | } | j | d <t | } | j | d <t t | j t | j S)NclsTr )r field_to_paramsr dictr< r- )rG rH rE Z
field_cls1Z
field_cls2Zparams1Zparams2r
r
r r@ s
r@ c K sf t j t | d d | } | j d k rb t | j rb t | j t rb | j | d <| S)Nc S s i S)Nr
)r r
r
r r s z!field_to_params.<locals>.<lambda>r )r getr r r r r )r rE r r
r
r r_ s $
r_ c O s. d | j j d t j d d | D f S)Nz migrator.change_fields('%s', %s)rZ c S s g | ] } t | d q S)F)rS )r' r r
r
r r: s z!change_fields.<locals>.<listcomp>)r! rY rU r3 )rV r; rE r
r
r rB s rB c C s/ | r d n d } d | | j j t | f S)NZ
drop_not_nullZ add_not_nullzmigrator.%s('%s', %s))r! rY r& )rV r r Z operationr
r
r rC s rC ) collectionsr r Zpeeweer Zplayhouse.reflectionr Z VanilaColumnrT rU r Z CharFieldZ DecimalFieldr r rI rQ rX rN rO r> r? rS r@ r_ rB rC r
r
r
r <module> s*
* (