a OfM@sddlmZddlZddlmZmZmZddlZddl m Z ddl m Z ddl mZddlmZmZddlmZmZerdd lmZdd lmZdd lmZmZmZmZmZmZGd d d eZ GdddedZ!dS)) annotationsN) TYPE_CHECKINGAnycast)import_optional_dependency)doc) _shared_docs)BaseExcelReader ExcelWriter)combine_kwargsvalidate_freeze_panesWorkbook) Serialisable)ExcelWriterIfSheetExistsFilePath ReadBufferScalarStorageOptionsWriteExcelBufferc s eZdZdZdZd6ddddddd d d d fd d ZeddddZeddddZd dddZ e dddddZ e ddZ e dd Z e d!d"Ze dd#d$d%Ze d&d'Ze d(d)Ze d*d+Ze d,d-Ze d.d/Zd7dd1d1d2d d3d4d5ZZS)8OpenpyxlWriteropenpyxl)z.xlsxz.xlsmNwz)FilePath | WriteExcelBuffer | ExcelWriterz str | NonestrStorageOptions | NonezExcelWriterIfSheetExists | Nonezdict[str, Any] | NoneNone) pathengine date_formatdatetime_formatmodestorage_optionsif_sheet_exists engine_kwargsreturnc  sddlm} t|| }tj|||||dd|jvrddlm} z| |jj fi||_ Wn t y||jj Yn0|jj dnRz| fi||_ Wn t y|jj Yn0|jjr|j|jjddS)Nrr )r r!r"r#r+ load_workbook)Zopenpyxl.workbookrr super__init___moderr'_handleshandle_book TypeErrorcloseseekbook worksheetsremove) selfrrrrr r!r"r#kwargsrr' __class__R/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/io/excel/_openpyxl.pyr),s0        zOpenpyxlWriter.__init__rr$cCs|jS)z Book instance of class openpyxl.workbook.Workbook. This attribute can be used to access engine-specific features. )r-r4r8r8r9r1[szOpenpyxlWriter.bookzdict[str, Any]csfddjjD}|S)z(Mapping of sheet names to sheet objects.csi|]}|j|qSr8)r1).0namer;r8r9 gz)OpenpyxlWriter.sheets..)r1Z sheetnames)r4resultr8r;r9sheetsdszOpenpyxlWriter.sheetscCs:|j|jjd|jvr6t|jjtjs6|jjdS)z( Save workbook to disk. r%N)r1saver+r,r* isinstancemmaptruncater;r8r8r9_savejszOpenpyxlWriter._savedictzdict[str, Serialisable]) style_dictr$cCsXddi}i}|D]>\}}|||}t|d|dd}||}|r|||<q|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. ZbordersborderZ _convert_to_cSsdSNr8)xr8r8r9r?z9OpenpyxlWriter._convert_to_style_kwargs..)itemsgetgetattr)clsrHZ_style_key_map style_kwargskvZ _conv_to_xnew_vr8r8r9_convert_to_style_kwargsss  z'OpenpyxlWriter._convert_to_style_kwargscCs0ddlm}t|tr||S|fi|SdS)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)ColorN)openpyxl.stylesrVrCr)rPZ color_specrVr8r8r9_convert_to_colors  z OpenpyxlWriter._convert_to_colorcCshddlm}ddddddd }i}|D].\}}|||}|d krP||}|||<q*|fi|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)Fontsizebolditalic underlineZ strikethroughZ vertAlign)szbiustrikeZ vertaligncolor)rWrYrMrNrX)rPZ font_dictrYZ _font_key_mapZ font_kwargsrRrSr8r8r9_convert_to_fonts     zOpenpyxlWriter._convert_to_fontcCs 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 )maprX)rPZstop_seqr8r8r9_convert_to_stopszOpenpyxlWriter._convert_to_stop) fill_dictc Csddlm}m}ddddddd}ddi}i}i}|D]l\}} ||} ||} | dvrh|| } | d krz|| } | r| || <q:| r| || <q:| ||<| ||<q:z|fi|WSty|fi|YS0d S) 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 PatternFillZ fill_type start_color end_color)Z patternTypeZ patterntypeZfgColorZfgcolorZbgColorbgcolortype)rjrkstopN)rWrhrirMrNrXrfr.) rPrgrhriZ_pattern_fill_key_mapZ_gradient_fill_key_mapZ pfill_kwargsZ gfill_kwargsrRrSpkZgkr8r8r9_convert_to_fills8         zOpenpyxlWriter._convert_to_fillcCsrddlm}ddi}t|tr(||dSi}|D].\}}|||}|dkrZ||}|||<q4|fi|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_stylestyle)rsrc)rWrqrCrrMrNrX)rPZ side_specrqZ _side_key_mapZ side_kwargsrRrSr8r8r9_convert_to_side5s      zOpenpyxlWriter._convert_to_sidecCsrddlm}ddd}i}|D]@\}}|||}|dkrH||}|dvrZ||}|||<q"|fi|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)BorderZ diagonalUpZ diagonalDown)Z diagonalupZ diagonaldownrc)leftrighttopbottomZdiagonal)rWrurMrNrXrt)rPZ border_dictruZ_border_key_mapZ border_kwargsrRrSr8r8r9_convert_to_borderVs      z!OpenpyxlWriter._convert_to_bordercCsddlm}|fi|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) Alignment)rWr{)rPZalignment_dictr{r8r8r9_convert_to_alignment~s z$OpenpyxlWriter._convert_to_alignmentcCs|dS)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 Z format_coder8)rPZnumber_format_dictr8r8r9_convert_to_number_formatsz(OpenpyxlWriter._convert_to_number_formatcCsddlm}|fi|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) Protection)rWr~)rPZprotection_dictr~r8r8r9_convert_to_protections z%OpenpyxlWriter._convert_to_protectionrintztuple[int, int] | None) sheet_namestartrowstartcol freeze_panesr$cCs||}i}||jvr|jdkrd|jvr|jdkrd|j|}|j|}|j|=|j||} q|jdkrtd|dq|jdkr|j|} qtd|jd q|j|} n|j} || _t |rt t t t f|}| j |d d |d d d | _|D]} | j || jd || jd d } || j\| _} | rL| | _i} | jrt| j}||} | dur|| j} | ||<| r| D]\}}t| ||q| jdur| jdur| j|| jd || jd || jd || jd d | r|| jd }|| jd }|| jd }|| jd }t||d D]b}t||d D]L}||kr|||kr|q`| j ||d} | D]\}}t| ||qq`qNqdS)Nnewr%replaceerrorzSheet '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)rowcolumn)Z start_rowZ start_columnZ end_columnZend_row)rr)Z_get_sheet_namerAZ_if_sheet_existsr*r1indexZ create_sheet ValueErrortitler rtuplercellrrcolZ_value_with_fmtvalvalueZ number_formatrsrrNrUrMsetattrZ mergestartZmergeendZ merge_cellsrange)r4cellsrrrr _style_cacheZold_wksZ target_indexZwksrZxcellfmtrQkeyrRrS first_rowlast_rowZ first_colZlast_colrrr8r8r9 _write_cellss|                       zOpenpyxlWriter._write_cells)NNNrNNN)NrrN)__name__ __module__ __qualname__Z_engineZ_supported_extensionsr)propertyr1rArF classmethodrUrXrdrfrprtrzr|r}rr __classcell__r8r8r6r9r(sP$/ #  / >  '   rcseZdZeeddd&dddddfd d Zed d d dZdddddZedd ddZ ddddZ ddddZ dd dd Z d'd!d"d#d$d%Z ZS)(OpenpyxlReaderr!)r!NzFilePath | ReadBuffer[bytes]rz dict | Noner)filepath_or_bufferr!r#r$cstdtj|||ddS)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)rr(r))r4rr!r#r6r8r9r)s zOpenpyxlReader.__init__ztype[Workbook]r:cCsddlm}|S)Nrr )rr)r4rr8r8r9_workbook_class/s zOpenpyxlReader._workbook_classr)rr$cCs,ddlm}dddd}||fi||BS)Nrr&TF) read_onlyZ data_onlyZ keep_links)rr')r4rr#r'Zdefault_kwargsr8r8r9r'5s  zOpenpyxlReader.load_workbookz list[str]cCsdd|jjDS)NcSsg|] }|jqSr8)r)r<sheetr8r8r9 Cr?z.OpenpyxlReader.sheet_names..)r1r2r;r8r8r9 sheet_namesAszOpenpyxlReader.sheet_namesr)r=cCs|||j|SrJ)Zraise_if_bad_sheet_by_namer1)r4r=r8r8r9get_sheet_by_nameEs z OpenpyxlReader.get_sheet_by_namer)rcCs|||jj|SrJ)Zraise_if_bad_sheet_by_indexr1r2)r4rr8r8r9get_sheet_by_indexIs z!OpenpyxlReader.get_sheet_by_indexrcCs`ddlm}m}|jdurdS|j|kr.tjS|j|krZt|j}||jkrP|St|jS|jS)Nr) TYPE_ERROR TYPE_NUMERIC) Zopenpyxl.cell.cellrrrZ data_typenpnanrfloat)r4rrrrr8r8r9 _convert_cellMs      zOpenpyxlReader._convert_cellz int | Nonezlist[list[Scalar]])file_rows_neededr$csjjr|g}d}t|jD]^\}}fdd|D}|rV|ddkrV|q<|r^|}|||dur"t||kr"qq"|d|d}t|dkrtdd|Dt d d|Dkrdgfd d|D}|S) Ncsg|]}|qSr8)r)r<rr;r8r9rhr?z1OpenpyxlReader.get_sheet_data..rrrcss|]}t|VqdSrJlenr<Zdata_rowr8r8r9 wr?z0OpenpyxlReader.get_sheet_data..css|]}t|VqdSrJrrr8r8r9rxr?cs g|]}|t|qSr8rr) empty_cell max_widthr8r9rzs) r1rZreset_dimensions enumeraterowspopappendrmaxmin)r4rrdataZlast_row_with_dataZ row_numberrZ converted_rowr8)rrr4r9get_sheet_data_s,    zOpenpyxlReader.get_sheet_data)NN)N)rrrrrr)rrr'rrrrrrr8r8r6r9rs  rr)" __future__rrDtypingrrrnumpyrZpandas.compat._optionalrZpandas.util._decoratorsrZpandas.core.shared_docsrZpandas.io.excel._baser r Zpandas.io.excel._utilr r rrZ!openpyxl.descriptors.serialisablerZpandas._typingrrrrrrrrr8r8r8r9s"       p