o jX*@stdZddlmZddlZddlmZddlmZGdddeZ Gdd d e Z Gd d d e Z Gd d d e Z dS)z"Chart builder and related objects.) annotationsN)contextmanager)Workbookcs@eZdZdZfddZeddZeddZdd Z Z S) _BaseWorkbookWriterz:Base class for workbook writers, providing shared members.cstt|||_dS)N)superr__init__ _chart_data)self chart_data __class__O/home/air/Certificate/back/venv/lib/python3.10/site-packages/pptx/chart/xlsx.pyrs z_BaseWorkbookWriter.__init__cCsPt}||\}}|||Wd|S1swY|S)z+bytes for Excel file containing chart_data.N)ioBytesIO_open_worksheet_populate_worksheetgetvaluer xlsx_fileworkbook worksheetr r r xlsx_blobs z_BaseWorkbookWriter.xlsx_blobccs.t|ddi}|}||fV|dS)z Enable XlsxWriter Worksheet object to be opened, operated on, and then automatically closed within a `with` statement. A filename or stream object (such as an `io.BytesIO` instance) is expected as *xlsx_file*. in_memoryTN)r add_worksheetcloserr r rrs   z#_BaseWorkbookWriter._open_worksheetcCstd)zy Must be overridden by each subclass to provide the particulars of writing the spreadsheet data. z!must be provided by each subclass)NotImplementedErrorr rrr r rr'sz'_BaseWorkbookWriter._populate_worksheet) __name__ __module__ __qualname____doc__rpropertyrrrr __classcell__r r r rr s    rc@s`eZdZdZeddZddZddZedd Z d d Z d d Z ddZ ddZ ddZdS)CategoryWorkbookWriterz Determines Excel worksheet layout and can write an Excel workbook from a CategoryChartData object. Serves as the authority for Excel worksheet ranges. cCsF|jj}|jdkr tdttd|jd}|jd}d||fS)z| The Excel worksheet reference to the categories for this chart (not including the column heading). rz!chart data contains no categoriesAzSheet1!$A$2:$%s$%d)r categoriesdepth ValueErrorchrord leaf_count)r r' right_col bottom_rowr r rcategories_ref6s    z%CategoryWorkbookWriter.categories_refcCsd||S)z Return the Excel worksheet reference to the cell containing the name for *series*. This also serves as the column heading for the series values. z Sheet1!$%s$1)_series_col_letterr seriesr r rseries_name_refCsz&CategoryWorkbookWriter.series_name_refcCs$djdi||t|ddS)zy The Excel worksheet reference to the values for this series (not including the column heading). z1Sheet1!${col_letter}$2:${col_letter}${bottom_row}r&) col_letterr.Nr )formatr0lenr1r r r values_refKs  z!CategoryWorkbookWriter.values_refcCsd|dks|dkr tdd}|r0|d}|dkrd}ttd|d}||}|dd}|s|S)zReturn str Excel column reference like 'BQ' for *column_number*. *column_number* is an int in the range 1-16384 inclusive, where 1 maps to column 'A'. r&i@z&column_number must be in range 1-16384rr%)r)r*r+) column_numbercol_ref remainderr4r r r_column_referenceWs z(CategoryWorkbookWriter._column_referencecCs||||||dS)a Write the chart data contents to *worksheet* in category chart layout. Write categories starting in the first column starting in the second row, and proceeding one column per category level (for charts having multi-level categories). Write series as columns starting in the next following column, placing the series title in the first cell. N)_write_categories _write_seriesrr r rrws z*CategoryWorkbookWriter._populate_worksheetcCsd|jj|j}||S)zj The letter of the Excel worksheet column in which the data for a series appears. r&)r'r(indexr=)r r2r:r r rr0s z)CategoryWorkbookWriter._series_col_letterc CsR|jj}|d|ji}|j}t|jD]\}}||d}|||||qdS)aF Write the categories column(s) to *worksheet*. Categories start in the first column starting in the second row, and proceeding one column per category level (for charts having multi-level categories). A date category is formatted as a date. All others are formatted `General`. num_formatr&N)rr' add_format number_formatr( enumeratelevels_write_cat_column) r rrr'rAr(idxlevelcolr r rr>s z(CategoryWorkbookWriter._write_categoriescCs8|||d|D]\}}|d}|||||q dS)z Write a category column defined by *level* to *worksheet* at offset *col* and formatted with *num_format*. r&N) set_columnwrite)r rrIrHrAoffnamerowr r rrFs  z(CategoryWorkbookWriter._write_cat_columncCs\|jjj}t|jD]!\}}|d|ji}||}|d||j|d||j |q dS)z Write the series column(s) to *worksheet*. Series start in the column following the last categories column, placing the series title in the first cell. rArr&N) rr'r(rDrBrCrLrN write_columnvalues)r rr col_offsetrGr2rA series_colr r rr?s z$CategoryWorkbookWriter._write_seriesN)rrr r!r"r/r3r7 staticmethodr=rr0r>rFr?r r r rr$/s     r$c@s8eZdZdZddZddZddZdd Zd d Zd S) XyWorkbookWriterz Determines Excel worksheet layout and can write an Excel workbook from XY chart data. Serves as the authority for Excel worksheet ranges. cCs||d}d|S)z Return the Excel worksheet reference to the cell containing the name for *series*. This also serves as the column heading for the series Y values. r&z Sheet1!$B$%d)series_table_row_offset)r r2rOr r rr3sz XyWorkbookWriter.series_name_refcCs|jd}|j}||S)zq Return the number of rows preceding the data table for *series* in the Excel worksheet. )r@data_point_offset)r r2title_and_spacer_rowsdata_point_rowsr r rrVs z(XyWorkbookWriter.series_table_row_offsetcC*||d}|t|d}d||fS)zx The Excel worksheet reference to the X values for this chart (not including the column label). rWr&zSheet1!$A$%d:$A$%drVr6r r2top_rowr.r r r x_values_ref zXyWorkbookWriter.x_values_refcCr[)zx The Excel worksheet reference to the Y values for this chart (not including the column label). rWr&zSheet1!$B$%d:$B$%dr\r]r r r y_values_refr`zXyWorkbookWriter.y_values_refcCsx|d|jji}|jD]-}|d|ji}||}||dd|j|||d|j||dd|j|q dS)a# Write chart data contents to *worksheet* in the standard XY chart layout. Write the data for each series to a separate two-column table, X values in column A and Y values in column B. Place the series label in the first (heading) cell of the column. rAr&rN) rBrrCrVrPx_valuesrLrNy_valuesr rrchart_num_formatr2series_num_formatoffsetr r rrs  z$XyWorkbookWriter._populate_worksheetN) rrr r!r3rVr_rarr r r rrUs   rUc@s eZdZdZddZddZdS)BubbleWorkbookWriterz^ Service object that knows how to write an Excel workbook from bubble chart data. cCr[)z The Excel worksheet reference to the range containing the bubble sizes for *series* (not including the column heading cell). rWr&zSheet1!$C$%d:$C$%dr\r]r r rbubble_sizes_refr`z%BubbleWorkbookWriter.bubble_sizes_refcCs|d|jji}|jD]?}|d|ji}||}||dd|j|||d|j||dd|j|||dd||dd|j |q dS)aN Write chart data contents to *worksheet* in the bubble chart layout. Write the data for each series to a separate three-column table with X values in column A, Y values in column B, and bubble sizes in column C. Place the series label in the first (heading) cell of the values column. rAr&rrWSizeN) rBrrCrVrPrbrLrNrc bubble_sizesrdr r rrs  z(BubbleWorkbookWriter._populate_worksheetN)rrr r!rirr r r rrhs rh) r! __future__rr contextlibr xlsxwriterrobjectrr$rUrhr r r rs   $<