3 import wx, wx.grid # noqa: E401 multiple imports on one line
4 from ..compat import string_type, unicode_type
5 from ..translations import translations
6 from .Grids import GridWindow, GridPanel
9 _ = getattr(translations, 'ugettext', None) or translations.gettext
12 class BooksDataTable(wx.grid.GridTableBase):
13 def __init__(self, rows_count, column_names):
14 wx.grid.GridTableBase.__init__(self)
15 self.rows_count = rows_count
16 self.column_names = column_names
18 for row in range(rows_count):
20 self.data.append(row_data)
21 for col in range(len(column_names)):
24 # required methods for the wxPyGridTableBase interface
26 def GetNumberRows(self):
27 return self.rows_count
29 def GetNumberCols(self):
30 return len(self.column_names)
32 def IsEmptyCell(self, row, col):
35 # Get/Set values in the table. The Python version of these
36 # methods can handle any data-type, (as long as the Editor and
37 # Renderer understands the type too,) not just strings as in the
39 def GetValue(self, row, col):
40 return self.data[row][col]
42 def SetValue(self, row, col, value):
43 self.data[row][col] = value
47 # Called when the grid needs to display labels
48 def GetRowLabelValue(self, row):
51 def GetColLabelValue(self, col):
52 return _(self.column_names[col])
54 # Called to determine the kind of editor/renderer to use by
55 # default, doesn't necessarily have to be the same type used
56 # natively by the editor/renderer if they know how to convert.
57 def GetTypeName(self, row, col):
58 return wx.grid.GRID_VALUE_STRING
61 class ListBooksPanel(GridPanel):
64 books_by_author = self.param['books_by_author']
65 columns = self.param['columns']
67 for author in books_by_author:
68 books = books_by_author[author]
69 series = {book.series for book in books}
70 total_rows += len(books) + len(series) + 1
72 grid.SetTable(BooksDataTable(total_rows, columns), takeOwnership=True)
73 grid.EnableEditing(False)
74 for col, col_name in enumerate(columns):
75 grid.AutoSizeColLabelSize(col)
76 if col_name in ('ser_no', 'size'):
77 cell_attr = wx.grid.GridCellAttr()
78 cell_attr.SetAlignment(wx.ALIGN_RIGHT, wx. ALIGN_CENTRE)
79 grid.SetColAttr(col, cell_attr)
81 for author in sorted(books_by_author):
82 grid.SetCellAlignment(row, 0, wx.ALIGN_LEFT, wx. ALIGN_CENTRE)
83 grid.SetCellSize(row, 0, 1, len(columns))
84 grid.SetCellValue(row, 0, u'%s' % (author,))
86 books = books_by_author[author]
89 if book.series != series:
94 grid.SetCellAlignment(row, 0,
95 wx.ALIGN_LEFT, wx. ALIGN_CENTRE)
96 grid.SetCellSize(row, 0, 1, len(columns))
97 grid.SetCellValue(row, 0,
98 u'%s — %s' % (book.author1, value))
101 for col, col_name in enumerate(columns):
102 value = getattr(book, col_name)
105 elif not isinstance(value, (string_type, unicode_type)):
107 grid.SetCellValue(row, col, value)
109 grid.AutoSizeColumns()
112 def OnDClick(self, event):
115 def OnKeyDown(self, event):
116 if event.GetKeyCode() == wx.WXK_ESCAPE:
122 class ListBooksWindow(GridWindow):
124 session_config_section_name = 'list_books'
125 window_title = u"m_Librarian: Список книг"
126 GridPanelClass = ListBooksPanel