o gimZ@sdZddlZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l m Z Gd d d e ZGd d d e ZdS)zs This class represents a common base for all LayoutElement implementations that attempt to represent tabular data. N)Decimal)Color)HexColor) Rectangle) Alignment) LayoutElement)Pagec)seZdZdZddeddededededddeddededededdddfdedeje d e d e d e d ed ededede de dede dededededejedejede f(fdd Z de de fdd Zd!d"de ddfd#d$Zd%e ddfd&d'Zde fd(d)Zde ffd*d+ Zdefd,d-Zdejefd.d/Zdejefd0d1Zdejefd2d3Zdejefd4d5Zdejefd6d7Zdejefd8d9Zdeje fd:d;Zde fdd?ZZS)@ TableCellz8 This class represents a single cell of a table NT000000rlayout_elementbackground_color border_bottom border_color border_leftborder_radius_bottom_leftborder_radius_bottom_rightborder_radius_top_leftborder_radius_top_right border_right border_top border_width column_spanpadding_bottom padding_left padding_right padding_toppreferred_heightpreferred_widthrow_spancstt|jdid|d|d|d|d|d|d|d| d | d | d | d d dtddtddtjdtddtddtddtdd|d|d|d|dtj||_|dkshJ| dksnJ||_ | |_ g|_ d|_ d|_ ||_d|_d|_||_d|_dS)Nr rrrrrrrrrrfont Helvetica font_colorz#000000 font_size horizontal_alignment margin_bottomr margin_left margin_right margin_toprrrrvertical_alignmentr )superr __init__rrr JUSTIFIEDTOP_layout_element _row_span _column_span_table_coordinates _min_width _max_width_preferred_width _min_height _max_height_preferred_height_forced_layout_box)selfr r rrrrrrrrrrrrrrrrrr __class__r+V/home/ubuntu/.local/lib/python3.10/site-packages/borb/pdf/canvas/layout/table/table.pyr-s|        zTableCell.__init__available_spacereturncCs|jdurttd}|jr||j7}|jr||j7}td}|jr%||j7}|jr-||j7}t|j|j |jr;|jntd|j |j |jrL|jntdt td|j |j |j|t td|j|j|j |S|j|S)Nr)r:r _border_left _border_width _border_right _border_top_border_bottomrget_x _padding_leftget_y_padding_bottommax get_width_padding_right get_height _padding_topr0get_layout_box)r;r?horizontal_border_widthvertical_border_widthr+r+r>_get_content_boxisX      zTableCell._get_content_boxpagercCs|j||dSN)r0paint)r;rSr?r+r+r>_paint_content_boxszTableCell._paint_content_box layout_boxcCs ||_|SrT)r:)r;rWr+r+r>_set_layout_boxszTableCell._set_layout_boxcC|jS)z This function returns the column span. This specifies the number of columns a cell should span. :return: The column span )r2r;r+r+r>get_column_spanzTableCell.get_column_spancs |jdur|jStt||S)z This function returns the previous result of layout :return: the Rectangle that was the result of the previous layout operation N)r:r,r rO)r;r?r<r+r>rOs zTableCell.get_layout_boxcCrY)z This function returns the LayoutElement in this TableCell :return: the LayoutElement inside this TableCell )r0rZr+r+r>get_layout_elementr\zTableCell.get_layout_elementcCrY)a This function returns the maximum width this TableCell (determined by the LayoutElement inside it). This function returns None if the maximum width has not yet been determined. :return: the maximum width of this TableCell )r5rZr+r+r> get_max_widthzTableCell.get_max_widthcCrY)a This function returns the maximum height this TableCell (determined by the LayoutElement inside it). This function returns None if the maximum height has not yet been determined. :return: the maximum height of this TableCell )r8rZr+r+r>get_max_heightr_zTableCell.get_max_heightcCrY)a This function returns the minimum width this TableCell (determined by the LayoutElement inside it). This function returns None if the minimum width has not yet been determined. :return: the minimum width of this TableCell )r4rZr+r+r> get_min_widthr_zTableCell.get_min_widthcCrY)a This function returns the minimum height this TableCell (determined by the LayoutElement inside it). This function returns None if the minimum height has not yet been determined. :return: the minimum height of this TableCell )r7rZr+r+r>get_min_heightr_zTableCell.get_min_heightcCrY)a  This function returns the preferred height of this TableCell (determined by a previous run of the LayoutElement framework). This function returns None if the preferred height has not yet been determined. :return: the preferred height of this TableCell )r9rZr+r+r>get_preferred_heightr_zTableCell.get_preferred_heightcCrY)a This function returns the preferred width of this TableCell (determined by a previous run of the LayoutElement framework). This function returns None if the preferred width has not yet been determined. :return: the preferred width of this TableCell )r6rZr+r+r>get_preferred_widthr_zTableCell.get_preferred_widthcCs |jS)z This function returns the previous result of layout of this LayoutElement :return: the Rectangle that was the result of the previous layout operation )r0get_previous_layout_boxrZr+r+r>res z!TableCell.get_previous_layout_boxcCrY)z This function returns the row span. This specifies the number of rows a cell should span. :return: The row span )r1rZr+r+r> get_row_spanr\zTableCell.get_row_spancCrY)a This function returns the coordinates (in the Table) of this TableCell. These coordinates may be a single typing.Tuple[int, int] or a typing.List of such tuples in case of row/column span. :return: the table coordinates )r3rZr+r+r>get_table_coordinatesszTableCell.get_table_coordinates) __name__ __module__ __qualname____doc__rrrtypingOptionalrboolintr-rrRrVrXr[rOr]r^r`rarbrcrdrerfListTuplerg __classcell__r+r+r<r>r s      M( &r c/s4eZdZdZddeddededededddedejededededededededejfde de d e j e d e d e d e d edededede de dededededededededededef.fdd Zd e d!e d"e j efd#d$Zd%ed"dfd&d'Z dId(e d)e d*e j e d"dfd+d,Zd!e d"e jefd-d.Zd e d"e jefd/d0Zd"e fd1d2Zd"e fd3d4ZdJd5d6ZdJd7d8ZdJd9d:Zed;fdZd e d"dfd?d@Zd e d"dfdAdBZded"dfdCdDZde de d e d e d"df dEdFZ dedededed"df dGdHZ!Z"S)KTablez This class represents a common base for all LayoutElement implementations that attempt to represent tabular data. NFr rr number_of_rowsnumber_of_columnsr rrrrrrrrrrr%r&r'r(r)rrrrr*cstt|jdid|d|d|d|d|d|d| d| d | d | d | d td d|d|d|d|d|d|d|d|d|d||dksSJ|dksYJ||_||_g|_dS)Nr rrrrrrrrrrr#r$r%r&r'r(r)rrrrr*r r+)r,rsr-r_number_of_rows_number_of_columns_content)r;rtrur rrrrrrrrrrr%r&r'r(r)rrrrr*r<r+r>r-sd        zTable.__init__rowcolumnr@cs8|jD]}tfdd|Ddkr|SqdS)Ncs(g|]}|dkr|dkr|qSrr r+.0przryr+r> Es z'Table._get_cells_at..r)rxlenrg)r;ryrztr+rr> _get_cells_atAs    zTable._get_cells_atr cst|ts t|}t|tsJj||}jdur$|_tfddtdj Dg D]} fdd| Dq:tfddtdj D}td|D]}td|D]}|j|||fqjqaS)z This function adds the given LayoutElement to this Table. This function returns self. :param layout_element: the LayoutElement to be added :return: self Ncs.g|]}tdd|Djkr|qS)cSsg|]}|qSr+)r[)r}yr+r+r>rnsz(Table.add...)sumget_cells_at_rowrwr}xrZr+r>rkszTable.add..rcs g|] }|dkr|dqSr{r+r)first_incomplete_rowr+r>rvs  csg|]}|vr|qSr+r+r)occupied_cols_in_rowr+r>r~s ) isinstancer rxappendr] _font_size get_font_sizeminrangervrextendrgrwrfr[r3)r;r inner_layout_elementcfirst_empty_columnijr+)rrr;r>addTs>           z Table.addeven_row_color odd_row_colorheader_row_colorcCsh|dur|}|dus Jtd|jD]}||D]}|dkr#||_q|ddkr-||_q||_qq|S)a This function colors the Table with the classic "zebra stripes" e.a. one color for all even rows, and a contrasting color for the odd rows. This function returns self. :param even_row_color: the Color to be used for even rows :param odd_row_color: the Color to be used for odd rows :param header_row_color: the Color to be used for the header row, if None is specified the even_row_color will be used :return: self Nr)rrvr_background_color)r;rrrrtcr+r+r>even_odd_row_colorss  zTable.even_odd_row_colorsc<g}|jD]}tfdd|Ddkr||q|S)z This function returns all TableCell elements at a given column :param column: the specified column :return: all TableCell elements at the given column cg|] }|dkr|qS)r r+r|rzr+r>rz-Table.get_cells_at_column..rrxrrgr)r;rzoutrr+rr>get_cells_at_column   zTable.get_cells_at_columncr)z This function returns all TableCell elements at a given row :param row: the specified row :return: all TableCell elements at the given row cr)rr+r|ryr+r>rrz*Table.get_cells_at_row..rr)r;ryrrr+rr>rrzTable.get_cells_at_rowcCrY)zu This function returns the number of columns in this Table :return: the number of columns )rwrZr+r+r>get_number_of_columnsr\zTable.get_number_of_columnscCrY)zo This function returns the number of rows in this Table :return: the number of rows )rvrZr+r+r>get_number_of_rowsr\zTable.get_number_of_rowscCs|jD]}d|_d|_d|_d|_q|dD]}d|_q||jdD]}d|_q%|dD]}d|_q0||jdD]}d|_q>|S)z This method sets the border(s) on all TableCell objects in this Table except for the borders that form the outside edge of the Table :return: self TrFr ) rxrDrCrErArrvrrw)r;rrr+r+r>internal_borderss zTable.internal_borderscCs|dddd|S)zr This method unsets the border(s) on all TableCell objects in this Table :return: self F)set_borders_on_all_cellsrZr+r+r> no_bordersszTable.no_borderscCsp||dD]}d|_q ||jdD]}d|_q|dD]}d|_q"||jdD]}d|_q0|S)z This method unsets the border(s) on all TableCell objects in this Table except for the borders that form the outside edge of the Table :return: self rTr ) rrrDrvrErrArwrC)r;rr+r+r> outer_bordersszTable.outer_borders border_radiuscCs|dd}|dur ||_|d|jd}|dur||_||jd|jd}|dur0||_||jdd}|dur@||_|S)z This function sets rounded borders on the outside of this Table :param border_radius: the desired border radius :return: self rNr )r_border_radius_top_leftrw_border_radius_top_rightrv_border_radius_bottom_right_border_radius_bottom_left)r;rulurlrllr+r+r>outer_borders_roundeds"  zTable.outer_borders_roundedcC|jD]}||_q|S)z This method sets the background Color on all TableCell objects in this Table :param background_color: the background Color :return: self )rxr)r;r er+r+r>!set_background_color_on_all_cells z'Table.set_background_color_on_all_cellscCr)z This method sets the border color on all TableCell objects in this Table :param border_color: the border Color :return: self )rx _border_color)r;rrr+r+r>set_border_color_on_all_cells'rz#Table.set_border_color_on_all_cellscCs"|dksJ|jD]}||_q |S)z This method sets the border width on all TableCell objects in this Table :param border_width: the border width :return: self r)rxrB)r;rrr+r+r>set_border_width_on_all_cells1s  z#Table.set_border_width_on_all_cellscC(|jD]}||_||_||_||_q|S)a This method sets the border(s) on all TableCell objects in this Table :param border_top: True if a top border should be drawn, False otherwise :param border_right: True if a right border should be drawn, False otherwise :param border_bottom: True if a bottom border should be drawn, False otherwise :param border_left: True if a left border should be drawn, False otherwise :return: self )rxrDrCrErA)r;rrrrrr+r+r>r< zTable.set_borders_on_all_cellscCr)a@ This method sets the padding on all TableCell objects in this Table :param padding_top: the top padding :param padding_right: the right padding :param padding_bottom: the bottom padding :param padding_left: the left padding :return: self )rxrNrLrIrG)r;rrrrrr+r+r>set_padding_on_all_cellsRrzTable.set_padding_on_all_cellsrT)r@rs)#rhrirjrkrrrLEFTr/rorlrmrrnr-r rrrrrprrrrrrrrrrrrrrrr+r+r<r>rss      <?          rs)rkrldecimalrborb.pdf.canvas.color.colorrr"borb.pdf.canvas.geometry.rectangler%borb.pdf.canvas.layout.layout_elementrrborb.pdf.page.pagerr rsr+r+r+r>s       i