import wx

class MyPrintout(wx.Printout):
    def __init__(self):
        super().__init__()

    def OnPrintPage(self, page):
        dc = self.GetDC()
        dc.SetFont(wx.Font(10, wx.FONTFAMILY_SWISS, wx.FONTSTYLE_NORMAL, wx.FONTWEIGHT_NORMAL))
        dc.DrawText("Hello, this is a print test!", 100, 100)
        return True

def draw_table_border(dc, x_pos, y_pos, width, height):
    dc.SetPen(wx.Pen(wx.Colour(0, 0, 0)))  # 设置边框颜色
    dc.SetBrush(wx.Brush(wx.Colour(255, 255, 255)))  # 设置背景颜色
    dc.DrawRectangle(x_pos, y_pos, width, height)

    def OnPrint(self, event: wx.Event) -> None:
        """打印表格"""

        # 创建打印数据对象
        pageData = wx.PrintData()
        pageData.SetPaperId(wx.PAPER_A4)  # 设置纸张类型为 A4
        pageData.SetOrientation(wx.LANDSCAPE)  # 设置纸张方向为纵向

        # 将打印数据设置到打印机对象
        data = wx.PrintDialogData(pageData)
        # 窗口标题
        page_title = self.GetTitle()

        # 创建打印输出对象
        printout = ctrl.MyPrintout(self.grid, page_title)
        printout2 = ctrl.MyPrintout(self.grid, page_title)

        # 创建打印机对象，并将打印数据传递给打印机
        # printer = wx.Printer(data)
        preview = wx.PrintPreview(printout, printout2, data=data)
        if preview.IsOk():
            preview.SetZoom(100)
            frame = wx.PreviewFrame(
                preview, self, "打印预览", pos=self.GetPosition(), size=self.GetSize()
            )
            frame.Initialize()
            frame.Show(True)

class MyPrintout(wx.Printout):
    """创建一个自定义的 wx.Printout 类，在其中定义如何打印 wx.Grid 的内容。"""

    def __init__(
        self,
        grid: gridlib.Grid,
        title: str = "打印标题",
        margins=(wx.Point(10, 10), wx.Point(10, 10)),
    ):
        super().__init__()
        self.grid = grid
        self.title = title
        self.margins = margins

        # 获取网格的行列数
        self.row_count = self.grid.GetNumberRows()
        self.col_count = self.grid.GetNumberCols()
        self.current_row = 0  # 记录当前打印的行数

        font = wx.Font(
            FONTSIZE, wx.FONTFAMILY_TELETYPE, wx.FONTSTYLE_NORMAL, wx.FONTWEIGHT_NORMAL
        )
        dc.SetFont(font)
        # 计算每行的高度
        self.lineHeight = dc.GetCharHeight() + 5  # 加上5像素的间距

        # 标题高度 + 标题底部空白行高度 + 表头高度
        y_offset = 20 + 20 + 20
        # 计算每页可以显示的行数
        self.linesPerPage = int((self.pageHeight - y_offset) // self.lineHeight)

        # 计算页面数
        self.numPages = self.row_count // self.linesPerPage
        if self.row_count % self.linesPerPage != 0:
            self.numPages += 1

    def HasPage(self, page):
        return page <= self.numPages

    def GetPageInfo(self):
        return (1, self.numPages, 1, self.numPages)

        # 绘制标题，居中显示
        text_width, text_height = dc.GetTextExtent(self.title)
        title_x = (width - text_width) // 2  # 计算居中的X位置
        dc.DrawText(self.title, int(title_x), int(y_offset))

        # 设置线条颜色
        dc.SetPen(wx.Pen(wx.Colour(0, 0, 0)))  # 黑色线条
        # 将所有的坐标转换为整数
        dc.DrawLine(line_x, line_y, line_x + line_width, line_y)

        # 绘制表头
        font.SetPointSize(FONTSIZE)
        dc.SetFont(font)
        for col in range(self.col_count):
            dc.DrawText(
                self.grid.GetColLabelValue(col),
                int(x_offset + sum(col_widths[:col])),
                int(y_offset),
            )

col_widths = [self.grid.GetColSize(col) for col in range(self.col_count)]

self.current_row = (page - 1) * self.linesPerPage  # 计算当前页面的起始行数

        # 绘制表格的内容
        lines_on_page = 0  # 记录当前页面的行数
        for row in range(self.current_row, self.row_count):
            y_offset += self.lineHeight  # 增加行的高度
            for col in range(self.col_count):
                cell_value = self.grid.GetCellValue(row, col)
                dc.DrawText(
                    cell_value,
                    x_offset + sum(col_widths[:col]),
                    y_offset,
                )
            lines_on_page += 1
            # 如果当前页面的行数已经达到最大值，则绘制页脚，并返回False，继续打印下一页
            if lines_on_page >= self.linesPerPage:
                self.draw_footer(dc, page) # 绘制页脚，底部的页码
                return False  # 打印完一页后返回False，继续打印下一页

        self.draw_footer(dc, page)
        return True

    def draw_footer(self, dc: wx.DC, page: int):
        # 绘制当前页码
        page_num_text = f"当前页码: {page} / {self.numPages}"
        page_num_width, page_num_height = dc.GetTextExtent(page_num_text)
        dc.DrawText(
            page_num_text,
            int((self.pageWidth - page_num_width) // 2),
            int(self.y2),
        )