o թZh@sxdZddlmZddlmZmZmZmZddlZddl Z ddl Z ddl m Z m Z mZmZddlZddlZddlmZddlmZddlmZdd lmZdd lmZmZdd lmZm Z m!Z!m"Z"ddl#m$m%Z&dd l'm(Z(dd l)m*Z*ddl+m,Z,m-Z-ddl.m/Z/ddl0m1Z1e rddl2m3Z3m4Z4m5Z5m6Z6ddlm7Z7GdddZ8Gddde8Z9GdddZ:GdddZ;dS)zC Utilities for conversion to writer-agnostic Excel representation. ) annotations)HashableIterableMappingSequenceN) TYPE_CHECKINGAnyCallablecast) is_list_like)doc)find_stack_level)missing)is_float is_scalar) DataFrameIndex MultiIndex PeriodIndex) _shared_docs) CSS4_COLORS) CSSResolver CSSWarning)get_level_lengths) pprint_thing)FilePath IndexLabelStorageOptionsWriteExcelBuffer ExcelWriterc@s&eZdZdZeZ   d dd d ZdS) ExcelCellrowcolvalstyle mergestartmergeendNr#intr$r' int | Noner(returnNonecCs(||_||_||_||_||_||_dSNr")selfr#r$r%r&r'r(r/N/var/www/html/lang_env/lib/python3.10/site-packages/pandas/io/formats/excel.py__init__Cs   zExcelCell.__init__)NNN) r#r)r$r)r'r*r(r*r+r,)__name__ __module__ __qualname__Z __fields__ __slots__r1r/r/r/r0r!?sr!cseZdZdfdd ZZS) CssExcelCellr#r)r$r& dict | None css_styles3dict[tuple[int, int], list[tuple[str, Any]]] | Nonecss_rowcss_col css_converterCallable | Noner+r,c  sR|r|rdd|||fD} t| } || }tjd||||d| dS)NcSsi|] \}}||qSr/lower).0propr%r/r/r0 csz)CssExcelCell.__init__..r#r$r%r&r/) frozensetitemssuperr1) r.r#r$r%r&r8r:r;r<kwargsZdeclaration_dictZunique_declarations __class__r/r0r1Us    zCssExcelCell.__init__)r#r)r$r)r&r7r8r9r:r)r;r)r<r=r+r,)r2r3r4r1 __classcell__r/r/rHr0r6Tsr6c@seZdZUdZeZdddddddZdddddddddddddd Zdddd Zd d d d dZ dddDZ de d<dgdhddZ e ZdiddZdidd Zdjd#d$Zdkd&d'Zdld(d)Zdmd+d,Zdnd.d/Zdod3d4Zdpd6d7Zdqd9d:Zdrd=d>Zdsd?d@ZdtdBdCZdudEdFZdmdGdHZdmdIdJZdvdLdMZdwdOdPZ dmdQdRZ!dvdSdTZ"dxdVdWZ#dydZd[Z$dzd]d^Z%d{dadbZ&d|dcddZ'd{dedfZ(dS)}CSSToExcelConverteraK A callable for converting CSS declarations to ExcelWriter styles Supports parts of CSS 2.2, with minimal CSS 3.0 support (e.g. text-shadow), focusing on font styling, backgrounds, borders and alignment. Operates by first computing CSS styles in a fairly generic way (see :meth:`compute_css`) then determining Excel style properties from CSS properties (see :meth:`build_xlstyle`). Parameters ---------- inherited : str, optional CSS declarations understood to be the containing scope for the CSS processed by :meth:`__call__`. topcenterbottom)rLztext-topmiddleZbaselinerNz text-bottomTF) boldZbolderZ600Z700Z800Z900normalZlighter100Z200Z300Z400500)rQitalicZoblique)Zserifz sans-serifZcursiveZfantasycCsi|]}||qSr/r>)r@r&r/r/r0rBszCSSToExcelConverter.) dashedZ mediumDashDotZ dashDotDothairdottedmediumDashDotDotdoubleZdashDotZ slantDashDot mediumDashedzdict[str, str] | None inheritedN str | Noner+r,cCs.|dur |||_nd|_t|j|_dSr-) compute_cssr_ functoolscache_call_uncached _call_cached)r.r_r/r/r0r1szCSSToExcelConverter.__init__ declarations str | frozenset[tuple[str, str]]dict[str, dict[str, str]]cCs ||S)a Convert CSS declarations to ExcelWriter style. Parameters ---------- declarations : str | frozenset[tuple[str, str]] CSS string or set of CSS declaration tuples. e.g. "font-weight: bold; background: blue" or {("font-weight", "bold"), ("background", "blue")} Returns ------- xlstyle : dict A style as interpreted by ExcelWriter when found in ExcelCell.style. )re)r.rfr/r/r0__call__s zCSSToExcelConverter.__call__cCs|||j}||Sr-)rar_ build_xlstyle)r.rf propertiesr/r/r0rds z"CSSToExcelConverter._call_uncachedpropsMapping[str, str]csH||||||||||d}dfdd ||S) N) alignmentborderfillfontZ number_formatddict[str, str | None]r+r,csFt|D]\}}|dur||=qt|tr ||s ||=qdS)z4Remove key where value is None, through nested dictsN)listrE isinstancedict)rrkv remove_noner/r0rzs z6CSSToExcelConverter.build_xlstyle..remove_none)rrrsr+r,)build_alignment build_border build_fill build_fontbuild_number_format)r.rloutr/ryr0rjs  z!CSSToExcelConverter.build_xlstyledict[str, bool | str | None]cCs|d||||dS)Nz text-align) horizontalverticalZ wrap_text)get_get_vertical_alignment_get_is_wrap_textr.rlr/r/r0r{sz#CSSToExcelConverter.build_alignmentcC|d}|r |j|SdS)Nzvertical-align)r VERTICAL_MAP)r.rlZvertical_alignr/r/r0r  z+CSSToExcelConverter._get_vertical_alignment bool | NonecCs"|ddur dSt|ddvS)Nz white-space)Znowrapprezpre-line)rboolrr/r/r0r sz%CSSToExcelConverter._get_is_wrap_text dict[str, dict[str, str | None]]csfdddDS)Ncsfi|]/}|d|dd|dd|dd|ddqS)zborder-z-stylez-widthz-color)r&color) _border_stylercolor_to_excel)r@Zsiderlr.r/r0rBs z4CSSToExcelConverter.build_border..rLrightrNleftr/rr/rr0r|s  z CSSToExcelConverter.build_borderr&widthrcCs|dur|dur|durdS|dur|durdS|dvrdS||}|dur)dS|dvr/|S|dkr5dS|dkrA|dvr?dSdS|dkrM|dvrKdSd S||jvrW|j|Stjd t|ttd dS) Nnone)rZhidden)NZgrooveZridgeZinsetZoutsetsolidr]r[)rZthinr\rYr^zUnhandled border style format:  stacklevel)_get_width_nameBORDER_STYLE_MAPwarningswarnreprrr )r.r&rrZ width_namer/r/r0rs:    z!CSSToExcelConverter._border_style width_inputcCs2||}|dkr dS|dkrdS|dkrdSdS)Ngh㈵>g?rgffffff@ZmediumZthick)_width_to_float)r.rrr/r/r0rVs z#CSSToExcelConverter._get_width_namefloatcCs|durd}||S)NZ2pt) _pt_to_float)r.rr/r/r0r`s z#CSSToExcelConverter._width_to_float pt_stringstrcCs|dsJt|dS)Npt)endswithrrstrip)r.rr/r/r0resz CSSToExcelConverter._pt_to_floatcCs&|d}|dvr||ddSdS)Nzbackground-color)N transparentrr)ZfgColorZ patternType)rr)r.rlZ fill_colorr/r/r0r}is zCSSToExcelConverter.build_fillrscCs,|d}t|tr|ddn|}d|iS)Nz number-format§;Z format_code)rrurreplace)r.rlZfcr/r/r0rps z'CSSToExcelConverter.build_number_format$dict[str, bool | float | str | None]c Csv||}||}|r|dnd||||||||d|vr'dndd|vp-d||d||d S)Nr underlinesinglez line-throughr) namefamilysizerPrTrstrikerZshadow) _get_font_names_get_decoration_select_font_family_get_font_size _get_is_bold_get_is_italicrr _get_shadow)r.rl font_names decorationr/r/r0r~us   zCSSToExcelConverter.build_fontcCr)Nz font-weight)rBOLD_MAP)r.rlweightr/r/r0rrz CSSToExcelConverter._get_is_boldcCr)Nz font-style)r ITALIC_MAP)r.rlZ font_styler/r/r0rrz"CSSToExcelConverter._get_is_italic Sequence[str]cCs|d}|dur |SdS)Nztext-decorationr/)rsplit)r.rlrr/r/r0rs z#CSSToExcelConverter._get_decorationrcCsd|vrdSdS)Nrrr/)r.rr/r/r0_get_underlinesz"CSSToExcelConverter._get_underlinecCs d|vrttd|dSdS)Nz text-shadowz ^[^#(]*[1-9])rresearchrr/r/r0rszCSSToExcelConverter._get_shadowcCstd|dd}g}|D]3}|dddkr#|dddd}n|dddkr6|ddd d}n|}|rA||q|S) Nz(?x) ( "(?:[^"]|\\")+" | '(?:[^']|\\')+' | [^'",]+ )(?=,|\s*$) z font-familyrU"z\"'z\')rfindallrrstripappend)r.rlZfont_names_tmprrr/r/r0rs   z#CSSToExcelConverter._get_font_names float | NonecCs |d}|dur |S||S)Nz font-size)rr)r.rlrr/r/r0rs  z"CSSToExcelConverter._get_font_sizerr*cCs(d}|D] }|j|}|r|Sq|Sr-) FAMILY_MAPr)r.rrrr/r/r0rs z'CSSToExcelConverter._select_font_familyr%cCs^|durdS||r||Sz|j|WSty.tjdt|ttdYdSw)NzUnhandled color format: r) _is_hex_color_convert_hex_to_excel NAMED_COLORSKeyErrorrrrrr r.r%r/r/r0rs     z"CSSToExcelConverter.color_to_excel color_stringrcCst|dS)N#)r startswith)r.rr/r/r0rsz!CSSToExcelConverter._is_hex_colorcCsD|d}||r|dd|dd|ddS|S)NrrrVrU)lstrip_is_shorthand_colorupperr.rcoder/r/r0rs  (z)CSSToExcelConverter._convert_hex_to_excelcCs8|d}t|dkr dSt|dkrdStd|)zcCheck if color code is shorthand. #FFF is a shorthand as opposed to full #FFFFFF. rTFzUnexpected color )rlen ValueErrorrr/r/r0rs   z'CSSToExcelConverter._is_shorthand_colorr-)r_r`r+r,)rfrgr+rh)rlrmr+rh)rlrmr+r)rlrmr+r`)rlrmr+r)rlrmr+r)r&r`rr`rr`)rr`r+r`)rr`r+r)rrr+r)rlrm)rlrmr+rs)rlrmr+r)rlrmr+r)rrr+r`)rlrmr+r)rrr+r*)r%r`r+r`)rrr+r)rrr+r))r2r3r4__doc__rrrrrrr__annotations__r1rrarirdrjr{rrr|rrrrr}rr~rrrrrrrrrrrrr/r/r/r0rKms            7              rKc@seZdZdZdZdZ         dHdIddZedJddZd d!Z dKd#d$Z dKd%d&Z dKd'd(Z dKd)d*Z dKd+d,ZdKd-d.ZedLd/d0ZdMd3d4ZdKd5d6Zeed7d8 9 : :    dNdOdFdGZdS)PExcelFormattera9 Class for formatting a DataFrame to a list of ExcelCells, Parameters ---------- df : DataFrame or Styler na_rep: na representation float_format : str, default None Format string for floating point numbers cols : sequence, optional Columns to write header : bool or sequence of str, default True Write out column names. If a list of string is given it is assumed to be aliases for the column names index : bool, default True output row names (index) index_label : str or sequence, default None Column label for index column(s) if desired. If None is given, and `header` and `index` are True, then the index names are used. A sequence should be given if the DataFrame uses MultiIndex. merge_cells : bool, default False Format MultiIndex and Hierarchical Rows as merged cells. inf_rep : str, default `'inf'` representation for np.inf values (which aren't representable in Excel) A `'-'` sign will be added in front of -inf. style_converter : callable, optional This translates Styler styles (CSS) into ExcelWriter styles. Defaults to ``CSSToExcelConverter()``. It should have signature css_declarations string -> excel style. This is only called for body cells. ii@rNTFinfna_repr float_formatr`colsSequence[Hashable] | NoneheaderSequence[Hashable] | boolindexr index_labelIndexLabel | None merge_cellsinf_repstyle_converterr=r+r,c Csd|_||_t|ts!||_|j|j}| durt} | |_nd|_d|_||_ |durWt t | |j st d|S)N-tzinfoz{Excel does not support datetimes with timezones. Please ensure that datetimes are timezone unaware before writing to Excel.) rrZisnarrZisposinf_scalarrZisneginf_scalarrrgetattrrrr/r/r0 _format_valueTs   zExcelFormatter._format_valueIterable[ExcelCell]ccs|jjdkr|jstd|js|jsdS|j}|j|jdd}t|}d}d}|jr=t |j jt r=t |j jdd}|jrt |jD]\}}t||||jdVqEt t||j|jD]B\}\}}} || } |D]1\} } d\} }| dkr||| | } }t||| d| | |jt|jdd|| |j| |d Vqnq^n+t t|D]$\} } d tt| }t||| d||jt|jdd|| |jd Vq||_dS) NrUz]Writing to Excel with MultiIndex columns and no index ('index'=False) is not yet implemented.FZsparsifyZ include_namesrrCNN ctx_columns r#r$r%r&r8r:r;r<r'r(.r#r$r%r&r8r:r;r<)rZnlevelsrNotImplementedError _has_aliasesr _format_multirrrurrr enumeratenamesr!rziplevelscodestakerEr6rrrjoinmaprr)r.r level_strs level_lengths coloffsetlnumrspansr  level_codesvaluesispan_valr'r(rxr/r/r0_format_header_mifsx          z ExcelFormatter._format_header_mic cs|js|jrgd}|jrd}t|jjtrt|jjj}|j}|jrGt t |j|_t|jt|jkrDt dt|jdt|jd|j}t |D]\}}t |j||||jt|jddd||jdVqKdSdS)NrrUzWriting z cols but got z aliasesrr)rrrrurrrr rr rrrr6rrrrr)r.rZcolnamescolindexZcolnamer/r/r0_format_header_regulars<   z%ExcelFormatter._format_header_regularcstjtr }n}d}jjjrGddjjjDdgtj}t dddd|DrGfd dt |D}j d 7_ t ||S) Nr/cSsg|] }|dur |ndqS)Nrr/r@xr/r/r0 sz1ExcelFormatter._format_header..rcSs|o|Sr-r/)ryr/r/r0sz/ExcelFormatter._format_header..css|]}|dkVqdS)rNr/rr/r/r0 sz0ExcelFormatter._format_header..c3s&|]\}}tj||jVqdSr-)r!rr)r@rr%rr/r0r!s  rU)rurrrrrrr rrbreducerr itertoolschain)r.genZgen2r#r/rr0_format_headers     zExcelFormatter._format_headercCst|jjtr |S|Sr-)rurrr_format_hierarchical_rows_format_regular_rowsrr/r/r0 _format_bodyszExcelFormatter._format_bodyc cs2|js|jr|jd7_|jr|jr%t|jtttj t fr%|jd}n|jr2t|jt r2|j}n|j jj d}t|jtrF|jd7_|rY|jdurYt|jdd||jV|j j}t|j jtrj|j j}t|D]\}}t|j|d||jt|jdd|d|jdVqnd}nd}||EdHdS)NrUrF ctx_indexr)rrrrrrurttuplenpndarrayrrrr rrr!rrZ to_timestamprr6rrr_generate_body)r.rZ index_valuesidxZidxvalrr/r/r0r(s@       z#ExcelFormatter._format_regular_rowsccs|js|jr|jd7_d}|jr|jjj}|jr)t|jtt t j t fr)|j}t|j tr9|jr9|jd7_tj|rX|jdurXt|D]\}}t|jd|||jVqG|jr|jjjddd}t|}t||jjj|jjjD]J\}}} |j| |j|jd} |D]2\} } d\} }| dkr|j| | d} |}t|j| || | |jt|j dd| ||j!| |d Vq|d7}qtn-t|jjD]&}t|D]\}}t|j||||jt|j dd|||j!d Vq|d7}q|"|EdHdS) NrUrFTr)Z allow_fillZ fill_valuerr*rr)#rrrrrr rrurtr+r,r-rrrrcomZ any_not_nonerr!rrrr r r r Z _can_hold_naZ _na_valuerEr6rrrr.)r.ZgcolidxZ index_labelsZcidxrrrrr rrrrr'r(Z indexcolvalsr/Z indexcolvalr/r/r0r'sz      z(ExcelFormatter._format_hierarchical_rowscCs t|jS)z1Whether the aliases for column names are present.)r rrr/r/r0rcs zExcelFormatter._has_aliasesrr)c csptt|jD]-}|jjdd|f}t|D]\}}t|j||||dt|j dd|||j dVqqdS)Nctxr) rangerrrZilocrr6rrrr)r.rZcolidxZseriesrr%r/r/r0r.hs   zExcelFormatter._generate_bodyccs4t||D] }||j|_|Vq dSr-)r#r$r&r)rr%)r.cellr/r/r0get_formatted_cellsxs z"ExcelFormatter.get_formatted_cellsstorage_options)r5Sheet1rwriter)FilePath | WriteExcelBuffer | ExcelWriter sheet_namestartrowstartcol freeze_panestuple[int, int] | NoneengineStorageOptions | None engine_kwargsr7c  Csddlm} |jj\} } | |jks| |jkr(td| d| d|jd|j|dur.i}|} t|| r:d} n | ||||d}d } z|j | ||||d W| rX| dSdS| ra| ww) ab writer : path-like, file-like, or ExcelWriter object File path or existing ExcelWriter sheet_name : str, default 'Sheet1' Name of sheet which will contain DataFrame startrow : upper left cell row to dump data frame startcol : upper left cell column to dump data frame freeze_panes : tuple of integer (length 2), default None Specifies the one-based bottommost row and rightmost column that is to be frozen engine : string, default None write engine to use if writer is a path - you can also set this via the options ``io.excel.xlsx.writer``, or ``io.excel.xlsm.writer``. {storage_options} engine_kwargs: dict, optional Arbitrary keyword arguments passed to excel engine. rrz-This sheet is too large! Your sheet size is: z, z Max sheet size is: NF)r>r5r@T)r:r;r<) Zpandas.io.excelr rshapemax_rowsmax_colsrr4ruZ _write_cellsclose)r.r7r9r:r;r<r>r5r@r Znum_rowsZnum_colsZformatted_cellsZ need_saver/r/r0write}sF "    zExcelFormatter.write) rNNTTNFrN)rrrr`rrrrrrrrrrrrrr=r+r,)r+r)r+r)r+r)rr)r+r)r6rrNNNN)r7r8r9rr:r)r;r)r<r=r>r`r5r?r@r7r+r,)r2r3r4rrBrCr1propertyrrrrr&r)r(r'rr.r4r rrEr/r/r/r0rsH  -   E    .O    r)s>