3
š(ƒhî  ã               @   sX   d dl mZ d dlmZ ddlmZ G dd„ deƒZG dd„ deƒZG d	d
„ d
eƒZdS )é    )Úunicode_literals)Úrouteré   )Ú	Operationc               @   sJ   e Zd ZdZddgZddd„Zdd„ Zd	d
„ Zdd„ Zdd„ Z	dd„ Z
dS )ÚSeparateDatabaseAndStatea"  
    Takes two lists of operations - ones that will be used for the database,
    and ones that will be used for the state change. This allows operations
    that don't support state change to have it applied, or have operations
    that affect the state or not the database, or so on.
    Údatabase_operationsÚstate_operationsNc             C   s   |pg | _ |pg | _d S )N)r   r   )Úselfr   r   © r
   úK/tmp/pip-install-q3hcpn_q/Django/django/db/migrations/operations/special.pyÚ__init__   s    
z!SeparateDatabaseAndState.__init__c             C   s2   i }| j r| j |d< | jr$| j|d< | jjg |fS )Nr   r   )r   r   Ú	__class__Ú__name__)r	   Úkwargsr
   r
   r   Údeconstruct   s    

z$SeparateDatabaseAndState.deconstructc             C   s    x| j D ]}|j||ƒ qW d S )N)r   Ústate_forwards)r	   Ú	app_labelÚstateÚstate_operationr
   r
   r   r   "   s    z'SeparateDatabaseAndState.state_forwardsc             C   s<   x6| j D ],}|jƒ }|j||ƒ |j||||ƒ |}qW d S )N)r   Úcloner   Údatabase_forwards)r	   r   Úschema_editorÚ
from_stateÚto_stateÚdatabase_operationr
   r
   r   r   &   s
    z*SeparateDatabaseAndState.database_forwardsc             C   sd   i }x*| j D ] }|||< |jƒ }|j||ƒ qW x.t| j ƒD ] }|}|| }|j||||ƒ q<W d S )N)r   r   r   ÚreversedÚdatabase_backwards)r	   r   r   r   r   Z	to_statesZdbopr   r
   r
   r   r   .   s    z+SeparateDatabaseAndState.database_backwardsc             C   s   dS )Nz(Custom state/database change combinationr
   )r	   r
   r
   r   Údescribe=   s    z!SeparateDatabaseAndState.describe)NN)r   Ú
__module__Ú__qualname__Ú__doc__Zserialization_expand_argsr   r   r   r   r   r   r
   r
   r
   r   r      s   
r   c               @   sZ   e Zd ZdZdZddd„Zdd„ Zed	d
„ ƒZdd„ Z	dd„ Z
dd„ Zdd„ Zdd„ ZdS )ÚRunSQLzá
    Runs some raw SQL. A reverse SQL statement may be provided.

    Also accepts a list of operations that represent the state change effected
    by this SQL change, in case it's custom column/table creation/deletion.
    Ú NFc             C   s*   || _ || _|pg | _|pi | _|| _d S )N)ÚsqlÚreverse_sqlr   ÚhintsÚelidable)r	   r#   r$   r   r%   r&   r
   r
   r   r   J   s
    

zRunSQL.__init__c             C   sL   d| j i}| jd k	r| j|d< | jr.| j|d< | jr>| j|d< | jjg |fS )Nr#   r$   r   r%   )r#   r$   r   r%   r   r   )r	   r   r
   r
   r   r   Q   s    




zRunSQL.deconstructc             C   s
   | j d k	S )N)r$   )r	   r
   r
   r   Ú
reversiblea   s    zRunSQL.reversiblec             C   s    x| j D ]}|j||ƒ qW d S )N)r   r   )r	   r   r   r   r
   r
   r   r   e   s    zRunSQL.state_forwardsc             C   s(   t j|jj|f| jŽr$| j|| jƒ d S )N)r   Úallow_migrateÚ
connectionÚaliasr%   Ú_run_sqlr#   )r	   r   r   r   r   r
   r
   r   r   i   s    zRunSQL.database_forwardsc             C   s:   | j d krtdƒ‚tj|jj|f| jŽr6| j|| j ƒ d S )Nz!You cannot reverse this operation)r$   ÚNotImplementedErrorr   r(   r)   r*   r%   r+   )r	   r   r   r   r   r
   r
   r   r   m   s    
zRunSQL.database_backwardsc             C   s   dS )NzRaw SQL operationr
   )r	   r
   r
   r   r   s   s    zRunSQL.describec             C   sœ   t |ttfƒrdxˆ|D ]J}d }t |ttfƒrPt|ƒ}|dkrD|\}}ntd| ƒ‚|j||d qW n4|tjkr˜|jj	j
|ƒ}x|D ]}|j|d d q‚W d S )Né   zExpected a 2-tuple but got %d)Úparams)Ú
isinstanceÚlistÚtupleÚlenÚ
ValueErrorÚexecuter!   Únoopr)   ÚopsZprepare_sql_script)r	   r   Zsqlsr#   r.   ÚelementsZ
statementsZ	statementr
   r
   r   r+   v   s    



zRunSQL._run_sql)NNNF)r   r   r   r    r5   r   r   Úpropertyr'   r   r   r   r   r+   r
   r
   r
   r   r!   A   s   
r!   c               @   s^   e Zd ZdZdZddd„Zdd„ Zedd	„ ƒZd
d„ Z	dd„ Z
dd„ Zdd„ Zedd„ ƒZdS )Ú	RunPythonzT
    Runs Python code in a context suitable for doing versioned ORM operations.
    FNc             C   sV   || _ t|ƒstdƒ‚|| _|d kr,d | _nt|ƒs<tdƒ‚|| _|pHi | _|| _d S )Nz*RunPython must be supplied with a callablez2RunPython must be supplied with callable arguments)ÚatomicÚcallabler3   ÚcodeÚreverse_coder%   r&   )r	   r<   r=   r:   r%   r&   r
   r
   r   r   Ž   s    
zRunPython.__init__c             C   sP   d| j i}| jd k	r| j|d< | jd k	r2| j|d< | jrB| j|d< | jjg |fS )Nr<   r=   r:   r%   )r<   r=   r:   r%   r   r   )r	   r   r
   r
   r   r   ž   s    





zRunPython.deconstructc             C   s
   | j d k	S )N)r=   )r	   r
   r
   r   r'   ®   s    zRunPython.reversiblec             C   s   d S )Nr
   )r	   r   r   r
   r
   r   r   ²   s    zRunPython.state_forwardsc             C   s(   t j|jj|f| jŽr$| j|j|ƒ d S )N)r   r(   r)   r*   r%   r<   Úapps)r	   r   r   r   r   r
   r
   r   r   ·   s    zRunPython.database_forwardsc             C   s:   | j d krtdƒ‚tj|jj|f| jŽr6| j |j|ƒ d S )Nz!You cannot reverse this operation)r=   r,   r   r(   r)   r*   r%   r>   )r	   r   r   r   r   r
   r
   r   r   ¿   s    
zRunPython.database_backwardsc             C   s   dS )NzRaw Python operationr
   )r	   r
   r
   r   r   Å   s    zRunPython.describec             C   s   d S )Nr
   )r>   r   r
   r
   r   r5   È   s    zRunPython.noop)NNNF)r   r   r   r    Zreduces_to_sqlr   r   r8   r'   r   r   r   r   Ústaticmethodr5   r
   r
   r
   r   r9   ‡   s   
r9   N)	Ú
__future__r   Z	django.dbr   Úbaser   r   r!   r9   r
   r
   r
   r   Ú<module>   s
   9F