Regroup filters

1 files changed, 4 insertions, 9 deletions

diff --git a/src/filters.py b/src/filters.py
index 07c0e0e..bac725c 100644
--- a/src/filters.py
+++ b/src/filters.py
@@ -6,13 +6,11 @@ import src.models as models
 
 class Filters(Gtk.Box):
 
-    def __init__(self, init_books, init_progress, init_genre, init_search, msg):
+    def __init__(self, init_books, init_filters, msg):
         Gtk.Box.__init__(self, spacing=10)
 
         genres = models.get_genres(init_books)
-        self._progress = init_progress
-        self._genre = init_genre
-        self._search = init_search
+        [self._progress, self._genre, self._search] = init_filters
         self._msg = msg
 
         self._progress_dropdown = Gtk.DropDown.new_from_strings(models.all_progress)
@@ -27,8 +25,8 @@ class Filters(Gtk.Box):
         self._search.connect('changed', self._on_update_search)
         self.append(self._search)
 
-    def get_progress(self):
-        return self._progress
+    def get(self):
+        return [self._progress, self._genre, self._search.get_text()]
 
     def get_genre(self):
         return self._genre
@@ -36,9 +34,6 @@ class Filters(Gtk.Box):
     def set_genre(self, genre):
         self._genre = genre
 
-    def get_search(self):
-        return self._search.get_text()
-
     def select_progress(self, progress):
         self._progress = progress
         self._progress_dropdown.set_selected(models.all_progress.index(progress))