customized TableBordersLayout

[horstweigelt65]

The customized TableBordersLayout example in
https://py-pdf.github.io/fpdf2/Tables.html#set-cell-borders does not work with python 3.13.0 and fpdf 2.8.7.
-> ImportError: cannot import name 'TableBorderStyle' from 'fpdf.table'
Is the example still 'up to date'? What does the fix look like?

[andersonhc]

It should look like:

from fpdf.enums import TableBordersLayout, TableBorderStyle, TableCellStyle

[horstweigelt65]

Thank you. Yes, that import works nicely now.
But now I receive the error:
TypeError: TableBordersLayout() takes no arguments

Here is the code pasted from the website https://py-pdf.github.io/fpdf2/Tables.html#set-cell-borders

custom_layout = TableBordersLayout(
    cell_style_getter=lambda row_num, col_num, num_heading_rows, num_rows, num_cols: TableCellStyle(
        left=(
            True if col_num == 0
            else TableBorderStyle(color=(150, 150, 150), dash=2) if col_num == 2
            else False
        ),        bottom=True if row_num == num_rows - 1 else False,
        right=True if col_num == num_cols - 1 else False,
        top=(
            True if row_num == 0
            else TableBorderStyle(thickness=1) if row_num == num_heading_rows
            else TableBorderStyle(color=red, dash=2)
        ),
    )
)

What is the best fix??

[andersonhc]

I believe the best way to rewrite that example should be:

gray = (150, 150, 150)
red = (255, 0, 0)
class CustomLayout(TableBordersLayout):
    def cell_style_getter(
            self,
            row_idx,
            col_idx,
            col_pos,
            num_heading_rows,
            num_rows,
            num_col_idx,
            num_col_pos,
    ):
        return TableCellStyle(
            left=(
                True if col_idx == 0
                else TableBorderStyle(color=(150, 150, 150), dash=2) if col_idx == 2
                else False
            ),
            bottom=True if row_idx == num_rows - 1 else False,
            right=True if num_col_idx == num_col_idx - 1 else False,
            top=(
                True if row_idx == 0
                else TableBorderStyle(thickness=1) if row_idx == num_heading_rows
                else TableBorderStyle(color=red, dash=2)
            ),
        )

with pdf.table(borders_layout=CustomLayout()) as table:
...

I will open an issue to get the documentation on tables reviewed.

[horstweigelt65]

Thank you. Now the example works :-)
Great idea to get this documentation and the examples reviewed!