U
    [¨+dÂ,  ã                   @   s´   d dl Z d dlZd dlmZ d dlmZ d dlmZ d dlm	Z	 d dl
mZ d dlmZ d dlmZmZ d d	lmZ d d
lmZ d dlmZ G dd„ dƒZG dd„ dƒZdZdZdS )é    N)Úimport_module)Úget_version)Úapps)ÚSettingsReference)Ú
migrations)ÚMigrationLoader)Ú
SerializerÚserializer_factory)Úget_func_args)Ú
module_dir)Únowc                   @   s>   e Zd Zddd„Zdd„ Zdd„ Zdd	„ Zd
d„ Zdd„ ZdS )ÚOperationWriteré   c                 C   s   || _ g | _|| _d S ©N)Ú	operationÚbuffÚindentation)Úselfr   r   © r   ú?/tmp/pip-unpacked-wheel-n7e__lmp/django/db/migrations/writer.pyÚ__init__   s    zOperationWriter.__init__c           
         s  ‡ ‡fdd„}t ƒ ‰ ˆj ¡ \}}}tˆjjƒ}tt|d ƒˆjjkrTˆ d| ¡ n,ˆ  	dˆjjj
 ¡ ˆ dˆjjj
|f ¡ ˆ ¡  t|ƒD ]\}}|}|| }	||	|ƒ qt|ƒ}||d … D ]}	|	|krÄ||	 }||	|ƒ qÄˆ ¡  ˆ d¡ ˆ ¡ ˆ fS )Nc                    s4  | ˆj jkr¨t|tttfƒr¨t|tƒrˆ d|  ¡ ˆ ¡  | ¡ D ]¤\}}t	 
|¡\}}t	 
|¡\}}| ¡ }t|ƒdkrÈˆ d||d f ¡ |dd… D ]}	ˆ |	¡ q¤ˆ d|d  ¡ nˆ d||f ¡ ˆ  |¡ ˆ  |¡ qJˆ ¡  ˆ d¡ n¢ˆ d	|  ¡ ˆ ¡  |D ]t}
t	 
|
¡\}}| ¡ }t|ƒdkrx|d d… D ]}	ˆ |	¡ qRˆ d|d  ¡ nˆ d| ¡ ˆ  |¡ qˆ ¡  ˆ d
¡ nˆt	 
|¡\}}| ¡ }t|ƒdkrˆ d| |d f ¡ |dd… D ]}	ˆ |	¡ qîˆ d|d  ¡ nˆ d| |f ¡ ˆ  |¡ d S )Nz%s={é   z%s: %sr   éÿÿÿÿz%s,z%s: %s,z},z%s=[z],z%s=%sz%s=%s,)r   Zserialization_expand_argsÚ
isinstanceÚlistÚtupleÚdictÚfeedÚindentÚitemsÚMigrationWriterÚ	serializeÚ
splitlinesÚlenÚupdateÚunindent)Z	_arg_nameZ
_arg_valueÚkeyÚvalueZ
key_stringZkey_importsÚ
arg_stringZarg_importsÚargsÚargÚitem©Úimportsr   r   r   Ú_write   sV     ÿ
z)OperationWriter.serialize.<locals>._writezmigrations.%s(z	import %sz%s.%s(z),)Úsetr   Zdeconstructr
   r   Úgetattrr   Ú	__class__r   ÚaddÚ
__module__r   Ú	enumerater#   r%   Úrender)
r   r.   Únamer)   ÚkwargsZoperation_argsÚir*   Ú	arg_valueZarg_namer   r,   r   r!      s*    1
zOperationWriter.serializec                 C   s   |  j d7  _ d S ©Nr   ©r   ©r   r   r   r   r   i   s    zOperationWriter.indentc                 C   s   |  j d8  _ d S r:   r;   r<   r   r   r   r%   l   s    zOperationWriter.unindentc                 C   s   | j  d| jd  | ¡ d S )Nú é   )r   Úappendr   )r   Úliner   r   r   r   o   s    zOperationWriter.feedc                 C   s   d  | j¡S )NÚ
)Újoinr   r<   r   r   r   r5   r   s    zOperationWriter.renderN)r   )	Ú__name__r3   Ú__qualname__r   r!   r   r%   r   r5   r   r   r   r   r      s   
Qr   c                   @   sj   e Zd ZdZddd„Zdd„ Zedd„ ƒZed	d
„ ƒZedd„ ƒZ	e
dd„ ƒZe
dd„ ƒZe
dd„ ƒZdS )r    zj
    Take a Migration instance and is able to produce the contents
    of the migration file from it.
    Tc                 C   s   || _ || _d| _d S )NF)Ú	migrationÚinclude_headerÚneeds_manual_porting)r   rE   rF   r   r   r   r   |   s    zMigrationWriter.__init__c                 C   sü  dddœ}t ƒ }g }| jjD ](}t|ƒ ¡ \}}| |¡ | |¡ q|rXd |¡d nd|d< g }| jjD ]F}|d dkrš| d|d  ¡ | 	d	¡ ql| d
|  |¡d  ¡ ql|rÆd |¡d nd|d< t ƒ }	t
|ƒD ]8}
t d|
¡rÜ|	 	|
 d¡d  ¡ ¡ | |
¡ d| _qÜd|kr6| d¡ | 	d¡ n
| 	d¡ t|dd„ d}|rdd |¡d nd|d< |	r|d  dd t|	ƒ¡ 7  < | jjr´d|  | jj¡d  |d< | jrÚttƒ tƒ  d¡dœ |d< nd|d< | jjrôd|d< t| S )z%Return a string of the file contents.Ú )Úreplaces_strÚinitial_strrA   Ú
operationsr   Z__setting__z5        migrations.swappable_dependency(settings.%s),r   z from django.conf import settingsz        %s,Údependenciesz^import (.*)\.\d+[^\s]*$ÚimportTzfrom django.db import modelsz(from django.db import migrations, modelsz from django.db import migrationsc                 S   s   |   ¡ d S r:   )Úsplit)r8   r   r   r   Ú<lambda>²   ó    z+MigrationWriter.as_string.<locals>.<lambda>)r&   r-   z½

# Functions from the following migrations need manual copying.
# Move them and any dependencies into this file, then update the
# RunPython operations to refer to the local versions:
# %sz
# z
    replaces = %s
rI   z%Y-%m-%d %H:%M)ÚversionÚ	timestampZmigration_headerz
    initial = True
rJ   )r/   rE   rK   r   r!   r$   r?   rB   rL   r2   r   ÚreÚmatchrN   ÚstripÚremoverG   ÚdiscardÚsortedZreplacesrF   ÚMIGRATION_HEADER_TEMPLATEr   r   ÚstrftimeÚinitialÚMIGRATION_TEMPLATE)r   r   r-   rK   r   Zoperation_stringZoperation_importsrL   Ú
dependencyZmigration_importsr@   Zsorted_importsr   r   r   Ú	as_string   sd    þ
ÿÿ



û
ÿ
þ
zMigrationWriter.as_stringc              
   C   s–  t  | jj¡\}}|d kr*td| jj ƒ‚zt|ƒ}W n tk
rJ   Y n"X z
t|ƒW S  tk
rj   Y nX t 	| jj¡}| 
d¡\}}}|j|kr¤tj |j|¡S | d¡g  }}|r,| d| d¡¡ ztd |¡ƒ}	W n ttfk
 rü   Y q´Y q´X zt|	ƒ}
W n tk
r$   Y q´Y q´X q8q´td| ƒ‚tjj|
f|žŽ }tj|dd |D ]4}tj |
|¡}
ttj |
d¡d	ƒ W 5 Q R X q\|S )
NztDjango can't create migrations for app '%s' because migrations have been disabled via the MIGRATION_MODULES setting.Ú.r   r   z„Could not locate an appropriate location to create migrations package %s. Make sure the toplevel package exists and can be imported.T)Úexist_okz__init__.pyÚw)r   Úmigrations_modulerE   Z	app_labelÚ
ValueErrorr   ÚImportErrorr   r   Zget_app_configÚ
rpartitionr6   ÚosÚpathrB   rN   ÚinsertÚpopÚmakedirsÚopen)r   Zmigrations_package_nameÚ_rb   Z
app_configZmaybe_app_nameZmigrations_package_basenameZexisting_dirsZmissing_dirsÚbase_moduleÚbase_dirZ	final_dirZmissing_dirr   r   r   ÚbasedirÎ   sb    ÿþÿ
ü


þÿzMigrationWriter.basedirc                 C   s   d| j j S )Nz%s.py)rE   r6   r<   r   r   r   Úfilename  s    zMigrationWriter.filenamec                 C   s   t j | j| j¡S r   )rf   rg   rB   ro   rp   r<   r   r   r   rg     s    zMigrationWriter.pathc                 C   s   t |ƒ ¡ S r   )r	   r!   )Úclsr'   r   r   r   r!     s    zMigrationWriter.serializec                 C   s   t  ||¡ d S r   )r   Úregister)rq   Útype_Ú
serializerr   r   r   Úregister_serializer  s    z#MigrationWriter.register_serializerc                 C   s   t  |¡ d S r   )r   Ú
unregister)rq   rs   r   r   r   Úunregister_serializer   s    z%MigrationWriter.unregister_serializerN)T)rC   r3   rD   Ú__doc__r   r^   Úpropertyro   rp   rg   Úclassmethodr!   ru   rw   r   r   r   r   r    v   s   
M
A



r    z4# Generated by Django %(version)s on %(timestamp)s

z»%(migration_header)s%(imports)s

class Migration(migrations.Migration):
%(replaces_str)s%(initial_str)s
    dependencies = [
%(dependencies)s    ]

    operations = [
%(operations)s    ]
)rf   rS   Ú	importlibr   Zdjangor   Zdjango.appsr   Zdjango.confr   Z	django.dbr   Zdjango.db.migrations.loaderr   Zdjango.db.migrations.serializerr   r	   Zdjango.utils.inspectr
   Zdjango.utils.module_loadingr   Zdjango.utils.timezoner   r   r    rY   r\   r   r   r   r   Ú<module>   s    d 0