Refactor(wx): Refactor common code

[m_librarian.git] / m_librarian / wx / ListAuthors.py

diff --git a/m_librarian/wx/ListAuthors.py b/m_librarian/wx/ListAuthors.py
index 89c294529da74668e26af471e272c5f96f9a962f..3d73a2827c7d3949707afe07298b43ffe930b15e 100644 (file)
--- a/m_librarian/wx/ListAuthors.py
+++ b/m_librarian/wx/ListAuthors.py
@@ -2,40 +2,19 @@
 
 import wx, wx.grid  # noqa: E401 multiple imports on one line
 from ..compat import string_type, unicode_type
+from ..search import books_by_author
 from ..translations import translations
-from .AWindow import AWindow
+from .Grids import GridWindow, GridPanel
+from .ListBooks import ListBooksWindow
 
 
-class ListAuthorsWindow(AWindow):
+class ListAuthorsPanel(GridPanel):
 
-    session_config_section_name = 'list_authors'
-    window_title = u"m_Librarian: Список авторов"
-
-    def __init__(self, parent, search_authors_results):
-        self.search_authors_results = search_authors_results
-        AWindow.__init__(self, parent)
-
-    def OnInit(self):
-        AWindow.OnInit(self)
-        ListAuthorsPanel(self, self.search_authors_results)
-
-
-class ListAuthorsPanel(wx.Panel):
-
-    def __init__(self, parent, search_authors_results):
-        wx.Panel.__init__(self, parent)
-        list_authors_sizer = wx.BoxSizer(wx.VERTICAL)
-        self.SetSizer(list_authors_sizer)
-
-        grid = wx.grid.Grid(self)
-        list_authors_sizer.Add(grid, 0, wx.EXPAND, 0)
-
-        self.InitGrid(grid, search_authors_results)
-
-    def InitGrid(self, grid, search_authors_results):
+    def InitGrid(self):
         _ = getattr(translations, 'ugettext', None) or translations.gettext
-        authors = search_authors_results['authors']
-        columns = search_authors_results['columns']
+        authors = self.param['authors']
+        columns = self.param['columns']
+        grid = self.grid
         grid.CreateGrid(len(authors), len(columns))
         grid.EnableEditing(False)
         for row in range(len(authors)):
@@ -56,3 +35,30 @@ class ListAuthorsPanel(wx.Panel):
                 grid.SetCellValue(row, col, value)
         grid.AutoSizeColumns()
         grid.AutoSizeRows()
+
+        grid.Bind(wx.grid.EVT_GRID_CELL_LEFT_DCLICK, self.OnDClick)
+        grid.Bind(wx.EVT_KEY_DOWN, self.OnKeyDown)
+
+    def listBooks(self, row):
+        authors = self.param['authors']
+        author = authors[row]
+        _books_by_author = books_by_author(author.id)
+        ListBooksWindow(self, _books_by_author)
+
+    def OnDClick(self, event):
+        row = event.GetRow()
+        self.listBooks(row)
+
+    def OnKeyDown(self, event):
+        if event.GetKeyCode() in (wx.WXK_RETURN, wx.WXK_NUMPAD_ENTER):
+            row = self.grid.GetGridCursorRow()
+            self.listBooks(row)
+        else:
+            event.Skip()
+
+
+class ListAuthorsWindow(GridWindow):
+
+    session_config_section_name = 'list_authors'
+    window_title = u"m_Librarian: Список авторов"
+    GridPanelClass = ListAuthorsPanel