/�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 )�field�params� 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_digits�decimal_places�
auto_round�rounding)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)N�defaultzmigrator.orm['%s'])�name�type� field_class�nullZnullableZ primary_keyZ db_column�index�uniquer r �callable�FIELD_TO_PARAMS�updateZ rel_modelZ related_nameZto_field�
isinstance�pw�ForeignKeyField�_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)�.0�k�vr
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, �join�sortedr- �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! �fields�set�append�
create_fields� drop_fields�compare_fields�pop�
change_fields�change_not_null)�model1Zmodel2�kwargs�changesZfields2Znames1r; Znames2Zfields_Znulls_r �field1�field2Zdiffr 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_topologically�reversedr 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; )�INDENT�NEWLINEr3 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)N�clsTr )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*
* (