from wxPython.wx import *
from wxPython.lib.printout import PrintTable

import os

#---------------------------------------------------------------------------

buttonDefs = {
    814 : ('PreviewWide',      'Preview print of a wide table'),
    815 : ('PreviewNarrow',   'Preview print of a narrow table with color highlights'),
    818 : ('OnPreviewMatrix',   'Preview print of a narrow column grid without a table header'),
    817 : ('PreviewLine',   'Preview print to demonstrate the use of line breaks'),
    819 : ('PrintWide', 'Direct print (no preview) of a wide table'),
    }


class TablePanel(wxPanel):
    def __init__(self, parent, log, frame):
        wxPanel.__init__(self, parent, -1)
        self.log = log
        self.frame = frame

        box = wxBoxSizer(wxVERTICAL)
        box.Add(20, 30)
        keys = buttonDefs.keys()
        keys.sort()
        for k in keys:
            text = buttonDefs[k][1]
            btn = wxButton(self, k, text)
            box.Add(btn, 0, wxALIGN_CENTER|wxALL, 15)
            EVT_BUTTON(self, k, self.OnButton)

        self.SetAutoLayout(true)
        self.SetSizer(box)

    def OnButton(self, evt):
        funct = buttonDefs[evt.GetId()][0]
        code = 'self.' + funct + '()'
        eval(code)

    def ReadData(self):
        test_file = "./data/TestTable.txt"
        file = open(test_file,'r',1)
        i = 0

        data = []
        while 1:
            text = file.readline()
            text = string.strip(text)
            if not text:
                break

            list_val = string.splitfields(text,'\t')
            data.append(list_val)
        file.close()

        self.header = data[0]
        self.data = data[1:]

    def PreviewWide(self):
        self.ReadData()
        prt = PrintTable(self.frame)
        prt.data = self.data
        prt.left_margin = 0.5
        prt.set_column = [1.0, 1.0, 1.0, 1.5, 1.0, 3.0]
        prt.label = self.header
        prt.SetLandscape()

        prt.SetColumnLineSize(2, 3)
        prt.SetColumnLineColour(3, wxNamedColour('RED'))

        prt.SetRowLineSize(1, 3)
        prt.SetRowLineColour(5, wxNamedColour('RED'))

        prt.SetHeader("wxWindows Applications")
        prt.SetFooter()
        prt.Preview()

    def PreviewNarrow(self):
        self.ReadData()
        new_data = []
        for val in self.data:
            new_data.append([val[0], val[1], val[2], val[4], val[5]])

        val = self.header
        new_header = [val[0], val[1], val[2], val[4], val[5]]

        prt = PrintTable(self.frame)
        prt.data = new_data
        prt.set_column = [ 1, 1, 1, 1, 2]
        prt.label = new_header
        prt.SetColAlignment(1, wxALIGN_CENTRE)
        prt.SetColBackgroundColour(0, wxNamedColour('RED'))
        prt.SetColTextColour(0, wxNamedColour('WHITE'))
        prt.SetCellColour(4, 0, wxNamedColour('LIGHT BLUE'))
        prt.SetCellColour(4, 1, wxNamedColour('LIGHT BLUE'))
        prt.SetCellColour(17, 1, wxNamedColour('LIGHT BLUE'))

        prt.SetColBackgroundColour(2, wxNamedColour('LIGHT BLUE'))
        prt.SetCellText(4, 2, wxNamedColour('RED'))

        prt.SetColTextColour(3, wxNamedColour('RED'))
        prt.label_font_colour = wxNamedColour('WHITE')
        prt.SetHeader("wxWindows Applications", colour = wxNamedColour('RED'))

        prt.SetHeader("Date", align=wxALIGN_RIGHT, indent = -2, colour = wxNamedColour('BLUE'))
        prt.SetFooter("Page No", colour = wxNamedColour('RED'), type ="Num")
        prt.Preview()

    def OnPreviewMatrix(self):
        total_col = 45
        total_row = 10
        hsize = 0.2
        vsize = 0.2

        data = []
        startx = 1.0
        columns = []
        for val in range(total_col):
            columns.append(hsize)

        prt = PrintTable(self.frame)

        for row in range(total_row):
            value = []
            for col in range(total_col):
                value.append(str(col))
            data.append(value)

        for col in range(total_col):
            prt.SetColAlignment(col, wxALIGN_CENTRE)

        prt.SetLandscape()
        prt.text_font_size = 8
        prt.cell_left_margin = 0

        prt.data = data
        prt.set_column = columns
        prt.SetHeader("Test of Small Grid Size")
        prt.Preview()

    def PreviewLine(self):
        prt = PrintTable(self.frame)
        prt.label = ["Header 1", "Header 2", "Header 3"]
        prt.set_column = []
        prt.data = [["Row 1", "1", "2"], ["Row 2", "3", "4\nNew Line to see if it also can wrap around the cell region properly\nAnother new line"]]
        prt.SetFooter()
        prt.Preview()

    def PrintWide(self):
        self.ReadData()
        prt = PrintTable(self.frame)
        prt.data = self.data

        prt.left_margin = 0.5
        prt.set_columns = [ 1, 1, 1, 1, 2, 1, 3 ]
        prt.label = self.header
        prt.SetLandscape()
        prt.Print()


#---------------------------------------------------------------------------

def runTest(frame, nb, log):
    win = TablePanel(nb, log, frame)
    return win

#---------------------------------------------------------------------------





import os
import wxPython.lib.printout




overview = """\
<html><body>
<h2>Table Printing</h2>

This demo shows various ways of using the <b><i>new
</i></b> PrintOut class.  To understand the class you need to examine the demo examples
and the library <a href="%s">printout.py</a> module classes.
<p>
The initial class primarily contains a Table preview/printing class.  There is alot of flexibility
in manipulating the placement, sizing, colours, alignment of the table text and cell background colors.
There are also a number of options for printing Header and Footer information on the page.
<p>
There is also a class to extract the parameters from a wxGrid and easily recreate a Table printout.
<p>
The data is printed from a list object containing the column and row values.  The label or table header
can be defined and will be repeated for all pages.
<p>
The correct "Total Page" does get calculated and used in the print out Footer.
<p>
There is still problems with the print framework to properly get the total pages in the preview unless
the program knows it before trying to parse through the available pages.  This will be fixed
when the framework allows for it.


""" % os.path.join(os.path.dirname(wxPython.lib.printout.__file__), "printout.py")