U
    ~fhV                     @   s   d dl mZ d dlmZmZmZmZ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 dd
lmZ ddlmZmZmZ ddlmZm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 )    )safe_string)TypedIntegerBoolStringSequence)ExtensionList)Serialisable   )PatternFillFill)Font)Border)	Alignment)
Protection)NumberFormatDescriptorBUILTIN_FORMATS_MAX_SIZEBUILTIN_FORMATS_REVERSE)
StyleArray	CellStylec                	       s   e Zd ZdZeedZeedZee	dZ
eedZe ZeedZeddZeddZe ZdZe Zddd	Z fd
dZdd Zdd Zdd Zdd Zdd Z dd Z!  Z"S )
NamedStylez#
    Named and editable styles
    expected_typeT
allow_noneNNormalFc
           
      C   sf   || _ |pt | _|pt | _|p&t | _|p2t | _|| _	|pDt
 | _|| _|	| _d | _t | _d S N)namer   fontr   fillr   borderr   	alignmentnumber_formatr   
protection	builtinIdhidden_wbr   _style)
selfr   r   r   r    r!   r"   r#   r$   r%    r)   @/tmp/pip-unpacked-wheel-eyn_au5k/openpyxl/styles/named_styles.py__init__2   s    zNamedStyle.__init__c                    s.   t  || t| dd r*|dkr*|   d S )Nr&   )r   r   r    r!   r"   r#   )super__setattr__getattr_recalculate)r(   attrvalue	__class__r)   r*   r-   J   s    zNamedStyle.__setattr__c                 c   s0   dD ]&}t | |d }|d k	r|t|fV  qd S )Nr   r$   r%   xfId)r.   r   )r(   keyr1   r)   r)   r*   __iter__R   s    zNamedStyle.__iter__c                 C   s   || _ |   dS )z2
        Bind a named style to a workbook
        N)r&   r/   )r(   wbr)   r)   r*   bindY   s    zNamedStyle.bindc                 C   s   | j j| j| j_| j j| j| j_| j j	| j
| j_| j j| j| j_| j j| j| j_| j}|tkr|t| }n| j j| jt }|| j_d S r   )r&   Z_fontsaddr   r'   ZfontIdZ_bordersr    ZborderIdZ_fillsr   ZfillIdZ_protectionsr#   ZprotectionIdZ_alignmentsr!   ZalignmentIdr"   r   Z_number_formatsr   ZnumFmtId)r(   fmtr)   r)   r*   r/   a   s    
zNamedStyle._recalculatec                 C   s   | j S )z3Return a style array representing the current style)r'   r(   r)   r)   r*   as_tuplep   s    zNamedStyle.as_tuplec                 C   sJ   t | j}d|_d|_d|_| jt kr2| j|_| jt	 krF| j|_|S )z+
        Return equivalent XfStyle
        N)
r   Z
from_arrayr'   r5   ZpivotButtonZquotePrefixr!   r   r#   r   )r(   Zxfr)   r)   r*   as_xfu   s    zNamedStyle.as_xfc                 C   s   t | j| j| j| jjd}|S )z.
        Return relevant named style

        r4   )_NamedCellStyler   r$   r%   r'   r5   )r(   namedr)   r)   r*   as_name   s    zNamedStyle.as_name)	r   NNNNNNNF)#__name__
__module____qualname____doc__r   r   r   r   r   r   r    r   r!   r   r"   r   r#   r   r$   r   r%   r   r   r&   r   r'   r+   r-   r7   r9   r/   r=   r>   rA   __classcell__r)   r)   r2   r*   r      s:   






         
r   c                       sF   e Zd ZdZd fdd	Zedd Z fddZ fd	d
Z  Z	S )NamedStyleListz
    Named styles are editable and can be applied to multiple objects

    As only the index is stored in referencing objects the order mus
    be preserved.

    Returns a list of NamedStyles
    r)   c                    s0   t |t| D ]\}}||j_qt | dS )zN
        Allow a list of named styles to be passed in and index them.
        N)	enumeratelenr'   r5   r,   r+   )r(   iterableidxsr2   r)   r*   r+      s    
zNamedStyleList.__init__c                 C   s   dd | D S )Nc                 S   s   g | ]
}|j qS r)   )r   ).0rL   r)   r)   r*   
<listcomp>   s     z(NamedStyleList.names.<locals>.<listcomp>r)   r<   r)   r)   r*   names   s    zNamedStyleList.namesc                    sP   t |trt |S t| jD ]\}}||kr | |   S q td|d S )Nz&No named style with the name{0} exists)
isinstanceintr,   __getitem__rH   rO   KeyErrorformat)r(   r6   rK   r   r2   r)   r*   rR      s    
zNamedStyleList.__getitem__c                    sL   t |tstdn|j| jkr0td|jt| |j_	t
 | d S )Nz&Only NamedStyle instances can be addedzStyle {0} exists already)rP   r   	TypeErrorr   rO   
ValueErrorrT   rI   r'   r5   r,   append)r(   styler2   r)   r*   rW      s    

zNamedStyleList.append)r)   )
rB   rC   rD   rE   r+   propertyrO   rR   rW   rF   r)   r)   r2   r*   rG      s   	

rG   c                   @   sb   e Zd ZdZdZe Ze ZeddZ	eddZ
eddZeddZeeddZdZd
dd	ZdS )r?   z
    Pointer-based representation of named styles in XML
    xfId refers to the corresponding CellStyleXfs

    Not used in client code.
    	cellStyleTr   )r   r   r)   Nc                 C   s(   || _ || _|| _|| _|| _|| _d S r   )r   r5   r$   iLevelr%   customBuiltin)r(   r   r5   r$   r[   r%   r\   extLstr)   r)   r*   r+      s    	z_NamedCellStyle.__init__)NNNNNNN)rB   rC   rD   rE   tagnamer   r   r   r5   r$   r[   r   r%   r\   r   r   r]   Z__elements__r+   r)   r)   r)   r*   r?      s$   



       r?   c                   @   sJ   e Zd ZdZdZeddZeedZ	dZ
dd	d
Zedd Zdd ZdS )_NamedCellStyleListzM
    Container for named cell style objects

    Not used in client code
    Z
cellStylesTr   r   )countNr)   c                 C   s
   || _ d S r   )rZ   )r(   r`   rZ   r)   r)   r*   r+      s    z_NamedCellStyleList.__init__c                 C   s
   t | jS r   )rI   rZ   r<   r)   r)   r*   r`      s    z_NamedCellStyleList.countc                 C   sh   dd }g }t  }t  }t| j|dD ]<}|j|ks&|j|kr@q&||j ||j || q&|S )a  
        Some applications contain duplicate definitions either by name or
        referenced style.

        As the references are 0-based indices, styles are sorted by
        index.

        Returns a list of style references with duplicates removed
        c                 S   s   | j S r   )r5   )vr)   r)   r*   sort_fn  s    z6_NamedCellStyleList.remove_duplicates.<locals>.sort_fn)r6   )setsortedrZ   r5   r   r:   rW   )r(   rb   ZstylesrO   idsnsr)   r)   r*   remove_duplicates   s    z%_NamedCellStyleList.remove_duplicates)Nr)   )rB   rC   rD   rE   r^   r   r`   r   r?   rZ   	__attrs__r+   rY   rg   r)   r)   r)   r*   r_      s   

  

r_   N)#Zopenpyxl.compatr   Zopenpyxl.descriptorsr   r   r   r   r   Zopenpyxl.descriptors.excelr   Z!openpyxl.descriptors.serialisabler	   Zfillsr   r   Zfontsr   Zbordersr   r!   r   r#   r   Znumbersr   r   r   Z
cell_styler   r   r   listrG   r?   r_   r)   r)   r)   r*   <module>   s   s-'