diff --git a/src/sioyek/sioyek.py b/src/sioyek/sioyek.py index c47e80a..e6576fd 100644 --- a/src/sioyek/sioyek.py +++ b/src/sioyek/sioyek.py @@ -1154,8 +1154,12 @@ def get_bookmarks(self): BOOKMARK_SELECT_QUERY = "select * from bookmarks where document_path='{}'".format(doc_hash) shared_database = self.sioyek.get_shared_database() cursor = shared_database.execute(BOOKMARK_SELECT_QUERY) - bookmarks = [Bookmark(self, desc, y_offset) for _, _, desc, y_offset in cursor.fetchall()] - return bookmarks + return [ + Bookmark(self, desc, y_offset) + for row in cursor.fetchall() + if len(row) >= 4 + for _, _, desc, y_offset, *_ in [row] + ] def get_highlights(self): doc_hash = self.get_hash() @@ -1163,8 +1167,12 @@ def get_highlights(self): shared_database = self.sioyek.get_shared_database() cursor = shared_database.execute(HIGHLIGHT_SELECT_QUERY) - highlights = [Highlight(self, text, highlight_type, (begin_x, begin_y), (end_x, end_y)) for _, _, text, highlight_type, begin_x, begin_y, end_x, end_y in cursor.fetchall()] - return highlights + return [ + Highlight(self, text, highlight_type, (begin_x, begin_y), (end_x, end_y)) + for row in cursor.fetchall() + if len(row) >= 8 + for _, _, text, highlight_type, begin_x, begin_y, end_x, end_y, *_ in [row] + ] def get_page_selection(self, page_number, selection_begin_x, selection_begin_y, selection_end_x, selection_end_y): in_range = False