a j=ic|@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,eZdZdZeZd ddddddddZdS) ExcelCellrowcolvalstyle mergestartmergeendNint int | NoneNone)r r!r$r%returncCs(||_||_||_||_||_||_dSNr)selfr r!r"r#r$r%r,h/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/pandas/io/formats/excel.py__init__=s zExcelCell.__init__)NNN)__name__ __module__ __qualname__Z __fields__ __slots__r.r,r,r,r-r9s rc s0eZdZdddddddddfdd ZZS) CssExcelCellr&z dict | Nonez3dict[tuple[int, int], list[tuple[str, Any]]] | NoneCallable | Noner()r r!r# css_stylescss_rowcss_col css_converterr)c  sR|r2|r2dd|||fD} t| } || }tjf||||d| dS)NcSsi|]\}}||qSr,)lower).0propr"r,r,r- ]sz)CssExcelCell.__init__..r r!r"r#) frozensetitemssuperr.) r+r r!r"r#r5r6r7r8kwargsZdeclaration_dictZunique_declarations __class__r,r-r.Os   zCssExcelCell.__init__)r/r0r1r. __classcell__r,r,rBr-r3Nsr3c@seZdZUdZeZdddddddZdddddddddddddd Zdddd Zd d d d dZ de d<dcdddddZ e Z edddddddZddddd Zdd!dd"d#Zdddd$d%Zdd&dd'd(Zdd)dd*d+Zdddd,d-d.Zddd/d0d1Zdd2d3d4d5Zd6d2d7d8d9Zdd:d;d<Zdd=dd>d?Zdd@ddAdBZdd&ddCdDZdd&ddEdFZddGddHdIZdGddJdKdLZdd&ddMdNZ ddGddOdPZ!ddQddRdSZ"dTdUdVdWZ#dddXdYdZZ$d6d[d\d]d^Z%d6d6d\d_d`Z&d6d[d\dadbZ'dS)dCSSToExcelConverteraK 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)rFztext-topmiddleZbaselinerHz text-bottomTF) boldZbolderZ600Z700Z800Z900normalZlighter100Z200Z300Z400500)rKitalicZoblique)Zserifz sans-serifZcursiveZfantasyzdict[str, str] | None inheritedN str | Noner()rSr)cCs |dur|||_nd|_dSr*) compute_cssrS)r+rSr,r,r-r.szCSSToExcelConverter.__init__)maxsizez str | frozenset[tuple[str, str]]zdict[str, dict[str, str]]) declarationsr)cCs|||j}||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. )rUrS build_xlstyle)r+rW propertiesr,r,r-__call__szCSSToExcelConverter.__call__zMapping[str, str])propsr)csN||||||||||d}dddfdd ||S)N) alignmentborderfillfontZ number_formatdict[str, str | None]r()dr)csFt|D]4\}}|dur$||=q t|tr ||s ||=q dS)z4Remove key where value is None, through nested dictsN)listr? isinstancedict)rakv remove_noner,r-rhs z6CSSToExcelConverter.build_xlstyle..remove_none)build_alignment build_border build_fill build_fontbuild_number_format)r+r[outr,rgr-rXs  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+r[r,r,r-risz#CSSToExcelConverter.build_alignmentcCs|d}|r|j|SdS)Nzvertical-align)rr VERTICAL_MAP)r+r[Zvertical_alignr,r,r-rss  z+CSSToExcelConverter._get_vertical_alignmentz bool | NonecCs"|ddurdSt|ddvS)Nz white-space)Znowrapprezpre-line)rrboolrur,r,r-rtsz%CSSToExcelConverter._get_is_wrap_textz dict[str, dict[str, str | None]]csfdddDS)Ncsfi|]^}|d|dd|dd|dd|ddqS)zborder-z-stylez-widthz-color)r#color) _border_stylerrcolor_to_excel)r:sider[r+r,r-r<s z4CSSToExcelConverter.build_border..rFrightrHleftr,rur,r}r-rjs z CSSToExcelConverter.build_border)r#widthrycCs|dur|dur|durdS|dur0|dur0dS|dks@|dkrDdS||}|durZdS|dvrf|S|dkrrdS|dkr|dvrdSdS|dkr|dvrdSd SdS) Nnonehidden)NZgrooveZridgeZinsetZoutsetsoliddoubleZdotted)ZhairthinZmediumDashDotDotZdashedZ mediumDashed)_get_width_name)r+r#rryZ width_namer,r,r-rzs* z!CSSToExcelConverter._border_style) width_inputr)cCs2||}|dkrdS|dkr"dS|dkr.dSdS)Ngh㈵>g?rgffffff@ZmediumZthick)_width_to_float)r+rrr,r,r-r.s z#CSSToExcelConverter._get_width_namefloat)rr)cCs|dur d}||S)NZ2pt) _pt_to_float)r+rr,r,r-r8sz#CSSToExcelConverter._width_to_floatstr) pt_stringr)cCs|dsJt|dS)Npt)endswithrrstrip)r+rr,r,r-r=sz CSSToExcelConverter._pt_to_float)r[cCs&|d}|dvr"||ddSdS)Nzbackground-color)N transparentrr)ZfgColorZ patternType)rrr{)r+r[Z fill_colorr,r,r-rkAs zCSSToExcelConverter.build_fillr`cCs,|d}t|tr |ddn|}d|iS)Nz number-format§;Z format_code)rrrcrreplace)r+r[Zfcr,r,r-rmHs z'CSSToExcelConverter.build_number_formatz$dict[str, bool | float | str | None]c Csv||}||}|r |dnd||||||||d|vrNdndd|vpZd||d||d S)Nr underlinesinglez line-throughry) namefamilysizerJrNrstrikeryZshadow) _get_font_names_get_decoration_select_font_family_get_font_size _get_is_bold_get_is_italicr{rr _get_shadow)r+r[ font_names decorationr,r,r-rlMs   zCSSToExcelConverter.build_fontcCs|d}|r|j|SdS)Nz font-weight)rrBOLD_MAP)r+r[weightr,r,r-r_s  z CSSToExcelConverter._get_is_boldcCs|d}|r|j|SdS)Nz font-style)rr ITALIC_MAP)r+r[Z font_styler,r,r-res  z"CSSToExcelConverter._get_is_italicz Sequence[str]cCs"|d}|dur|SdSdS)Nztext-decorationr,)rrsplit)r+r[rr,r,r-rks z#CSSToExcelConverter._get_decoration)rr)cCsd|vr dSdS)Nrrr,)r+rr,r,r-_get_underlinersz"CSSToExcelConverter._get_underlinecCs d|vrttd|dSdS)Nz text-shadowz ^[^#(]*[1-9])rxresearchrur,r,r-rwszCSSToExcelConverter._get_shadowcCstd|dd}g}|D]f}|dddkrF|dddd}n.|dddkrl|ddd d}n|}|r||q|S) Nz(?x) ( "(?:[^"]|\\")+" | '(?:[^']|\\')+' | [^'",]+ )(?=,|\s*$) z font-familyrO"z\"'z\')rfindallrrrstripappend)r+r[Zfont_names_tmprrr,r,r-r|s   z#CSSToExcelConverter._get_font_namesz float | NonecCs |d}|dur|S||S)Nz font-size)rrr)r+r[rr,r,r-rs z"CSSToExcelConverter._get_font_sizer'r)cCs&d}|D]}|j|}|rq"q|Sr*) FAMILY_MAPrr)r+rrrr,r,r-rs  z'CSSToExcelConverter._select_font_family)r"r)cCs`|dur dS||r ||Sz |j|WStyZtjdt|ttdYn0dS)NzUnhandled color format: ) stacklevel) _is_hex_color_convert_hex_to_excel NAMED_COLORSKeyErrorwarningswarnreprrrr+r"r,r,r-r{s      z"CSSToExcelConverter.color_to_excelrx) color_stringr)cCst|dS)N#)rx startswith)r+rr,r,r-rsz!CSSToExcelConverter._is_hex_colorcCsH|d}||r<|dd|dd|ddS|SdS)NrrrPrO)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/r0r1__doc__rrrvrrr__annotations__r.rrUrrZrXrirsrtrjrzrrrrkrmrlrrrrrrrrr{rrrr,r,r,r-rEgsz   -  rEc @seZdZdZdZdZd8d d d d d dd d ddd ddZeddddZddZ ddddZ ddddZ dddd Z ddd!d"Z ddd#d$Zddd%d&Zed dd'd(Zd)dd*d+d,Zddd-d.Zeed/d0d9d d)d)d3d d4dd5d6d7ZdS):ExcelFormattera9 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@rNTFinfrrTzSequence[Hashable] | NonezSequence[Hashable] | boolrxzIndexLabel | Noner4r() na_rep float_formatcolsheaderindex index_label merge_cellsinf_repstyle_converterr)c Csd|_||_t|tsB||_|j|j}| dur:t} | |_n d|_d|_||_ |durt 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_computedatarErdfrr intersectionrrsetZreindexrrrrrr) r+rrrrrrrrrrr,r,r-r.s2     zExcelFormatter.__init__z dict[str, dict[str, str | bool]]rcCs ddidddddddddS) NrJTrr~rGrF)rorp)r_Zbordersr\r,r+r,r,r- header_style szExcelFormatter.header_stylecCst|rt|r|j}nJt|rdt|r4|j}n0t|rLd|j}n|jdurdt |j|}t |dddur|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_value-s   zExcelFormatter._format_valuezIterable[ExcelCell]ccs|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) NrOz]Writing to Excel with MultiIndex columns and no index ('index'=False) is not yet implemented.FZsparsifyZadjoinnamesrr=NN ctx_columns r r!r"r#r5r6r7r8r$r%.r r!r"r#r5r6r7r8)rZnlevelsrNotImplementedError _has_aliasesrformatrrrcrrr enumeraterrrziplevelscodestaker?r3rrrjoinmaprr)r+r level_strs level_lengths coloffsetlnumrspansr level_codesvaluesispan_valr$r%rfr,r,r-_format_header_mi?sr          z ExcelFormatter._format_header_mic cs|js |jrd}|jr6d}t|jjtr6t|jjj}|j}|jrt t |j|_t|jt|jkrt dt|jdt|jdn|j}t |D]6\}}t |j||||jt|jddd||jdVqdS)NrrOzWriting z cols but got z aliasesrr)rrrrcrrrrrr r rrr3rrrrr)r+rZcolnamescolindexZcolnamer,r,r-_format_header_regulars6  z%ExcelFormatter._format_header_regularcstjtr}n}d}jjjrddjjjDdgtj}t ddt dd|rfdd t |D}j d 7_ t ||S) Nr,cSsg|]}|dur|ndqSNrr,)r:xr,r,r- z1ExcelFormatter._format_header..rcSs|o|Sr*r,)ryr,r,r-rz/ExcelFormatter._format_header..cSs|dkSrr,)rr,r,r-rrc3s$|]\}}tj||jVqdSr*)rrr)r:rr"rr,r- sz0ExcelFormatter._format_header..rO)rcrrrrrrrrrrrr itertoolschain)r+genZgen2r r,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|jdurt|jdd||jV|j j}t|j jtr|j j}t|D]6\}}t|j|d||jt|jdd|d|jdVqd}nd}||EdHdS)NrOrF ctx_indexr)rrrrrrcrbtuplenpndarrayrrrrrrrrrZ to_timestamprr3rrr_generate_body)r+rZ index_valuesidxZidxvalrr,r,r-r s>      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|jdurt|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) NrOrFTr)Z allow_fillZ fill_valuerrrr)#rrrrrrrrcrbrrrrrrrcomZ any_not_nonerrrrrrrrrZ _can_hold_naZ _na_valuer?r3rrrr)r+ZgcolidxZ index_labelsZcidxrrrrrrrrrr$r%Z indexcolvalsrZ indexcolvalr,r,r-r sv       z(ExcelFormatter._format_hierarchical_rowscCs t|jS)z1Whether the aliases for column names are present.)r rrr,r,r-r?szExcelFormatter._has_aliasesr&)rr)c csntt|jD]Z}|jjdd|f}t|D]8\}}t|j||||dt|j dd|||j dVq.qdS)Nctxr) rangerrrZilocrr3rrrr)r+rZcolidxZseriesrr"r,r,r-rDs zExcelFormatter._generate_bodyccs2t||D]}||j|_|VqdSr*)rrr r rr")r+cellr,r,r-get_formatted_cellsTsz"ExcelFormatter.get_formatted_cellsstorage_options)rSheet1rztuple[int, int] | Noner) sheet_namestartrowstartcol freeze_panesenginerr)c Csddlm}|jj\} } | |jks,| |jkrPtd| d| d|jd|j|} t||rhd} n||||d}d} z$|j | ||||d W| r| n| r| 0d 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)rrT)rrrN) Zpandas.io.excelr rshapemax_rowsmax_colsrrrcZ _write_cellsclose) r+writerrrrrrrr Znum_rowsZnum_colsZformatted_cellsZ need_saver,r,r-writeYs:&     zExcelFormatter.write) rNNTTNFrN)rrrNNN)r/r0r1rr"r#r.propertyrrrrr r r r rrrrrr&r,r,r,r-rsD ". E .Q r)9r __future__r functoolsrrrrtypingrrrrr r r rnumpyrZpandas._libs.libr Zpandas._typingr rZpandas.util._decoratorsrZpandas.util._exceptionsrZpandas.core.dtypesrZpandas.core.dtypes.commonrrZpandasrrrrZpandas.core.commoncorecommonrZpandas.core.shared_docsrZpandas.io.formats._color_datarZpandas.io.formats.cssrrZpandas.io.formats.formatrZpandas.io.formats.printingrrr3rErr,r,r,r-s4 $         i