A spot of refactoring

This commit is contained in:
Kovid Goyal 2017-05-03 08:32:31 +05:30
parent c88c951699
commit 358a3bf541
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C

View File

@ -284,9 +284,7 @@ class AddAction(InterfaceAction):
fmts = [pt.name] fmts = [pt.name]
ids.append(db.import_book(mi, fmts)) ids.append(db.import_book(mi, fmts))
tuple(map(os.remove, orig_fmts)) tuple(map(os.remove, orig_fmts))
self.gui.library_view.model().books_added(num) self.refresh_gui(num)
self.gui.refresh_cover_browser()
self.gui.tags_view.recount()
if ids: if ids:
ids.reverse() ids.reverse()
self.gui.library_view.select_rows(ids) self.gui.library_view.select_rows(ids)
@ -438,6 +436,13 @@ class AddAction(InterfaceAction):
Adder(paths, db=None if to_device else self.gui.current_db, Adder(paths, db=None if to_device else self.gui.current_db,
parent=self.gui, callback=partial(self._files_added, on_card=on_card), pool=self.gui.spare_pool()) parent=self.gui, callback=partial(self._files_added, on_card=on_card), pool=self.gui.spare_pool())
def refresh_gui(self, num, set_current_row=-1):
self.gui.library_view.model().books_added(num)
if set_current_row > -1:
self.gui.library_view.set_current_row(0)
self.gui.refresh_cover_browser()
self.gui.tags_view.recount()
def _files_added(self, adder, on_card=None): def _files_added(self, adder, on_card=None):
if adder.items: if adder.items:
paths, infos, names = [], [], [] paths, infos, names = [], [], []
@ -451,10 +456,7 @@ class AddAction(InterfaceAction):
return return
if adder.number_of_books_added > 0: if adder.number_of_books_added > 0:
self.gui.library_view.model().books_added(adder.number_of_books_added) self.refresh_gui(adder.number_of_books_added, set_current_row=0)
self.gui.library_view.set_current_row(0)
self.gui.refresh_cover_browser()
self.gui.tags_view.recount()
if adder.merged_books: if adder.merged_books:
merged = defaultdict(list) merged = defaultdict(list)