U
/�X2 � @ s� d dl mZmZ d dlZd dlmZ dZde Z dd� Z
ej dd � ej d
d � ej
e
iZG d d � d e�Zd
d� Zd'dd�Zdd� 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 )�field�params� 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| jd�S )N�Z
max_digitsZdecimal_placesZ
auto_roundZroundingr r r
r
r r s
�c s0 e Zd Zd dd�Z� fdd�Zd
dd�Z� 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|tj�r�|jj | _|r�|jjj |jkr�d|jjj | _n
|jj| _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| ��� }|�dd� | j�� D �� |S )Nc S s i | ]\}}|t |��qS r
)�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< | � � }d�dd� t|�� �D ��}dj| j|| jj|d�S )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, �join�sortedr- �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 �fields�set�append�
create_fields� drop_fields�compare_fields�pop�
change_fields�change_not_null)�model1Zmodel2�kwargs�changesZfields2Znames1r> Znames2Zfields_Znulls_r �field1�field2Zdiffr r
r< r �diff_oneG s2
rL Fc s� t �� �� t ����|r(t� �� t���tdd� � D ��� tdd� �D ���g }�fdd�� D �D ]}|�t� | |d�� qb� fdd��D �D ]}|�t�| �� q�� �� D ](\}}|�kr�q�|t|�| |d�7 }q�|S )Nc S s g | ]}|j j|f�qS r
�r r �r' �mr
r
r r= w s z diff_many.<locals>.<listcomp>c S s g | ]}|j j|f�qS 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_topologically�reversedr 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|d�S )Nz&class {classname}(pw.Model):
{fields}
c s. g | ]&}t |tj�r |jd kst|f� ��qS )�id)r r ZPrimaryKeyFieldr �
field_to_code�r' r �rH r
r r= � s
�z!model_to_code.<locals>.<listcomp>)r4 r> )�INDENT�NEWLINEr5 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 �F�rX 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 jd�tt|��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|�}|�|rdnd�S )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 )N�clsTr )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| j�sBt| jt�rB| 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
� *(