hiX*dZddlmZddlZddlmZddlmZGddeZ Gdd e Z Gd d e Z Gd d e Z dS)z"Chart builder and related objects.) annotationsN)contextmanager)WorkbookcTeZdZdZfdZedZedZdZ xZ S)_BaseWorkbookWriterz:Base class for workbook writers, providing shared members.cdtt|||_dS)N)superr__init__ _chart_data)self chart_data __class__s g/var/www/development/aibuddy-work/election-extract/venv/lib/python3.11/site-packages/pptx/chart/xlsx.pyr z_BaseWorkbookWriter.__init__s. !4((11333%ctj}||5\}}|||dddn #1swxYwY|S)z+bytes for Excel file containing chart_data.N)ioBytesIO_open_worksheet_populate_worksheetgetvaluer xlsx_fileworkbook worksheets r xlsx_blobz_BaseWorkbookWriter.xlsx_blobsJLL  ! !) , , :0E9  $ $Xy 9 9 9 : : : : : : : : : : : : : : :!!###sAAAc#Kt|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_worksheetclosers rrz#_BaseWorkbookWriter._open_worksheetsTI T':;;**,,  !!!!rc td)zy Must be overridden by each subclass to provide the particulars of writing the spreadsheet data. z!must be provided by each subclass)NotImplementedErrorr rrs rrz'_BaseWorkbookWriter._populate_worksheet's ""EFFFr) __name__ __module__ __qualname____doc__r propertyrrrr __classcell__)rs@rrr sDD&&&&&$$X$  ^ GGGGGGGrrcheZdZdZedZdZdZedZ dZ dZ dZ d Z d Zd S) CategoryWorkbookWriterz Determines Excel worksheet layout and can write an Excel workbook from a CategoryChartData object. Serves as the authority for Excel worksheet ranges. c|jj}|jdkrtdt t d|jzdz }|jdz}d||fzS)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_rows rcategories_refz%CategoryWorkbookWriter.categories_ref6si %0  q @AA AC:#33a788 *Q. #y*&===rc2d||zS)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 seriess rseries_name_refz&CategoryWorkbookWriter.series_name_refCs  7 7 ? ???rchdjdi||t|dzdS)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_letterr5)formatr8lenr9s r values_refz!CategoryWorkbookWriter.values_refKsO JBI  "55f==!&kkAo   rc|dks|dkrtdd}|r>|dz}|dkrd}ttd|zdz }||z}|dz dz}|>|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,)r0r1r2) column_numbercol_ref remainderr=s r_column_referencez(CategoryWorkbookWriter._column_referenceWs 1    5 5EFF F  6%*IA~~ SXX 1A566J 7*G+Q.25M 6rc^||||||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_seriesr"s rrz*CategoryWorkbookWriter._populate_worksheetws6 x333 8Y/////rcZd|jjz|jz}||S)zj The letter of the Excel worksheet column in which the data for a series appears. r-)r.r/indexrH)r r:rEs rr8z)CategoryWorkbookWriter._series_col_letters/ F-33flB %%m444rc|jj}|d|ji}|j}t |jD]%\}}||z dz }|||||&dS)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)r r. add_format number_formatr/ enumeratelevels_write_cat_column) r rrr.rOr/idxlevelcols rrJz(CategoryWorkbookWriter._write_categoriess%0 ((, 8P)QRR  #J$566 F FJC#+/C  " "9c5* E E E E F Frc~|||d|D]"\}}|dz}|||||#dS)z Write a category column defined by *level* to *worksheet* at offset *col* and formatted with *num_format*. r-N) set_columnwrite)r rrWrVrOoffnamerows rrTz(CategoryWorkbookWriter._write_cat_columns] S#r*** 8 8IC'C OOCdJ 7 7 7 7 8 8rc|jjj}t|jD]_\}}|d|ji}||z}|d||j|d||j |`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. rOrr-N) r r.r/rRrPrQr[r] write_columnvalues)r rr col_offsetrUr:rO series_cols rrKz$CategoryWorkbookWriter._write_seriess %06 $T%566 M MKC!,,lFrrr*r*/s  > >X >@@@    \> 0 0 0555 F F F888 M M M M Mrr*c0eZdZdZdZdZdZdZdZdS)XyWorkbookWriterz Determines Excel worksheet layout and can write an Excel workbook from XY chart data. Serves as the authority for Excel worksheet ranges. c<||dz}d|zS)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 r:r^s rr;z XyWorkbookWriter.series_name_refs' **622Q6##rc.|jdz}|j}||zS)zq Return the number of rows preceding the data table for *series* in the Excel worksheet. )rMdata_point_offset)r r:title_and_spacer_rowsdata_point_rowss rrhz(XyWorkbookWriter.series_table_row_offsets$ !' q 0 2$66rcj||dz}|t|zdz }d||fzS)zx The Excel worksheet reference to the X values for this chart (not including the column label). rjr-zSheet1!$A$%d:$A$%drhr@r r:top_rowr5s r x_values_refzXyWorkbookWriter.x_values_refA ..v66:s6{{*Q. #w &;;;rcj||dz}|t|zdz }d||fzS)zx The Excel worksheet reference to the Y values for this chart (not including the column label). rjr-zSheet1!$B$%d:$B$%drorps r y_values_refzXyWorkbookWriter.y_values_refrsrcv|d|jji}|jD]}|d|ji}||}||dzd|j|||d|j||dzd|j|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. rOr-rN) rPr rQrhr`x_valuesr[r]y_valuesr rrchart_num_formatr:series_num_formatoffsets rrz$XyWorkbookWriter._populate_worksheets$.. d>N>\/]^^& V VF ( 3 3\6CW4X Y Y 11&99F  " "6A:q&/CS T T T OOFAv{ 3 3 3  " "6A:q&/CT U U U U V VrN) r#r$r%r&r;rhrrrurr>rrrfrfsn $$$777<<<<<<VVVVVrrfceZdZdZdZdZdS)BubbleWorkbookWriterz^ Service object that knows how to write an Excel workbook from bubble chart data. cj||dz}|t|zdz }d||fzS)z The Excel worksheet reference to the range containing the bubble sizes for *series* (not including the column heading cell). rjr-zSheet1!$C$%d:$C$%drorps rbubble_sizes_refz%BubbleWorkbookWriter.bubble_sizes_refrsrc|d|jji}|jD]}|d|ji}||}||dzd|j|||d|j||dzd|j|||dd||dzd|j |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. rOr-rrjSizeN) rPr rQrhr`rwr[r]rx bubble_sizesrys rrz(BubbleWorkbookWriter._populate_worksheets$.. d>N>\/]^^& Y YF ( 3 3\6CW4X Y Y 11&99F  " "6A:q&/CS T T T OOFAv{ 3 3 3  " "6A:q&/CT U U U OOFAv . . .  " "6A:q&2EGW X X X X Y YrN)r#r$r%r&rrr>rrr~r~sA <<<YYYYYrr~) r& __future__rr contextlibr xlsxwriterrobjectrr*rfr~r>rrrs,(("""""" %%%%%%!G!G!G!G!G&!G!G!GH@M@M@M@M@M0@M@M@MF9V9V9V9V9V*9V9V9Vx"Y"Y"Y"Y"Y+"Y"Y"Y"Y"Yr