
    OwgM                        d dl mZ d dlZd dlmZmZm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 erd d	lmZ d d
lmZ d dlmZmZmZmZmZmZ  G d de      Z  G d ded         Z!y)    )annotationsN)TYPE_CHECKINGAnycast)import_optional_dependency)doc)_shared_docs)BaseExcelReaderExcelWriter)combine_kwargsvalidate_freeze_panesWorkbook)Serialisable)ExcelWriterIfSheetExistsFilePath
ReadBufferScalarStorageOptionsWriteExcelBufferc                  N    e Zd ZdZdZ	 	 	 	 	 	 	 d	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 d fdZedd       Zedd       ZddZ	e
dd       Ze
d        Ze
d	        Ze
d
        Ze
dd       Ze
d        Ze
d        Ze
d        Ze
d        Ze
d        Z	 	 	 	 d	 	 	 	 	 	 	 	 	 ddZ xZS )OpenpyxlWriteropenpyxl)z.xlsxz.xlsmc	                   ddl m}
 t        ||	      }t        |   |||||       d| j
                  v rOddlm} 	  || j                  j                  fi || _
        | j                  j                  j                  d       y 	  |
di || _
        | j                  j                  r3| j                  j!                  | j                  j                  d          y y # t        $ r& | j                  j                  j                           w xY w# t        $ r& | j                  j                  j                           w xY w)Nr   r   )modestorage_optionsif_sheet_existsengine_kwargsr+load_workbook )openpyxl.workbookr   r   super__init___moder   r!   _handleshandle_book	TypeErrorcloseseekbook
worksheetsremove)selfpathenginedate_formatdatetime_formatr   r   r   r   kwargsr   r!   	__class__s               P/var/www/horilla/myenv/lib/python3.12/site-packages/pandas/io/excel/_openpyxl.pyr%   zOpenpyxlWriter.__init__,   s    	/&}f=++' 	 	
 4::.*4==+?+?Q=Q
 MM  %%a(%66

 yy##		  !5!5a!89 $  $$**,  $$**,s   "C D /D/D>c                    | j                   S )z
        Book instance of class openpyxl.workbook.Workbook.

        This attribute can be used to access engine-specific features.
        )r)   r0   s    r7   r-   zOpenpyxlWriter.book[   s     zz    c                r    | j                   j                  D ci c]  }|| j                   |    }}|S c c}w )z(Mapping of sheet names to sheet objects.)r-   
sheetnames)r0   nameresults      r7   sheetszOpenpyxlWriter.sheetsd   s9     59II4H4HID$		$'II Js   4c                &   | j                   j                  | j                  j                         d| j                  v rTt        | j                  j                  t        j                        s%| j                  j                  j                          yyy)z(
        Save workbook to disk.
        r   N)r-   saver'   r(   r&   
isinstancemmaptruncater9   s    r7   _savezOpenpyxlWriter._savej   s`     			t}}++,4::j1E1Etyy&QMM  ))+ 'Rr:   c                    ddi}i }|j                         D ]8  \  }}|j                  ||      }t        | d| d       } ||      }|s4|||<   : |S )a  
        Convert a style_dict to a set of kwargs suitable for initializing
        or updating-on-copy an openpyxl v2 style object.

        Parameters
        ----------
        style_dict : dict
            A dict with zero or more of the following keys (or their synonyms).
                'font'
                'fill'
                'border' ('borders')
                'alignment'
                'number_format'
                'protection'

        Returns
        -------
        style_kwargs : dict
            A dict with the same, normalized keys as ``style_dict`` but each
            value has been replaced with a native openpyxl style object of the
            appropriate class.
        bordersborder_convert_to_c                     y Nr"   )xs    r7   <lambda>z9OpenpyxlWriter._convert_to_style_kwargs.<locals>.<lambda>   s    r:   )itemsgetgetattr)cls
style_dict_style_key_mapstyle_kwargskv
_conv_to_xnew_vs           r7   _convert_to_style_kwargsz'OpenpyxlWriter._convert_to_style_kwargss   ss    0 $X.02$$& 	(DAq""1a(A QC&8.IJqME"'Q	( r:   c                N    ddl m} t        |t              r ||      S  |di |S )a  
        Convert ``color_spec`` to an openpyxl v2 Color object.

        Parameters
        ----------
        color_spec : str, dict
            A 32-bit ARGB hex string, or a dict with zero or more of the
            following keys.
                'rgb'
                'indexed'
                'auto'
                'theme'
                'tint'
                'index'
                'type'

        Returns
        -------
        color : openpyxl.styles.Color
        r   )Colorr"   )openpyxl.stylesr[   rB   str)rQ   
color_specr[   s      r7   _convert_to_colorz OpenpyxlWriter._convert_to_color   s*    , 	*j#&$$&:&&r:   c                    ddl m} ddddddd	}i }|j                         D ]2  \  }}|j                  ||      }|d
k(  r| j	                  |      }|||<   4  |di |S )a  
        Convert ``font_dict`` to an openpyxl v2 Font object.

        Parameters
        ----------
        font_dict : dict
            A dict with zero or more of the following keys (or their synonyms).
                'name'
                'size' ('sz')
                'bold' ('b')
                'italic' ('i')
                'underline' ('u')
                'strikethrough' ('strike')
                'color'
                'vertAlign' ('vertalign')
                'charset'
                'scheme'
                'family'
                'outline'
                'shadow'
                'condense'

        Returns
        -------
        font : openpyxl.styles.Font
        r   )Fontsizebolditalic	underlinestrikethrough	vertAlign)szbiustrike	vertaligncolorr"   )r\   ra   rN   rO   r_   )rQ   	font_dictra   _font_key_mapfont_kwargsrU   rV   s          r7   _convert_to_fontzOpenpyxlWriter._convert_to_font   s    8 	) %$
 OO% 	DAq!!!Q'AG|))!,KN		 "k""r:   c                .    t        | j                  |      S )a  
        Convert ``stop_seq`` to a list of openpyxl v2 Color objects,
        suitable for initializing the ``GradientFill`` ``stop`` parameter.

        Parameters
        ----------
        stop_seq : iterable
            An iterable that yields objects suitable for consumption by
            ``_convert_to_color``.

        Returns
        -------
        stop : list of openpyxl.styles.Color
        )mapr_   )rQ   stop_seqs     r7   _convert_to_stopzOpenpyxlWriter._convert_to_stop   s      3(((33r:   c                r   ddl m}m} ddddddd}ddi}i }i }|j                         D ]l  \  }}	|j	                  |      }
|j	                  |      }|
dv r| j                  |	      }	|d	k(  r| j                  |	      }	|
r|	||
<   [|r|	||<   c|	||<   |	||<   n 	  |d
i |S # t        $ r  |d
i |cY S w xY w)a  
        Convert ``fill_dict`` to an openpyxl v2 Fill object.

        Parameters
        ----------
        fill_dict : dict
            A dict with one or more of the following keys (or their synonyms),
                'fill_type' ('patternType', 'patterntype')
                'start_color' ('fgColor', 'fgcolor')
                'end_color' ('bgColor', 'bgcolor')
            or one or more of the following keys (or their synonyms).
                'type' ('fill_type')
                'degree'
                'left'
                'right'
                'top'
                'bottom'
                'stop'

        Returns
        -------
        fill : openpyxl.styles.Fill
        r   )GradientFillPatternFill	fill_typestart_color	end_color)patternTypepatterntypefgColorfgcolorbgColorbgcolortype)r{   r|   stopr"   )r\   rx   ry   rN   rO   r_   rv   r*   )rQ   	fill_dictrx   ry   _pattern_fill_key_map_gradient_fill_key_mappfill_kwargsgfill_kwargsrU   rV   pkgks               r7   _convert_to_fillzOpenpyxlWriter._convert_to_fill   s    2	
 '&$$""!
 #.v!6OO% 	$DAq&**1-B'++A.B11))!,V|((+#$R #$R "#Q"#Q	$	0... 	0/,//	0s   B" "B65B6c                    ddl m} ddi}t        |t              r	 ||      S i }|j	                         D ]2  \  }}|j                  ||      }|dk(  r| j                  |      }|||<   4  |di |S )a  
        Convert ``side_spec`` to an openpyxl v2 Side object.

        Parameters
        ----------
        side_spec : str, dict
            A string specifying the border style, or a dict with zero or more
            of the following keys (or their synonyms).
                'style' ('border_style')
                'color'

        Returns
        -------
        side : openpyxl.styles.Side
        r   )Sideborder_stylestyle)r   rn   r"   )r\   r   rB   r]   rN   rO   r_   )rQ   	side_specr   _side_key_mapside_kwargsrU   rV   s          r7   _convert_to_sidezOpenpyxlWriter._convert_to_side5  s    " 	)'1i%i((OO% 	DAq!!!Q'AG|))!,KN		 "k""r:   c                    ddl m} ddd}i }|j                         D ]G  \  }}|j                  ||      }|dk(  r| j	                  |      }|dv r| j                  |      }|||<   I  |di |S )	aq  
        Convert ``border_dict`` to an openpyxl v2 Border object.

        Parameters
        ----------
        border_dict : dict
            A dict with zero or more of the following keys (or their synonyms).
                'left'
                'right'
                'top'
                'bottom'
                'diagonal'
                'diagonal_direction'
                'vertical'
                'horizontal'
                'diagonalUp' ('diagonalup')
                'diagonalDown' ('diagonaldown')
                'outline'

        Returns
        -------
        border : openpyxl.styles.Border
        r   )Border
diagonalUpdiagonalDown)
diagonalupdiagonaldownrn   )leftrighttopbottomdiagonalr"   )r\   r   rN   rO   r_   r   )rQ   border_dictr   _border_key_mapborder_kwargsrU   rV   s          r7   _convert_to_borderz!OpenpyxlWriter._convert_to_borderV  s    2 	+)5~V%%' 	!DAq##Aq)AG|))!,BB((+ M!	! &&&r:   c                    ddl m}  |di |S )a  
        Convert ``alignment_dict`` to an openpyxl v2 Alignment object.

        Parameters
        ----------
        alignment_dict : dict
            A dict with zero or more of the following keys (or their synonyms).
                'horizontal'
                'vertical'
                'text_rotation'
                'wrap_text'
                'shrink_to_fit'
                'indent'
        Returns
        -------
        alignment : openpyxl.styles.Alignment
        r   )	Alignmentr"   )r\   r   )rQ   alignment_dictr   s      r7   _convert_to_alignmentz$OpenpyxlWriter._convert_to_alignment~  s    & 	.*>**r:   c                    |d   S )aO  
        Convert ``number_format_dict`` to an openpyxl v2.1.0 number format
        initializer.

        Parameters
        ----------
        number_format_dict : dict
            A dict with zero or more of the following keys.
                'format_code' : str

        Returns
        -------
        number_format : str
        format_coder"   )rQ   number_format_dicts     r7   _convert_to_number_formatz(OpenpyxlWriter._convert_to_number_format  s      "-00r:   c                    ddl m}  |di |S )a'  
        Convert ``protection_dict`` to an openpyxl v2 Protection object.

        Parameters
        ----------
        protection_dict : dict
            A dict with zero or more of the following keys.
                'locked'
                'hidden'

        Returns
        -------
        r   )
Protectionr"   )r\   r   )rQ   protection_dictr   s      r7   _convert_to_protectionz%OpenpyxlWriter._convert_to_protection  s     	/,O,,r:   c           	        | j                  |      }i }|| j                  v r| j                  dk7  rd| j                  v r| j                  dk(  rT| j                  |   }| j                  j                  |      }| j                  |= | j                  j                  ||      }	n| j                  dk(  rt        d| d      | j                  dk(  r| j                  |   }	nJt        d| j                   d	      | j                  |   }	n!| j                  j                         }	||	_        t        |      rAt        t        t        t        f   |      }|	j                  |d
   dz   |d   dz         |	_        |D ]  }
|	j                  ||
j                  z   dz   ||
j                   z   dz         }| j#                  |
j$                        \  |_        }|r||_        i }|
j*                  rHt-        |
j*                        }|j/                  |      }| | j1                  |
j*                        }|||<   |r%|j3                         D ]  \  }}t5        |||        |
j6                  |
j8                  |	j;                  ||
j                  z   dz   ||
j                   z   dz   ||
j8                  z   dz   ||
j6                  z   dz          |sS||
j                  z   dz   }||
j6                  z   dz   }||
j                   z   dz   }||
j8                  z   dz   }t=        ||dz         D ]Y  }t=        ||dz         D ]E  }||k(  r||k(  r|	j                  ||      }|j3                         D ]  \  }}t5        |||        G [  y )Nnewr   replaceerrorzSheet 'z7' already exists and if_sheet_exists is set to 'error'.overlay'z^' is not valid for if_sheet_exists. Valid options are 'error', 'new', 'replace' and 'overlay'.r      )rowcolumn)	start_rowstart_column
end_columnend_row)r   r   )_get_sheet_namer?   _if_sheet_existsr&   r-   indexcreate_sheet
ValueErrortitler   r   tupleintcellfreeze_panesr   col_value_with_fmtvalvaluenumber_formatr   r]   rO   rY   rN   setattr
mergestartmergeendmerge_cellsrange)r0   cells
sheet_namestartrowstartcolr   _style_cacheold_wkstarget_indexwksr   xcellfmtrT   keyrU   rV   	first_rowlast_row	first_collast_colr   r   s                          r7   _write_cellszOpenpyxlWriter._write_cells  s    ))*5
;=$)>)>%)Gtzz!((I5"kk*5G#'99??7#;L		*-))00\JC**g5$!* .= >  **i7++j1C$D112 3U U 
 kk*-))((*C"CI .c3h>L"xx Oa'Q!0C  (  C  ,	5DHHtxx'!+Htxx4G!4K  E  $33DHH=EK&)#;=Lzz$**o+//4'#'#@#@#LL(4L%(..0 )DAqE1a() *t}}/H&1A5!)DHH!4q!8'$--7!;$t6:	      (488 3a 7I'$//9A=H (488 3a 7I'$--7!;H$Y1= 5#(HqL#A 5C"i/C94D ($'HHCSH$AE(4(:(:(< 51 'q! 4555K,	5r:   )NNNwNNN)r1   z)FilePath | WriteExcelBuffer | ExcelWriterr2   
str | Noner3   r   r4   r   r   r]   r   StorageOptions | Noner   zExcelWriterIfSheetExists | Noner   zdict[str, Any] | NonereturnNone)r   r   )r   dict[str, Any])r   r   )rR   dictr   zdict[str, Serialisable])r   r   )Nr   r   N)
r   r   r   r   r   r   r   ztuple[int, int] | Noner   r   )__name__
__module____qualname___engine_supported_extensionsr%   propertyr-   r?   rE   classmethodrY   r_   rr   rv   r   r   r   r   r   r   r   __classcell__r6   s   @r7   r   r   (   s   G.
 ""&&*15;?/3-:7-: -:  	-:
 $-: -: /-: 9-: --: 
-:^    
, ! !F ' '8 -# -#^ 4 4" <0 <0| # #@ %' %'N + +, 1 1" - -* "&/3X5 X5 	X5
 X5 -X5 
X5r:   r   c                       e Zd Z eed         	 	 d	 	 	 	 	 	 	 d fd       Zedd       Z	 	 	 	 ddZedd       Z	ddZ
ddZdd	Z	 d	 	 	 dd
Z xZS )OpenpyxlReaderr   )r   c                @    t        d       t        | 	  |||       y)a2  
        Reader using openpyxl engine.

        Parameters
        ----------
        filepath_or_buffer : str, path object or Workbook
            Object to be parsed.
        {storage_options}
        engine_kwargs : dict, optional
            Arbitrary keyword arguments passed to excel engine.
        r   )r   r   N)r   r$   r%   )r0   filepath_or_bufferr   r   r6   s       r7   r%   zOpenpyxlReader.__init__  s(    $ 	#:.+' 	 	
r:   c                    ddl m} |S )Nr   r   )r   r   )r0   r   s     r7   _workbook_classzOpenpyxlReader._workbook_class/  s
    %r:   c                2    ddl m} dddd} ||fi ||z  S )Nr   r    TF)	read_only	data_only
keep_links)r   r!   )r0   r   r   r!   default_kwargss        r7   r!   zOpenpyxlReader.load_workbook5  s0     	+'+$eT
-
 	
r:   c                h    | j                   j                  D cg c]  }|j                   c}S c c}w rK   )r-   r.   r   )r0   sheets     r7   sheet_nameszOpenpyxlReader.sheet_namesA  s$    )-)=)=>>>>s   /c                B    | j                  |       | j                  |   S rK   )raise_if_bad_sheet_by_namer-   )r0   r=   s     r7   get_sheet_by_namez OpenpyxlReader.get_sheet_by_nameE  s    ''-yyr:   c                V    | j                  |       | j                  j                  |   S rK   )raise_if_bad_sheet_by_indexr-   r.   )r0   r   s     r7   get_sheet_by_indexz!OpenpyxlReader.get_sheet_by_indexI  s%    ((/yy##E**r:   c                   ddl m}m} |j                  y|j                  |k(  rt
        j                  S |j                  |k(  r;t        |j                        }||j                  k(  r|S t        |j                        S |j                  S )Nr   )
TYPE_ERRORTYPE_NUMERIC )	openpyxl.cell.cellr  r  r   	data_typenpnanr   float)r0   r   r  r  r   s        r7   _convert_cellzOpenpyxlReader._convert_cellM  sn    	

 ::^^z)66M^^|+djj/Cdjj 
$$zzr:   c                @   | j                   j                  r|j                          g }d}t        |j                        D ]p  \  }}|D cg c]  }| j                  |       }}|r#|d   dk(  r|j                          |r	|d   dk(  r|r|}|j                  |       |at        |      |k\  sp n |d |dz    }t        |      dkD  rKt        d |D              }	t        d |D              |	k  r$dg}
|D cg c]  }||	t        |      z
  |
z  z    }}|S c c}w c c}w )Nr  r   r   c              3  2   K   | ]  }t        |        y wrK   len.0data_rows     r7   	<genexpr>z0OpenpyxlReader.get_sheet_data.<locals>.<genexpr>w  s     ?hCM?   c              3  2   K   | ]  }t        |        y wrK   r  r  s     r7   r  z0OpenpyxlReader.get_sheet_data.<locals>.<genexpr>x  s     6X3x=6r  )r-   r   reset_dimensions	enumeraterowsr  popappendr  maxmin)r0   r   file_rows_neededdatalast_row_with_data
row_numberr   r   converted_row	max_width
empty_cellr  s               r7   get_sheet_datazOpenpyxlReader.get_sheet_data_  sH    99""$#%(4 		OJBEF$T//5FMFM"$5$;!!#  M"$5$; %/"KK&+D	=M0M		 ,(1,-t9q=?$??I666B,.4
 %)  	CM 9ZGG 
 / G$s   
D7D)NN)r   FilePath | ReadBuffer[bytes]r   r   r   zdict | Noner   r   )r   ztype[Workbook])r   r)  r   r   )r   z	list[str])r=   r]   )r   r   )r   r   rK   )r!  z
int | Noner   zlist[list[Scalar]])r   r   r   r   r	   r%   r   r   r!   r   r  r  r  r(  r   r   s   @r7   r   r     s    &789 26%)	
8
 /
 #	

 

 :
0  


">

	

 ? ?+& 59 '1 	 r:   r   r   )"
__future__r   rC   typingr   r   r   numpyr  pandas.compat._optionalr   pandas.util._decoratorsr   pandas.core.shared_docsr	   pandas.io.excel._baser
   r   pandas.io.excel._utilr   r   r   r   !openpyxl.descriptors.serialisabler   pandas._typingr   r   r   r   r   r   r   r   r"   r:   r7   <module>r4     sd    "    > ' 0
 !> j5[ j5Zj_Z0 jr:   