U +ifZy@sLdZddlmZddlmZddlZddlZddlmZm Z m Z m Z m Z m Z mZddlZddlZddlmZddlmZmZddlmZdd 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/ddl0m1Z1GdddZ2Gddde2Z3GdddZ4GdddZ5dS)zC Utilities for conversion to writer-agnostic Excel representation. ) annotations)reduceN)AnyCallableHashableIterableMappingSequencecast) is_list_like) IndexLabelStorageOptions)doc)missing)is_float is_scalar) DataFrameIndex MultiIndex PeriodIndex) _shared_docs) CSS4_COLORS) CSSResolver CSSWarning)get_level_lengths) pprint_thingc@s*eZdZdZeZddddddddZdS) ExcelCellrowcolvalstyle mergestartmergeendNint int | None)rrr"r#cCs(||_||_||_||_||_||_dSNr)selfrrr r!r"r#r(J/home/mars/bis/venv/lib/python3.8/site-packages/pandas/io/formats/excel.py__init__9s zExcelCell.__init__)NNN)__name__ __module__ __qualname__Z __fields__ __slots__r*r(r(r(r)r5s rcs.eZdZddddddddfdd ZZS) CssExcelCellr$z dict | Nonez3dict[tuple[int, int], list[tuple[str, Any]]] | NoneCallable | None)rrr! css_stylescss_rowcss_col css_converterc  sH|r,|r,ddd|||fD} || }tjf||||d| S)N;cSs g|]\}}|dt|qS):)str).0avr(r(r) Ysz)CssExcelCell.__init__..rrr r!)joinsuperr*) r'rrr r!r1r2r3r4kwargscss __class__r(r)r*Ks zCssExcelCell.__init__)r+r,r-r* __classcell__r(r(rAr)r/Jsr/c@s eZdZUdZeZdddddddZdddddddddddddd Zdddd Zd d d d dZ de d<d`ddddZ e Z dddddZdddddZdddd d!Zdddd"d#Zdd$dd%d&Zdd'dd(d)Zddd*d+d,Zddd-d.d/Zdd0d1d2d3Zdd0d4d5d6Zdd7d8d9Zdd:dd;d<Zdd=dd>d?Zdd$dd@dAZdd$ddBdCZddDddEdFZdDddGdHdIZdd$ddJdKZddDddLdMZ ddNddOdPZ!dQdRdSdTZ"dddUdVdWZ#ddXdYdZd[Z$dddYd\d]Z%ddXdYd^d_Z&dS)aCSSToExcelConverteraK 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)rEztext-topmiddleZbaselinerGz text-bottomTF) boldZbolderZ600Z700Z800Z900normalZlighter100Z200Z300Z400500)rJitalicZoblique)Zserifz sans-serifZcursiveZfantasyzdict[str, str] | None inheritedN str | None)rRcCs |dk r|||_nd|_dSr&) compute_cssrR)r'rRr(r(r)r*szCSSToExcelConverter.__init__r7zdict[str, dict[str, str]])declarations_strreturncCs|||j}||S)ax Convert CSS declarations to ExcelWriter style. Parameters ---------- declarations_str : str List of CSS declarations. e.g. "font-weight: bold; background: blue" Returns ------- xlstyle : dict A style as interpreted by ExcelWriter when found in ExcelCell.style. )rTrR build_xlstyle)r'rU propertiesr(r(r)__call__szCSSToExcelConverter.__call__zMapping[str, str])propsrVcsN||||||||||d}dddfdd ||S)N) alignmentborderfillfontZ number_formatzdict[str, str]None)drVcsFt|D]4\}}|dkr$||=q t|tr ||s ||=q dS)z4Remove key where value is None, through nested dictsN)listitems isinstancedict)r`kr: remove_noner(r)rgs z6CSSToExcelConverter.build_xlstyle..remove_none)build_alignment build_border build_fill build_fontbuild_number_format)r'rZoutr(rfr)rWs  z!CSSToExcelConverter.build_xlstylezdict[str, bool | str | None]cCs|d||||dS)Nz text-align) horizontalvertical wrap_text)get_get_vertical_alignment_get_is_wrap_textr'rZr(r(r)rhsz#CSSToExcelConverter.build_alignmentcCs|d}|r|j|SdS)Nzvertical-align)rq VERTICAL_MAP)r'rZZvertical_alignr(r(r)rrs  z+CSSToExcelConverter._get_vertical_alignmentz bool | NonecCs"|ddkrdSt|ddkS)Nz white-space)Znowrapprezpre-line)rqboolrtr(r(r)rssz%CSSToExcelConverter._get_is_wrap_textz dict[str, dict[str, str | None]]csfdddDS)Nc sPi|]H}|d|dd|dd|ddqS)zborder-z-stylez-widthz-color)r!color) _border_stylerqcolor_to_excel)r8ZsiderZr'r(r) sz4CSSToExcelConverter.build_border..rErightrGleftr(rtr(r{r)ris z CSSToExcelConverter.build_border)r!widthcCs|dkr|dkrdS|dks$|dkr(dS||}|dkr>dS|dkrJ|S|dkrVdS|dkrn|dkrjdSdS|dkr|dkrdSd SdS) NnoneZhidden)NZgrooveZridgeZinsetZoutsetsoliddoubleZdotted)ZhairthinZmediumDashDotDotZdashedZ mediumDashed)_get_width_name)r'r!rZ width_namer(r(r)rys& z!CSSToExcelConverter._border_style) width_inputrVcCs2||}|dkrdS|dkr"dS|dkr.dSdS)Ngh㈵>g?rgffffff@ZmediumZthick)_width_to_float)r'rrr(r(r)rs z#CSSToExcelConverter._get_width_namefloat)rrVcCs|dkr d}||S)NZ2pt) _pt_to_float)r'rr(r(r)r'sz#CSSToExcelConverter._width_to_float) pt_stringrVcCs|dstt|dS)Npt)endswithAssertionErrorrrstrip)r'rr(r(r)r,sz CSSToExcelConverter._pt_to_float)rZcCs&|d}|dkr"||ddSdS)Nzbackground-color)N transparentrr)ZfgColorZ patternType)rqrz)r'rZZ fill_colorr(r(r)rj0s zCSSToExcelConverter.build_fillzdict[str, str | None]cCs,|d}t|tr |ddn|}d|iS)Nz number-format§r5Z format_code)rqrcr7replace)r'rZfcr(r(r)rl7s z'CSSToExcelConverter.build_number_formatz*dict[str, bool | int | float | str | None]c Csv||}||}|r |dnd||||||||d|krNdndd|kpZd||d||d S)Nr underlinesinglez line-throughrx) namefamilysizerIrMrstrikerxZshadow) _get_font_names_get_decoration_select_font_family_get_font_size _get_is_bold_get_is_italicrzrq _get_shadow)r'rZ font_names decorationr(r(r)rk<s   zCSSToExcelConverter.build_fontcCs|d}|r|j|SdS)Nz font-weight)rqBOLD_MAP)r'rZweightr(r(r)rNs  z CSSToExcelConverter._get_is_boldcCs|d}|r|j|SdS)Nz font-style)rq ITALIC_MAP)r'rZZ font_styler(r(r)rTs  z"CSSToExcelConverter._get_is_italicz Sequence[str]cCs"|d}|dk r|SdSdS)Nztext-decorationr()rqsplit)r'rZrr(r(r)rZs z#CSSToExcelConverter._get_decoration)rrVcCsd|kr dSdS)Nrrr()r'rr(r(r)_get_underlineasz"CSSToExcelConverter._get_underlinecCs d|krttd|dSdS)Nz text-shadowz ^[^#(]*[1-9])rwresearchrtr(r(r)rfszCSSToExcelConverter._get_shadowcCstd|dd}g}|D]f}|dddkrF|dddd}n.|dddkrl|ddd d}n|}|r||q|S) Nz(?x) ( "(?:[^"]|\\")+" | '(?:[^']|\\')+' | [^'",]+ )(?=,|\s*$) z font-familyrN"z\"'z\')rfindallrqrstripappend)r'rZZfont_names_tmprrr(r(r)rks   z#CSSToExcelConverter._get_font_namesz float | NonecCs |d}|dkr|S||S)Nz font-size)rqr)r'rZrr(r(r)rs z"CSSToExcelConverter._get_font_sizer%rVcCs&d}|D]}|j|}|rq"q|Sr&) FAMILY_MAPrq)r'rrrr(r(r)rs  z'CSSToExcelConverter._select_font_family)r rVcCs\|dkr dS||r ||Sz |j|WStk rVtdt|tYnXdS)NzUnhandled color format: ) _is_hex_color_convert_hex_to_excel NAMED_COLORSKeyErrorwarningswarnreprrr'r r(r(r)rzs   z"CSSToExcelConverter.color_to_excelrw) color_stringrVcCst|dS)N#)rw startswith)r'rr(r(r)rsz!CSSToExcelConverter._is_hex_colorcCsH|d}||r<|dd|dd|ddS|SdS)NrrrOrN)lstrip_is_shorthand_colorupperr'rcoder(r(r)rs  (z)CSSToExcelConverter._convert_hex_to_excelcCs<|d}t|dkrdSt|dkr*dStd|dS)zcCheck if color code is shorthand. #FFF is a shorthand as opposed to full #FFFFFF. rTFzUnexpected color N)rlen ValueErrorrr(r(r)rs    z'CSSToExcelConverter._is_shorthand_color)N)'r+r,r-__doc__rrrurrr__annotations__r*rrTrYrWrhrrrsriryrrrrjrlrkrrrrrrrrrzrrrr(r(r(r)rD`sx   '   rDc @seZdZdZdZdZd4d d d d d dd d dd ddZeddZddZ ddddZ ddddZ ddddZ dddd Z ddd!d"Zddd#d$Zed dd%d&Zd'dd(d)d*Zddd+d,Zeed-d.d5d1d.d2d3ZdS)6ExcelFormattera9 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@rNTFinfr7rSzSequence[Hashable] | NonezSequence[Hashable] | boolrwzIndexLabel | Noner0) na_rep float_formatcolsheaderindex index_label merge_cellsinf_repstyle_converterc Csd|_||_t|tsB||_|j|j}| dkr:t} | |_n d|_d|_||_ |dk rt t | |j sxtdt t | |j t t|krtd|j|d|_ |j j |_ ||_||_||_||_||_| |_dS)Nrz,passes columns are not ALL present dataframez.Not all names specified in 'columns' are found)columns) rowcounterrrcrstylerZ_computedatarDrdfrr intersectionrrsetZreindexrrrrrr) r'rrrrrrrrrrr(r(r)r*s2     zExcelFormatter.__init__cCs ddidddddddddS) NrITrr}rFrE)rnro)r^Zbordersr[r(r'r(r(r) header_style szExcelFormatter.header_stylecCst|rt|r|j}nJt|rdt|r4|j}n0t|rLd|j}n|jdk rdt |j|}t |dddk r|t 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_values   zExcelFormatter._format_valuezIterable[ExcelCell]rccs|jjdkr|jstd|js*|js*dS|j}|j|jddd}t|}d}d}|jrzt |j jt rzt |j jdd}|jrHt |jD]\}}t||||jdVqt t||j|jD]\}\}}} || } |D]d\} } d\} }| dkr ||| | } }t||| d| | |jt|jdd|| |j| |d VqqnXt t|D]J\} } d tt| }t||| d||jt|jdd|| |jd VqT||_dS) NrNz]Writing to Excel with MultiIndex columns and no index ('index'=False) is not yet implemented.FZsparsifyZadjoinnamesrr<NN ctx_columns rrr r!r1r2r3r4r"r#.rrr r!r1r2r3r4)rZnlevelsrNotImplementedError _has_aliasesrformatrrrcrrr enumeraterrrziplevelscodestakerbr/rrrr=maprr)r'r level_strs level_lengths coloffsetlnumrspansr level_codesvaluesispan_valr"r#r:r(r(r)_format_header_mi*sv          z ExcelFormatter._format_header_mic cs|js |jrd}|jr8d}t|jjtr8t|jjd}|j}|jrtt |j|_t|jt|jkrt dt|jdt|jdn|j}t |D]6\}}t |j ||||jt|jddd||jdVqdS)NrrNzWriting z cols but got z aliasesrr)rrrrcrrrrr r rrr/rrrrr)r'rZcolnamescolindexZcolnamer(r(r)_format_header_regularos2  z%ExcelFormatter._format_header_regularcstjtr}n}d}jjjrddjjjDdgtj}t ddt dd|rfdd t |D}j d 7_ t ||S) Nr(cSsg|]}|dk r|ndqSNrr()r8xr(r(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)r8rr rr(r) sz0ExcelFormatter._format_header..rN)rcrrrrrrrrrrrr itertoolschain)r'genZgen2rr(rr)_format_headers     zExcelFormatter._format_headercCs"t|jjtr|S|SdSr&)rcrrr_format_hierarchical_rows_format_regular_rowsrr(r(r) _format_bodyszExcelFormatter._format_bodyc cs2|js |jr|jd7_|jr|jrJt|jtttj t frJ|jd}n(|jrdt|jt rd|j}n|j jj d}t|jtr|jd7_|r|jdk rt|jdd||jV|j j}t|j jtr|j j}t|D]6\}}t|j|d||jt|jdd|d|jdVqd}nd}||EdHdS)NrNrF ctx_indexr)rrrrrrcratuplenpndarrayrr7rrrrrrrZ to_timestamprr/rrr_generate_body)r'rZ index_valuesidxZidxvalrr(r(r)rs@       z#ExcelFormatter._format_regular_rowsccs|js |jr|jd7_d}|jr|jjj}|jrRt|jtt t j t frR|j}t|j trr|jrr|jd7_tj|r|jdk rt|D] \}}t|jd|||jVq|jr|jjjdddd}t|}t||jjj|jjjD]\}}} |j| |j|jd} |D]h\} } d\} }| dkrB|j| | d} |}t|j| || | |jt|j dd| ||j!| |d Vq|d7}qn^t|jjD]P}t|D]8\}}t|j||||jt|j dd|||j!d Vq|d7}q|"|EdHdS) NrNrFTr)Z allow_fillZ fill_valuerr rr)#rrrrrrrrcrar r r rrrrcomZ any_not_nonerrrrrrrrrZ _can_hold_naZ _na_valuerbr/rrrr )r'ZgcolidxZ index_labelsZcidxrrrrrrrrrr"r#Z indexcolvalsrZ indexcolvalr(r(r)rs        z(ExcelFormatter._format_hierarchical_rowscCs t|jS)z1Whether the aliases for column names are present.)r rrr(r(r)r)szExcelFormatter._has_aliasesr$)rrVc csntt|jD]Z}|jjdd|f}t|D]8\}}t|j||||dt|j dd|||j dVq.qdS)Nctxr) rangerrrZilocrr/rrrr)r'rZcolidxZseriesrr r(r(r)r .s zExcelFormatter._generate_bodyccs2t||D]}||j|_|VqdSr&)rrrrrr )r'cellr(r(r)get_formatted_cells>sz"ExcelFormatter.get_formatted_cellsstorage_options)rSheet1rr c Csddlm}|jj\} } | |jks,| |jkrPtd| d| d|jd|j|} t||rhd} n||||d}d} z|j | ||||d W5| r| Xd S) 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``, ``io.excel.xls.writer``, and ``io.excel.xlsm.writer``. .. deprecated:: 1.2.0 As the `xlwt `__ package is no longer maintained, the ``xlwt`` engine will be removed in a future version of pandas. {storage_options} .. versionadded:: 1.2.0 r) ExcelWriterz-This sheet is too large! Your sheet size is: z, z Max sheet size is: F)enginerT)startrowstartcol freeze_panesN) Zpandas.io.excelrrshapemax_rowsmax_colsrrrccloseZ write_cells) r'writerZ sheet_namerrrrrrZnum_rowsZnum_colsZformatted_cellsZ need_saver(r(r)writeCs2&    zExcelFormatter.write) rNNTTNFrN)rrrNNN)r+r,r-rrrr*propertyrrrrrrrrrr rrrr r(r(r(r)rsD  . E .Q r)6r __future__r functoolsrrrtypingrrrrrr r rnumpyr Zpandas._libs.libr Zpandas._typingr r Zpandas.util._decoratorsrZpandas.core.dtypesrZpandas.core.dtypes.commonrrZpandasrrrrZpandas.core.commoncorecommonrZpandas.core.shared_docsrZpandas.io.formats._color_datarZpandas.io.formats.cssrrZpandas.io.formats.formatrZpandas.io.formats.printingrrr/rDrr(r(r(r)s2  $        [