o d@s\dZddlmZddlmZmZddlZddlZddlm Z m Z m Z m Z m Z mZmZddlZddlZddlmZddlmZmZddlmZdd lmZdd lmZdd lmZm Z dd l!m"Z"m#Z#m$Z$m%Z%ddl&m'm(Z)dd l*m+Z+ddl,m-Z-ddl.m/Z/m0Z0ddl1m2Z2ddl3m4Z4GdddZ5Gddde5Z6GdddZ7GdddZ8dS)zC Utilities for conversion to writer-agnostic Excel representation. ) annotations) lru_cachereduceN)AnyCallableHashableIterableMappingSequencecast) is_list_like) IndexLabelStorageOptions)doc)find_stack_level)missing)is_float is_scalar) DataFrameIndex MultiIndex PeriodIndex) _shared_docs) CSS4_COLORS) CSSResolver CSSWarning)get_level_lengths) pprint_thingc@s&eZdZdZeZ   d dd d ZdS) ExcelCellrowcolvalstyle mergestartmergeendNr intr!r$ int | Noner%returnNonecCs(||_||_||_||_||_||_dSNr)selfr r!r"r#r$r%r,K/app/.heroku/python/lib/python3.10/site-packages/pandas/io/formats/excel.py__init__=s   zExcelCell.__init__)NNN) r r&r!r&r$r'r%r'r(r))__name__ __module__ __qualname__Z __fields__ __slots__r.r,r,r,r-r9srcseZdZdfdd 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,r- ]sz)CssExcelCell.__init__..r r!r"r#r,) frozensetitemssuperr.) r+r r!r"r#r5r7r8r9kwargsZdeclaration_dictZunique_declarations __class__r,r-r.Os    zCssExcelCell.__init__)r r&r!r&r#r4r5r6r7r&r8r&r9r:r(r))r/r0r1r. __classcell__r,r,rEr-r3Nsr3c@seZdZUdZeZdddddddZdddddddddddddd Zdddd Zd d d d dZ dddDZ de d<dfdgddZ e ZdhddZdhdd Zdid#d$Zdjd&d'Zdkd(d)Zdld+d,Zdmd.d/Zdnd3d4Zdod6d7Zdpd9d:Zdqd=d>Zdrd?d@ZdsdBdCZdtdEdFZdldGdHZdldIdJZdudLdMZdvdOdPZ dldQdRZ!dudSdTZ"dwdVdWZ#dxdYdZZ$dyd\d]Z%dzd`daZ&d{dbdcZ'dzdddeZ(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)rIztext-topmiddleZbaselinerKz text-bottomTF) boldZbolderZ600Z700Z800Z900normalZlighter100Z200Z300Z400500)rNitalicZoblique)Zserifz sans-serifZcursiveZfantasycCsi|]}||qSr,r;)r=r#r,r,r-r?szCSSToExcelConverter.) dashedZ mediumDashDotZ dashDotDothairdottedmediumDashDotDotdoubleZdashDotZ slantDashDot mediumDashedzdict[str, str] | None inheritedN str | Noner(r)cCs2|dur |||_nd|_tdd|j|_dS)N)maxsize) compute_cssr\r_call_uncached _call_cached)r+r\r,r,r-r.szCSSToExcelConverter.__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. )ra)r+rbr,r,r-__call__s zCSSToExcelConverter.__call__cCs|||j}||Sr*)r_r\ build_xlstyle)r+rb propertiesr,r,r-r`s 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)listrB isinstancedict)rnkv remove_noner,r-rvs z6CSSToExcelConverter.build_xlstyle..remove_none)rnror(r))build_alignment build_border build_fill build_fontbuild_number_format)r+rhoutr,rur-rfs  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+rhr,r,r-rwsz#CSSToExcelConverter.build_alignmentcC|d}|r |j|SdS)Nzvertical-align)r VERTICAL_MAP)r+rhZvertical_alignr,r,r-r  z+CSSToExcelConverter._get_vertical_alignment bool | NonecCs"|ddur dSt|ddvS)Nz white-space)Znowrapprezpre-line)rboolrr,r,r-rsz%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=Zsiderhr+r,r-r? s z4CSSToExcelConverter.build_border..rIrightrKleftr,rr,rr-rx 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)NZgrooveZridgeZinsetZoutsetsolidrZrX)rWthinrYrVr[zUnhandled border style format:  stacklevel)_get_width_nameBORDER_STYLE_MAPwarningswarnreprrr)r+r#rrZ width_namer,r,r-rs:    z!CSSToExcelConverter._border_style width_inputcCs2||}|dkr dS|dkrdS|dkrdSdS)Ngh㈵>g?rgffffff@ZmediumZthick)_width_to_float)r+rrr,r,r-rPs z#CSSToExcelConverter._get_width_namefloatcCs|durd}||S)NZ2pt) _pt_to_float)r+rr,r,r-rZs z#CSSToExcelConverter._width_to_float pt_stringstrcCs|dsJt|dS)Npt)endswithrrstrip)r+rr,r,r-r_sz CSSToExcelConverter._pt_to_floatcCs&|d}|dvr||ddSdS)Nzbackground-color)N transparentrr)ZfgColorZ patternType)rr)r+rhZ fill_colorr,r,r-rycs zCSSToExcelConverter.build_fillrocCs,|d}t|tr|ddn|}d|iS)Nz number-format§;Z format_code)rrqrreplace)r+rhfcr,r,r-r{js 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) namefamilysizerMrQrstrikerZshadow) _get_font_names_get_decoration_select_font_family_get_font_size _get_is_bold_get_is_italicrr _get_shadow)r+rh font_names decorationr,r,r-rzos   zCSSToExcelConverter.build_fontcCr)Nz font-weight)rBOLD_MAP)r+rhweightr,r,r-rrz CSSToExcelConverter._get_is_boldcCr)Nz font-style)r ITALIC_MAP)r+rhZ font_styler,r,r-rrz"CSSToExcelConverter._get_is_italic Sequence[str]cCs|d}|dur |SdS)Nztext-decorationr,)rsplit)r+rhrr,r,r-rs z#CSSToExcelConverter._get_decorationrcCsd|vrdSdS)Nrrr,)r+rr,r,r-_get_underlinesz"CSSToExcelConverter._get_underlinecCs d|vrttd|dSdS)Nz text-shadowz ^[^#(]*[1-9])rresearchrr,r,r-rszCSSToExcelConverter._get_shadowcCstd|dd}g}|D]3}|dddkr#|dddd}n|dddkr6|ddd d}n|}|rA||q|S) Nz(?x) ( "(?:[^"]|\\")+" | '(?:[^']|\\')+' | [^'",]+ )(?=,|\s*$) z font-familyrR"z\"'z\')rfindallrrstripappend)r+rhZfont_names_tmprrr,r,r-rs   z#CSSToExcelConverter._get_font_names float | NonecCs |d}|dur |S||S)Nz font-size)rr)r+rhrr,r,r-rs  z"CSSToExcelConverter._get_font_sizer'cCs(d}|D] }|j|}|r|Sq|Sr*) FAMILY_MAPr)r+rrrr,r,r-rs 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_COLORSKeyErrorrrrrrr+r"r,r,r-rs     z"CSSToExcelConverter.color_to_excel color_stringrcCst|dS)N#)r startswith)r+rr,r,r-rsz!CSSToExcelConverter._is_hex_colorcCsD|d}||r|dd|dd|ddS|S)NrrrSrR)lstrip_is_shorthand_colorupperr+rcoder,r,r-rs  (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,r-rs   z'CSSToExcelConverter._is_shorthand_colorr*)r\r]r(r))rbrcr(rd)rhrir(rd)rhrir(r})rhrir(r])rhrir(r)rhrir(r)r#r]rr]rr])rr]r(r])rr]r(r)rrr(r)rhri)rhrir(ro)rhrir(r)rhrir(r)rrr(r])rhrir(r)r(r')r"r]r(r])rrr(r)rrr(r))r/r0r1__doc__rrrrrrr__annotations__r.rr_rer`rfrwrrrxrrrrryr{rzrrrrrrrrrrrrr,r,r,r-rHgs            7              rHc@seZdZdZdZdZ         dDdEddZedFddZd d!Z dGd#d$Z dGd%d&Z dGd'd(Z dGd)d*Z dGd+d,ZdGd-d.ZedHd/d0ZdId3d4ZdGd5d6Zeed7d8 9 : :   dJdKdBdCZdS)LExcelFormattera9 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,r- _format_valueNs   zExcelFormatter._format_valueIterable[ExcelCell]ccs|jjdkr|jstd|js|jsdS|j}|j|jddd}t|}d}d}|jr>t |j jt r>t |j jdd}|jrt |jD]\}}t||||jdVqFt t||j|jD]B\}\}}} || } |D]1\} } d\} }| dkr||| | } }t||| d| | |jt|jdd|| |j| |d Vqoq_n+t t|D]$\} } d tt| }t||| d||jt|jdd|| |jd Vq||_dS) NrRz]Writing to Excel with MultiIndex columns and no index ('index'=False) is not yet implemented.FZsparsifyZadjoinnamesrr@NN ctx_columns r r!r"r#r5r7r8r9r$r%.r r!r"r#r5r7r8r9)rZnlevelsrNotImplementedError _has_aliasesrformatrrrqrrr enumeraterrrziplevelscodestakerBr3rrrjoinmaprr)r+r level_strs level_lengths coloffsetlnumrspansr level_codesvaluesispan_valr$r%rtr,r,r-_format_header_mi`sx          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)NrrRzWriting z cols but got z aliasesrr)rrrrqrrrrrr r rrr3rrrrr)r+rZcolnamescolindexZcolnamer,r,r-_format_header_regulars<   z%ExcelFormatter._format_header_regularcstjtr }n}d}jjjrFddjjjDdgtj}t ddt dd|rFfdd t |D}j d 7_ t ||S) Nr,cSsg|] }|dur |ndqSNrr,)r=xr,r,r- sz1ExcelFormatter._format_header..rcSs|o|Sr*r,)ryr,r,r-z/ExcelFormatter._format_header..cSs|dkSrr,)rr,r,r-rrc3s&|]\}}tj||jVqdSr*)rrr)r=rr"rr,r- s  z0ExcelFormatter._format_header..rR)rqrrrrrrrrrr rr itertoolschain)r+genZgen2r r,rr-_format_headers     zExcelFormatter._format_headercCst|jjtr |S|Sr*)rqrrr_format_hierarchical_rows_format_regular_rowsrr,r,r- _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)NrRrF ctx_indexr)rrrrrrqrptuplenpndarrayrrrrrrrrrZ to_timestamprr3rrr_generate_body)r+rZ index_valuesidxZidxvalrr,r,r-r%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|jjjdddd}t|}t||jjj|jjjD]J\}}} |j| |j|jd} |D]2\} } d\} }| dkr|j| | d} |}t|j| || | |jt|j dd| ||j!| |d Vq|d7}qun-t|jjD]&}t|D]\}}t|j||||jt|j dd|||j!d Vq|d7}q|"|EdHdS) NrRrFTr)Z allow_fillZ fill_valuerr'rr)#rrrrrrrrqrpr(r)r*rrrrcomZ any_not_nonerrrrrrrr r Z _can_hold_naZ _na_valuerBr3rrrr+)r+ZgcolidxZ index_labelsZcidxrr rrrrrrrr$r%Z indexcolvalsr,Z indexcolvalr,r,r-r$sz      z(ExcelFormatter._format_hierarchical_rowscCs t|jS)z1Whether the aliases for column names are present.)r rrr,r,r-r]s zExcelFormatter._has_aliasesrr&c csptt|jD]-}|jjdd|f}t|D]\}}t|j||||dt|j dd|||j dVqqdS)Nctxr) rangerrrZilocrr3rrrr)r+rZcolidxZseriesrr"r,r,r-r+bs   zExcelFormatter._generate_bodyccs4t||D] }||j|_|Vq dSr*)r r!r#r&rr")r+cellr,r,r-get_formatted_cellsrs z"ExcelFormatter.get_formatted_cellsstorage_options)r2Sheet1r sheet_namestartrowstartcol freeze_panestuple[int, int] | Noneenginerc Csddlm}|jj\} } | |jks| |jkr(td| d| d|jd|j|} t||r4d} n ||||d}d} z|j | ||||d W| rQ| d Sd S| rZ| ww) a  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} .. versionadded:: 1.2.0 r) ExcelWriterz-This sheet is too large! Your sheet size is: z, z Max sheet size is: F)r9r2T)r5r6r7N) Zpandas.io.excelr:rshapemax_rowsmax_colsrr1rqZ _write_cellsclose) r+writerr4r5r6r7r9r2r:Znum_rowsZnum_colsZformatted_cellsZ need_saver,r,r-writews<    zExcelFormatter.write) rNNTTNFrN)rrrr]rrrrrrrrrrrrrr:r(r))r(r)r(r)r(r)rr&r(r)r3rrNNN)r4rr5r&r6r&r7r8r9r]r2rr(r))r/r0r1rr<r=r.propertyrrrrr#r&r%r$rr+r1rrr@r,r,r,r-rsF  -   E    .O    r)9r __future__r functoolsrrr rtypingrrrrr r r rnumpyr)Zpandas._libs.libr Zpandas._typingr rZpandas.util._decoratorsrZpandas.util._exceptionsrZpandas.core.dtypesrZpandas.core.dtypes.commonrrZpandasrrrrZpandas.core.commoncorecommonr-Zpandas.core.shared_docsrZpandas.io.formats._color_datarZpandas.io.formats.cssrrZpandas.io.formats.formatrZpandas.io.formats.printingrrr3rHrr,r,r,r-s8 $