U
    }®c   ã                   @   sÎ   d dl mZ d dlmZmZ d dlmZ dddddd	d
g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dS )é    )ÚNotSupportedError)ÚFuncÚIndex)Úcached_propertyÚ
BloomIndexÚ	BrinIndexÚ
BTreeIndexÚGinIndexÚ	GistIndexÚ	HashIndexÚSpGistIndexc                       s:   e Zd Zedd„ ƒZd
‡ fdd„	Zdd„ Zdd	„ Z‡  ZS )ÚPostgresIndexc                 C   s   t jtt jƒ t| jƒ S ©N)r   Úmax_name_lengthÚlenÚsuffix©Úself© r   úC/tmp/pip-unpacked-wheel-7m9rsep5/django/contrib/postgres/indexes.pyr      s    zPostgresIndex.max_name_lengthÚ c                    sX   |   |¡ tƒ j||fdd| j i|—Ž}|  ¡ }|rTdd |¡|jd f |jd< |S )NÚusingz	 USING %szWITH (%s) %sz, Úextra)Úcheck_supportedÚsuperÚ
create_sqlr   Úget_with_paramsÚjoinÚparts)r   ÚmodelÚschema_editorr   ÚkwargsZ	statementÚwith_params©Ú	__class__r   r   r      s    
 þzPostgresIndex.create_sqlc                 C   s   d S r   r   ©r   r    r   r   r   r       s    zPostgresIndex.check_supportedc                 C   s   g S r   r   r   r   r   r   r   #   s    zPostgresIndex.get_with_params)r   )	Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   Ú__classcell__r   r   r#   r   r      s
   
r   c                       s<   e Zd ZdZdddœ‡ fdd„
Z‡ fdd„Zd	d
„ Z‡  ZS )r   ZbloomNr   )ÚlengthÚcolumnsc                   s¤   t ƒ j||Ž t| jƒdkr$tdƒ‚t|ttfƒs:tdƒ‚t|ƒt| jƒkrTtdƒ‚tdd„ |D ƒƒsntdƒ‚|d k	r”d|  k rŠd	ks”n td
ƒ‚|| _	|| _
d S )Né    z-Bloom indexes support a maximum of 32 fields.z+BloomIndex.columns must be a list or tuple.z7BloomIndex.columns cannot have more values than fields.c                 s   s&   | ]}d |  k odkn  V  qdS )r   iÿ  Nr   )Ú.0Úcolr   r   r   Ú	<genexpr>4   s     z&BloomIndex.__init__.<locals>.<genexpr>z8BloomIndex.columns must contain integers from 1 to 4095.r   i   z<BloomIndex.length must be None or an integer from 1 to 4096.)r   Ú__init__r   ÚfieldsÚ
ValueErrorÚ
isinstanceÚlistÚtupleÚallr*   r+   )r   r*   r+   Úexpressionsr!   r#   r   r   r0   *   s&    ÿÿÿzBloomIndex.__init__c                    s>   t ƒ  ¡ \}}}| jd k	r$| j|d< | jr4| j|d< |||fS )Nr*   r+   )r   Údeconstructr*   r+   ©r   ÚpathÚargsr!   r#   r   r   r8   ?   s    


zBloomIndex.deconstructc                 C   sF   g }| j d k	r| d| j  ¡ | jrB| dd„ t| jdd�D ƒ¡ |S )Nzlength = %dc                 s   s   | ]\}}d ||f V  qdS )z
col%d = %dNr   )r-   ÚiÚvr   r   r   r/   L   s   ÿz-BloomIndex.get_with_params.<locals>.<genexpr>é   )Ústart)r*   Úappendr+   ÚextendÚ	enumerate©r   r"   r   r   r   r   G   s    

þzBloomIndex.get_with_params©r&   r'   r(   r   r0   r8   r   r)   r   r   r#   r   r   '   s   c                       sD   e Zd ZdZdddœ‡ fdd„
Z‡ fdd„Zdd	„ Zd
d„ Z‡  ZS )r   ZbrinN)ÚautosummarizeÚpages_per_rangec                   s6   |d k	r|dkrt dƒ‚|| _|| _tƒ j||Ž d S )Nr   z2pages_per_range must be None or a positive integer)r2   rE   rF   r   r0   )r   rE   rF   r7   r!   r#   r   r   r0   V   s
    zBrinIndex.__init__c                    sB   t ƒ  ¡ \}}}| jd k	r$| j|d< | jd k	r8| j|d< |||fS )NrE   rF   )r   r8   rE   rF   r9   r#   r   r   r8   ]   s    



zBrinIndex.deconstructc                 C   s   | j r|jjjstdƒ‚d S )Nz2BRIN option autosummarize requires PostgreSQL 10+.)rE   Ú
connectionÚfeaturesZhas_brin_autosummarizer   r%   r   r   r   r   e   s    zBrinIndex.check_supportedc                 C   sD   g }| j d k	r&| d| j rdnd ¡ | jd k	r@| d| j ¡ |S )Nzautosummarize = %sÚonÚoffzpages_per_range = %d)rE   r@   rF   rC   r   r   r   r   i   s    

zBrinIndex.get_with_params)	r&   r'   r(   r   r0   r8   r   r   r)   r   r   r#   r   r   S   s
   c                       s:   e Zd ZdZddœ‡ fdd„
Z‡ fdd„Zdd	„ Z‡  ZS )
r   ZbtreeN©Ú
fillfactorc                   s   || _ tƒ j||Ž d S r   ©rL   r   r0   ©r   rL   r7   r!   r#   r   r   r0   u   s    zBTreeIndex.__init__c                    s.   t ƒ  ¡ \}}}| jd k	r$| j|d< |||fS ©NrL   ©r   r8   rL   r9   r#   r   r   r8   y   s    

zBTreeIndex.deconstructc                 C   s"   g }| j d k	r| d| j  ¡ |S ©Núfillfactor = %d©rL   r@   rC   r   r   r   r      s    
zBTreeIndex.get_with_paramsrD   r   r   r#   r   r   r   s   c                       s<   e Zd ZdZdddœ‡ fdd„
Z‡ fdd„Zdd	„ Z‡  ZS )
r	   ZginN)Ú
fastupdateÚgin_pending_list_limitc                   s   || _ || _tƒ j||Ž d S r   )rT   rU   r   r0   )r   rT   rU   r7   r!   r#   r   r   r0   ‰   s    zGinIndex.__init__c                    sB   t ƒ  ¡ \}}}| jd k	r$| j|d< | jd k	r8| j|d< |||fS )NrT   rU   )r   r8   rT   rU   r9   r#   r   r   r8   Ž   s    



zGinIndex.deconstructc                 C   sD   g }| j d k	r| d| j  ¡ | jd k	r@| d| jr8dnd ¡ |S )Nzgin_pending_list_limit = %dzfastupdate = %srI   rJ   )rU   r@   rT   rC   r   r   r   r   –   s    

zGinIndex.get_with_paramsrD   r   r   r#   r   r	   †   s   c                       sD   e Zd ZdZdddœ‡ fdd„
Z‡ fdd„Zdd	„ Zd
d„ Z‡  ZS )r
   ZgistN)Ú	bufferingrL   c                   s   || _ || _tƒ j||Ž d S r   )rV   rL   r   r0   )r   rV   rL   r7   r!   r#   r   r   r0   ¢   s    zGistIndex.__init__c                    sB   t ƒ  ¡ \}}}| jd k	r$| j|d< | jd k	r8| j|d< |||fS )NrV   rL   )r   r8   rV   rL   r9   r#   r   r   r8   §   s    



zGistIndex.deconstructc                 C   sD   g }| j d k	r&| d| j rdnd ¡ | jd k	r@| d| j ¡ |S )Nzbuffering = %srI   rJ   rR   )rV   r@   rL   rC   r   r   r   r   ¯   s    

zGistIndex.get_with_paramsc                 C   s   | j r|jjjstdƒ‚d S )Nz.Covering GiST indexes requires PostgreSQL 12+.)ÚincluderG   rH   Zsupports_covering_gist_indexesr   r%   r   r   r   r   ·   s    zGistIndex.check_supported)	r&   r'   r(   r   r0   r8   r   r   r)   r   r   r#   r   r
   Ÿ   s
   c                       s:   e Zd ZdZddœ‡ fdd„
Z‡ fdd„Zdd	„ Z‡  ZS )
r   ÚhashNrK   c                   s   || _ tƒ j||Ž d S r   rM   rN   r#   r   r   r0   ¿   s    zHashIndex.__init__c                    s.   t ƒ  ¡ \}}}| jd k	r$| j|d< |||fS rO   rP   r9   r#   r   r   r8   Ã   s    

zHashIndex.deconstructc                 C   s"   g }| j d k	r| d| j  ¡ |S rQ   rS   rC   r   r   r   r   É   s    
zHashIndex.get_with_paramsrD   r   r   r#   r   r   ¼   s   c                       s:   e Zd ZdZddœ‡ fdd„
Z‡ fdd„Zdd	„ Z‡  ZS )
r   ZspgistNrK   c                   s   || _ tƒ j||Ž d S r   rM   rN   r#   r   r   r0   Ó   s    zSpGistIndex.__init__c                    s.   t ƒ  ¡ \}}}| jd k	r$| j|d< |||fS rO   rP   r9   r#   r   r   r8   ×   s    

zSpGistIndex.deconstructc                 C   s"   g }| j d k	r| d| j  ¡ |S rQ   rS   rC   r   r   r   r   Ý   s    
zSpGistIndex.get_with_paramsrD   r   r   r#   r   r   Ð   s   c                       s    e Zd ZdZ‡ fdd„Z‡  ZS )ÚOpClassz%(expressions)s %(name)sc                    s   t ƒ j||d� d S )N)Úname)r   r0   )r   Z
expressionrZ   r#   r   r   r0   ç   s    zOpClass.__init__)r&   r'   r(   Útemplater0   r)   r   r   r#   r   rY   ä   s   rY   N)Z	django.dbr   Zdjango.db.modelsr   r   Zdjango.utils.functionalr   Ú__all__r   r   r   r   r	   r
   r   r   rY   r   r   r   r   Ú<module>   s&        þ,