U
    mÇf_3  ã                   @   sx  d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl	Z	d dl
Z
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mZ d dlmZmZ d dlmZ G dd	„ d	ƒZG d
d„ deƒZG dd„ deƒZG dd„ deƒZG dd„ deƒZG dd„ deƒZ G dd„ deƒZ!G dd„ deƒZ"G dd„ deƒZ#G dd„ deƒZ$G dd„ deƒZ%G dd„ deƒZ&G d d!„ d!eƒZ'G d"d#„ d#eƒZ(G d$d%„ d%eƒZ)G d&d'„ d'e"ƒZ*G d(d)„ d)e"ƒZ+G d*d+„ d+eƒZ,G d,d-„ d-eƒZ-G d.d/„ d/eƒZ.G d0d1„ d1eƒZ/G d2d3„ d3eƒZ0G d4d5„ d5eƒZ1G d6d7„ d7eƒZ2G d8d9„ d9eƒZ3G d:d;„ d;eƒZ4G d<d=„ d=eƒZ5G d>d?„ d?ƒZ6d@dA„ Z7dS )Bé    N)ÚSettingsReference)Úmodels)Ú	Operation)ÚCOMPILED_REGEX_TYPEÚRegexObject)Ú
LazyObjectÚPromise)Úget_docs_versionc                   @   s   e Zd Zdd„ Zdd„ ZdS )ÚBaseSerializerc                 C   s
   || _ d S ©N©Úvalue)Úselfr   © r   úC/tmp/pip-unpacked-wheel-siwebuq3/django/db/migrations/serializer.pyÚ__init__   s    zBaseSerializer.__init__c                 C   s   t dƒ‚d S )NzCSubclasses of BaseSerializer must implement the serialize() method.©ÚNotImplementedError©r   r   r   r   Ú	serialize   s    ÿzBaseSerializer.serializeN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r   r
      s   r
   c                   @   s   e Zd Zdd„ Zdd„ ZdS )ÚBaseSequenceSerializerc                 C   s   t dƒ‚d S )NzISubclasses of BaseSequenceSerializer must implement the _format() method.r   r   r   r   r   Ú_format!   s    ÿzBaseSequenceSerializer._formatc                 C   sT   t ƒ }g }| jD ](}t|ƒ ¡ \}}| |¡ | |¡ q|  ¡ }|d |¡ |fS )Nú, )Úsetr   Úserializer_factoryr   ÚupdateÚappendr   Újoin©r   ÚimportsÚstringsÚitemZitem_stringZitem_importsr   r   r   r   r   &   s    

z BaseSequenceSerializer.serializeN)r   r   r   r   r   r   r   r   r   r       s   r   c                   @   s   e Zd Zdd„ ZdS )ÚBaseSimpleSerializerc                 C   s   t | jƒtƒ fS r   )Úreprr   r   r   r   r   r   r   2   s    zBaseSimpleSerializer.serializeN©r   r   r   r   r   r   r   r   r%   1   s   r%   c                   @   s   e Zd Zdd„ ZdS )ÚChoicesSerializerc                 C   s   t | jjƒ ¡ S r   )r   r   r   r   r   r   r   r   7   s    zChoicesSerializer.serializeNr'   r   r   r   r   r(   6   s   r(   c                   @   s   e Zd ZdZdd„ ZdS )ÚDateTimeSerializerz)For datetime.*, except datetime.datetime.c                 C   s   t | jƒdhfS ©Nzimport datetime©r&   r   r   r   r   r   r   >   s    zDateTimeSerializer.serializeN©r   r   r   Ú__doc__r   r   r   r   r   r)   ;   s   r)   c                   @   s   e Zd ZdZdd„ ZdS )ÚDatetimeDatetimeSerializerzFor datetime.datetime.c                 C   sF   | j jd k	r.| j jtjjkr.| j  tjj¡| _ dg}t| j ƒt|ƒfS r*   )r   ÚtzinfoÚdatetimeÚtimezoneÚutcÚ
astimezoner&   r   )r   r"   r   r   r   r   E   s    z$DatetimeDatetimeSerializer.serializeNr,   r   r   r   r   r.   B   s   r.   c                   @   s   e Zd Zdd„ ZdS )ÚDecimalSerializerc                 C   s   t | jƒdhfS )Nzfrom decimal import Decimalr+   r   r   r   r   r   M   s    zDecimalSerializer.serializeNr'   r   r   r   r   r4   L   s   r4   c                   @   s,   e Zd Zedd„ ƒZedd„ ƒZdd„ ZdS )ÚDeconstructableSerializerc           
      C   s˜   t  | ¡\}}g }|D ](}t|ƒ ¡ \}}| |¡ | |¡ qt| ¡ ƒD ]4\}	}t|ƒ ¡ \}}| |¡ | d|	|f ¡ qLd|d |¡f |fS )Nz%s=%sz%s(%s)r   )	r5   Ú_serialize_pathr   r   r   r   ÚsortedÚitemsr    )
ÚpathÚargsÚkwargsÚnamer"   r#   ÚargÚ
arg_stringZarg_importsÚkwr   r   r   Úserialize_deconstructedR   s    

z1DeconstructableSerializer.serialize_deconstructedc                 C   s>   |   dd¡\}}|dkr(dh}d| }nd| h}| }||fS )NÚ.é   zdjango.db.modelsúfrom django.db import modelsz	models.%sú	import %s)Úrsplit)r9   Úmoduler<   r"   r   r   r   r6   `   s    

z)DeconstructableSerializer._serialize_pathc                 C   s   | j | j ¡ Ž S r   )r@   r   Údeconstructr   r   r   r   r   k   s    z#DeconstructableSerializer.serializeN)r   r   r   Ústaticmethodr@   r6   r   r   r   r   r   r5   Q   s
   


r5   c                   @   s   e Zd Zdd„ ZdS )ÚDictionarySerializerc           	      C   s€   t ƒ }g }t| j ¡ ƒD ]J\}}t|ƒ ¡ \}}t|ƒ ¡ \}}| |¡ | |¡ | ||f¡ qdd dd„ |D ƒ¡ |fS )Nú{%s}r   c                 s   s   | ]\}}d ||f V  qdS )z%s: %sNr   )Ú.0ÚkÚvr   r   r   Ú	<genexpr>y   s     z1DictionarySerializer.serialize.<locals>.<genexpr>)	r   r7   r   r8   r   r   r   r   r    )	r   r"   r#   rL   rM   Zk_stringZ	k_importsZv_stringZ	v_importsr   r   r   r   p   s    

zDictionarySerializer.serializeNr'   r   r   r   r   rI   o   s   rI   c                   @   s   e Zd Zdd„ ZdS )ÚEnumSerializerc                 C   s,   | j j}|j}d||j| j jf d| hfS )Nz	%s.%s[%r]rD   )r   Ú	__class__r   r   r<   )r   Ú
enum_classrF   r   r   r   r   }   s
    þzEnumSerializer.serializeNr'   r   r   r   r   rO   |   s   rO   c                       s   e Zd Z‡ fdd„Z‡  ZS )ÚFloatSerializerc                    s4   t  | j¡st  | j¡r*d | j¡tƒ fS tƒ  ¡ S )Nzfloat("{}"))ÚmathÚisnanr   ÚisinfÚformatr   Úsuperr   r   ©rP   r   r   r   ‡   s    zFloatSerializer.serialize)r   r   r   r   Ú__classcell__r   r   rX   r   rR   †   s   rR   c                   @   s   e Zd Zdd„ ZdS )ÚFrozensetSerializerc                 C   s   dS )Nzfrozenset([%s])r   r   r   r   r   r   Ž   s    zFrozensetSerializer._formatN©r   r   r   r   r   r   r   r   rZ      s   rZ   c                   @   s   e Zd Zdd„ ZdS )ÚFunctionTypeSerializerc                 C   sÀ   t | jdd ƒrHt| jjtƒrH| jj}|j}d||j| jjf d| hfS | jjdkr\tdƒ‚| jjd krvtd| j ƒ‚| jj}d| jjkr¨d|| jjf d| jj hfS td	| jj|f ƒ‚d S )
NÚ__self__z%s.%s.%srD   z<lambda>z!Cannot serialize function: lambdaz'Cannot serialize function %r: No moduleú<ú%s.%sz"Could not find function %s in %s.
)	Úgetattrr   Ú
isinstancer]   Útyper   r   Ú
ValueErrorr   )r   ÚklassrF   Úmodule_namer   r   r   r   “   s*     ÿÿ
ÿÿz FunctionTypeSerializer.serializeNr'   r   r   r   r   r\   ’   s   r\   c                   @   s   e Zd Zdd„ ZdS )ÚFunctoolsPartialSerializerc                 C   sd   t | jjƒ ¡ \}}t | jjƒ ¡ \}}t | jjƒ ¡ \}}dh|||™}d| jjj|||f |fS )Nzimport functoolszfunctools.%s(%s, *%s, **%s))r   r   Úfuncr   r:   ÚkeywordsrP   r   )r   Zfunc_stringZfunc_importsÚargs_stringZargs_importsZkeywords_stringZkeywords_importsr"   r   r   r   r   ¯   s    ÿüÿøz$FunctoolsPartialSerializer.serializeNr'   r   r   r   r   rf   ®   s   rf   c                   @   s   e Zd Zdd„ ZdS )ÚIterableSerializerc                 C   s`   t ƒ }g }| jD ](}t|ƒ ¡ \}}| |¡ | |¡ qt|ƒdkrJdnd}|d |¡ |fS )NrB   ú(%s)ú(%s,)r   )r   r   r   r   r   r   Úlenr    r!   r   r   r   r   Å   s    

zIterableSerializer.serializeNr'   r   r   r   r   rj   Ä   s   rj   c                   @   s   e Zd Zdd„ ZdS )ÚModelFieldSerializerc                 C   s    | j  ¡ \}}}}|  |||¡S r   )r   rG   r@   )r   Ú	attr_namer9   r:   r;   r   r   r   r   Ó   s    zModelFieldSerializer.serializeNr'   r   r   r   r   rn   Ò   s   rn   c                   @   s   e Zd Zdd„ ZdS )ÚModelManagerSerializerc                 C   sD   | j  ¡ \}}}}}|r2|  |¡\}}d| |fS |  |||¡S d S )Nz%s.as_manager())r   rG   r6   r@   )r   Z
as_managerZmanager_pathZqs_pathr:   r;   r<   r"   r   r   r   r   Ù   s
    z ModelManagerSerializer.serializeNr'   r   r   r   r   rp   Ø   s   rp   c                   @   s   e Zd Zdd„ ZdS )ÚOperationSerializerc                 C   s0   ddl m} || jdd ¡ \}}| d¡|fS )Nr   )ÚOperationWriter)Úindentationú,)Zdjango.db.migrations.writerrr   r   r   Úrstrip)r   rr   Ústringr"   r   r   r   r   ã   s    zOperationSerializer.serializeNr'   r   r   r   r   rq   â   s   rq   c                   @   s   e Zd Zdd„ ZdS )ÚPathLikeSerializerc                 C   s   t t | j¡ƒi fS r   )r&   ÚosÚfspathr   r   r   r   r   r   ì   s    zPathLikeSerializer.serializeNr'   r   r   r   r   rw   ë   s   rw   c                   @   s   e Zd Zdd„ ZdS )ÚPathSerializerc                 C   s*   t | jtjƒrdnd}d|| jf dhfS )NZPureÚ zpathlib.%s%rzimport pathlib)ra   r   ÚpathlibÚPath)r   Úprefixr   r   r   r   ñ   s    zPathSerializer.serializeNr'   r   r   r   r   rz   ð   s   rz   c                   @   s   e Zd Zdd„ ZdS )ÚRegexSerializerc                 C   sj   t | jjƒ ¡ \}}| jjt d¡jA }t |ƒ ¡ \}}dh||™}|g}|rX| |¡ dd |¡ |fS )Nr{   z	import rezre.compile(%s)r   )	r   r   Úpatternr   ÚflagsÚreÚcompiler   r    )r   Zregex_patternZpattern_importsr   Zregex_flagsZflag_importsr"   r:   r   r   r   r   ù   s    ÿ
zRegexSerializer.serializeNr'   r   r   r   r   r   ø   s   r   c                   @   s   e Zd Zdd„ ZdS )ÚSequenceSerializerc                 C   s   dS )Nz[%s]r   r   r   r   r   r   	  s    zSequenceSerializer._formatNr[   r   r   r   r   r„     s   r„   c                   @   s   e Zd Zdd„ ZdS )ÚSetSerializerc                 C   s   | j r
dS dS )NrJ   zset(%s)r   r   r   r   r   r     s    zSetSerializer._formatNr[   r   r   r   r   r…     s   r…   c                   @   s   e Zd Zdd„ ZdS )ÚSettingsReferenceSerializerc                 C   s   d| j j dhfS )Nzsettings.%sz from django.conf import settings)r   Zsetting_namer   r   r   r   r     s    
ÿz%SettingsReferenceSerializer.serializeNr'   r   r   r   r   r†     s   r†   c                   @   s   e Zd Zdd„ ZdS )ÚTupleSerializerc                 C   s   t | jƒdkrdS dS )NrB   rk   rl   )rm   r   r   r   r   r   r     s    zTupleSerializer._formatNr[   r   r   r   r   r‡     s   r‡   c                   @   s   e Zd Zdd„ ZdS )ÚTypeSerializerc                 C   s   t jddgftd ƒdg fg}|D ]$\}}}|| jkr |t|ƒf  S q t| jdƒrŒ| jj}|tjkrr| jjtƒ fS d|| jj	f d| hfS d S )Nzmodels.ModelrC   z
type(None)r   r_   rD   )
r   ZModelrb   r   r   Úhasattrr   Úbuiltinsr   r   )r   Zspecial_casesZcaserv   r"   rF   r   r   r   r   #  s    þ

ÿzTypeSerializer.serializeNr'   r   r   r   r   rˆ   "  s   rˆ   c                   @   s   e Zd Zdd„ ZdS )ÚUUIDSerializerc                 C   s   dt | jƒ dhfS )Nzuuid.%szimport uuidr+   r   r   r   r   r   6  s    zUUIDSerializer.serializeNr'   r   r   r   r   r‹   5  s   r‹   c                (   @   s¾   e Zd Zeeeeeee	e
eeejeejeejeejejejfeeeeeeeedƒe e!e"fe#e$j%e&e'j(e'j)fe*e+j,e+j-e+j.fe/e0j1j2e3e4e5fe6e7j8e9e:j;e<e=j>e?iZ@eAdd„ ƒZBeAdd„ ƒZCdS )Ú
SerializerNc                 C   s&   t |tƒstd|j ƒ‚|| j|< d S )Nz('%s' must inherit from 'BaseSerializer'.)Ú
issubclassr
   rc   r   Ú	_registry)ÚclsÚtype_Ú
serializerr   r   r   ÚregisterW  s
    
ÿzSerializer.registerc                 C   s   | j  |¡ d S r   )rŽ   Úpop)r   r   r   r   r   Ú
unregister_  s    zSerializer.unregister)Dr   r   r   Ú	frozensetrZ   Úlistr„   r   r…   Útupler‡   ÚdictrI   r   ZChoicesr(   ÚenumÚEnumrO   r0   r.   ÚdateÚ	timedeltaÚtimer)   r   r†   ÚfloatrR   ÚboolÚintrb   ÚbytesÚstrÚranger%   ÚdecimalÚDecimalr4   Ú	functoolsÚpartialÚpartialmethodrf   ÚtypesÚFunctionTypeÚBuiltinFunctionTypeÚ
MethodTyper\   ÚcollectionsÚabcÚIterablerj   r   r   r   ÚuuidÚUUIDr‹   r|   ÚPurePathrz   rx   ÚPathLikerw   rŽ   Úclassmethodr’   r”   r   r   r   r   rŒ   :  s`                
 ý     ç
rŒ   c                 C   sÎ   t | tƒrt| ƒ} nt | tƒr.|  ¡ d d } t | tjƒrBt| ƒS t | tjj	ƒrXt
| ƒS t | tƒrjt| ƒS t | tƒr|t| ƒS t| dƒrŽt| ƒS tj ¡ D ]\}}t | |ƒr˜|| ƒ  S q˜td| tƒ f ƒ‚d S )NrB   r   rG   z´Cannot serialize: %r
There are some values Django cannot serialize into migration files.
For more, see https://docs.djangoproject.com/en/%s/topics/migrations/#migration-serializing)ra   r   r¢   r   Ú
__reduce__r   ZFieldrn   ÚmanagerZBaseManagerrp   r   rq   rb   rˆ   r‰   r5   rŒ   rŽ   r8   rc   r	   )r   r   Zserializer_clsr   r   r   r   d  s,    






þÿr   )8rŠ   Úcollections.abcr­   r0   r¤   r™   r¦   rS   rx   r|   r‚   r©   r°   Zdjango.confr   Z	django.dbr   Z$django.db.migrations.operations.baser   Zdjango.db.migrations.utilsr   r   Zdjango.utils.functionalr   r   Zdjango.utils.versionr	   r
   r   r%   r(   r)   r.   r4   r5   rI   rO   rR   rZ   r\   rf   rj   rn   rp   rq   rw   rz   r   r„   r…   r†   r‡   rˆ   r‹   rŒ   r   r   r   r   r   Ú<module>   s\   



	*