#!/usr/bin/env python3 import os import sys from PyQt5 import QtWidgets, QtGui, QtCore import sorter import database from resources import mainwindow from widgets import LibraryToolBar, BookToolBar, Tab, LibraryDelegate from threaded import BackGroundTabUpdate, BackGroundBookAddition from library import Library from settings import Settings from settingsdialog import SettingsUI class MainUI(QtWidgets.QMainWindow, mainwindow.Ui_MainWindow): def __init__(self): super(MainUI, self).__init__() self.setupUi(self) # Empty variables that will be infested soon self.current_view = None self.last_open_books = None self.last_open_tab = None self.last_open_path = None self.thread = None # Background Thread self.current_contentView = None # For fullscreening purposes self.display_profiles = None self.current_profile_index = None self.database_path = None self.table_header_sizes = None self.settings_dialog_settings = None # Initialize application Settings(self).read_settings() # This should populate all variables that need # to be remembered across sessions # Create the database in case it doesn't exist database.DatabaseInit(self.database_path) # Initialize settings dialog self.settings_dialog = SettingsUI(self) # Create and right align the statusbar label widget self.statusMessage = QtWidgets.QLabel() self.statusMessage.setObjectName('statusMessage') self.statusBar.addPermanentWidget(self.statusMessage) self.sorterProgress = QtWidgets.QProgressBar() self.sorterProgress.setObjectName('sorterProgress') sorter.progressbar = self.sorterProgress # This is so that updates can be # connected to setValue self.statusBar.addWidget(self.sorterProgress) self.sorterProgress.setVisible(False) # Application wide temporary directory self.temp_dir = QtCore.QTemporaryDir() # Init the Library self.lib_ref = Library(self) # Library toolbar self.libraryToolBar = LibraryToolBar(self) self.libraryToolBar.addButton.triggered.connect(self.add_books) self.libraryToolBar.deleteButton.triggered.connect(self.delete_books) self.libraryToolBar.coverViewButton.triggered.connect(self.switch_library_view) self.libraryToolBar.tableViewButton.triggered.connect(self.switch_library_view) self.libraryToolBar.settingsButton.triggered.connect(self.show_settings) self.libraryToolBar.searchBar.textChanged.connect(self.lib_ref.update_proxymodel) self.libraryToolBar.searchBar.textChanged.connect(self.lib_ref.update_table_proxy_model) self.libraryToolBar.sortingBox.activated.connect(self.lib_ref.update_proxymodel) if self.current_view == 0: self.libraryToolBar.coverViewButton.trigger() elif self.current_view == 1: self.libraryToolBar.tableViewButton.trigger() self.addToolBar(self.libraryToolBar) # Book toolbar self.bookToolBar = BookToolBar(self) self.bookToolBar.fullscreenButton.triggered.connect(self.set_fullscreen) for count, i in enumerate(self.display_profiles): self.bookToolBar.profileBox.setItemData(count, i, QtCore.Qt.UserRole) self.bookToolBar.profileBox.currentIndexChanged.connect(self.format_contentView) self.bookToolBar.profileBox.setCurrentIndex(self.current_profile_index) self.bookToolBar.fontBox.currentFontChanged.connect(self.modify_font) self.bookToolBar.fontSizeBox.currentTextChanged.connect(self.modify_font) self.bookToolBar.lineSpacingUp.triggered.connect(self.modify_font) self.bookToolBar.lineSpacingDown.triggered.connect(self.modify_font) self.bookToolBar.paddingUp.triggered.connect(self.modify_font) self.bookToolBar.paddingDown.triggered.connect(self.modify_font) self.bookToolBar.resetProfile.triggered.connect(self.reset_profile) self.bookToolBar.zoomIn.triggered.connect(self.modify_comic_view) self.bookToolBar.zoomOut.triggered.connect(self.modify_comic_view) self.bookToolBar.fitWidth.triggered.connect(self.modify_comic_view) self.bookToolBar.bestFit.triggered.connect(self.modify_comic_view) self.bookToolBar.originalSize.triggered.connect(self.modify_comic_view) self.bookToolBar.comicBGColor.clicked.connect(self.get_color) self.bookToolBar.colorBoxFG.clicked.connect(self.get_color) self.bookToolBar.colorBoxBG.clicked.connect(self.get_color) self.bookToolBar.tocBox.currentIndexChanged.connect(self.set_toc_position) self.addToolBar(self.bookToolBar) # Make the correct toolbar visible self.tab_switch() self.tabWidget.currentChanged.connect(self.tab_switch) # Tab closing self.tabWidget.setTabsClosable(True) # Get list of available parsers self.available_parsers = '*.' + ' *.'.join(sorter.available_parsers) print('Available parsers: ' + self.available_parsers) self.reloadLibrary = QtWidgets.QToolButton() self.reloadLibrary.setIcon(QtGui.QIcon.fromTheme('reload')) self.reloadLibrary.setAutoRaise(True) self.reloadLibrary.setPopupMode(QtWidgets.QToolButton.InstantPopup) self.reloadLibrary.triggered.connect(self.switch_library_view) self.tabWidget.tabBar().setTabButton( 0, QtWidgets.QTabBar.RightSide, self.reloadLibrary) self.tabWidget.tabCloseRequested.connect(self.tab_close) # Init display models self.lib_ref.generate_model('build') self.lib_ref.create_table_model() self.lib_ref.create_proxymodel() # ListView self.listView.setGridSize(QtCore.QSize(175, 240)) self.listView.setMouseTracking(True) self.listView.verticalScrollBar().setSingleStep(9) self.listView.doubleClicked.connect(self.library_doubleclick) self.listView.setItemDelegate(LibraryDelegate(self.temp_dir.path())) # TableView self.tableView.doubleClicked.connect(self.library_doubleclick) self.tableView.horizontalHeader().setSectionResizeMode( QtWidgets.QHeaderView.Interactive) self.tableView.horizontalHeader().setSortIndicator(0, QtCore.Qt.AscendingOrder) self.tableView.horizontalHeader().setHighlightSections(False) if self.table_header_sizes: for count, i in enumerate(self.table_header_sizes): self.tableView.horizontalHeader().resizeSection(count, int(i)) self.tableView.horizontalHeader().setStretchLastSection(True) self.tableView.horizontalHeader().setSectionResizeMode(3, QtWidgets.QHeaderView.Stretch) # Keyboard shortcuts self.ks_close_tab = QtWidgets.QShortcut(QtGui.QKeySequence('Ctrl+W'), self) self.ks_close_tab.setContext(QtCore.Qt.ApplicationShortcut) self.ks_close_tab.activated.connect(self.tab_close) self.ks_exit_all = QtWidgets.QShortcut(QtGui.QKeySequence('Ctrl+Q'), self) self.ks_exit_all.setContext(QtCore.Qt.ApplicationShortcut) self.ks_exit_all.activated.connect(self.closeEvent) self.listView.setFocus() # Open last... open books. # Then set the value to None for the next run self.open_files(self.last_open_books) self.last_open_books = None def resizeEvent(self, event=None): if event: # This implies a vertical resize event only # We ain't about that lifestyle if event.oldSize().width() == event.size().width(): return # The hackiness of this hack is just... default_size = 170 # This is size of the QIcon (160 by default) + # minimum margin is needed between thumbnails # for n icons, the n + 1th icon will appear at > n +1.11875 # First, calculate the number of images per row i = self.listView.viewport().width() / default_size rem = i - int(i) if rem >= .11875 and rem <= .9999: num_images = int(i) else: num_images = int(i) - 1 # The rest is illustrated using informative variable names space_occupied = num_images * default_size # 12 is the scrollbar width # Larger numbers keep reduce flickering but also increase # the distance from the scrollbar space_left = ( self.listView.viewport().width() - space_occupied - 19) try: layout_extra_space_per_image = space_left // num_images self.listView.setGridSize( QtCore.QSize(default_size + layout_extra_space_per_image, 250)) except ZeroDivisionError: # Initial resize is ignored return def add_books(self): # TODO # Maybe expand this to traverse directories recursively opened_files = QtWidgets.QFileDialog.getOpenFileNames( self, 'Open file', self.last_open_path, f'eBooks ({self.available_parsers})') if opened_files[0]: self.last_open_path = os.path.dirname(opened_files[0][0]) self.sorterProgress.setVisible(True) self.statusMessage.setText('Adding books...') self.thread = BackGroundBookAddition(self, opened_files[0], self.database_path) self.thread.finished.connect(self.move_on) self.thread.start() def move_on(self): self.sorterProgress.setVisible(False) self.lib_ref.create_table_model() self.lib_ref.create_proxymodel() def delete_books(self): # TODO # Use maptosource() here to get the view_model # indices selected in the listView # Implement this for the tableview selected_books = self.listView.selectedIndexes() if selected_books: def ifcontinue(box_button): if box_button.text() == '&Yes': selected_hashes = [] for i in selected_books: data = i.data(QtCore.Qt.UserRole + 3) selected_hashes.append(data['hash']) database.DatabaseFunctions( self.database_path).delete_from_database(selected_hashes) self.lib_ref.generate_model('build') self.lib_ref.create_table_model() self.lib_ref.create_proxymodel() selected_number = len(selected_books) msg_box = QtWidgets.QMessageBox() msg_box.setText('Delete %d book(s)?' % selected_number) msg_box.setIcon(QtWidgets.QMessageBox.Question) msg_box.setWindowTitle('Confirm deletion') msg_box.setStandardButtons( QtWidgets.QMessageBox.Yes | QtWidgets.QMessageBox.No) msg_box.buttonClicked.connect(ifcontinue) msg_box.show() msg_box.exec_() def switch_library_view(self): if self.libraryToolBar.coverViewButton.isChecked(): self.stackedWidget.setCurrentIndex(0) self.libraryToolBar.sortingBoxAction.setVisible(True) else: self.stackedWidget.setCurrentIndex(1) self.libraryToolBar.sortingBoxAction.setVisible(False) self.resizeEvent() def tab_switch(self): if self.tabWidget.currentIndex() == 0: self.resizeEvent() self.bookToolBar.hide() self.libraryToolBar.show() if self.lib_ref.proxy_model: # Making the proxy model available doesn't affect # memory utilization at all. Bleh. self.statusMessage.setText( str(self.lib_ref.proxy_model.rowCount()) + ' Books') else: self.bookToolBar.show() self.libraryToolBar.hide() current_metadata = self.tabWidget.widget( self.tabWidget.currentIndex()).metadata if self.bookToolBar.fontButton.isChecked(): self.bookToolBar.customize_view_on() current_title = current_metadata['title'] current_author = current_metadata['author'] current_position = current_metadata['position'] current_toc = current_metadata['content'].keys() self.bookToolBar.tocBox.blockSignals(True) self.bookToolBar.tocBox.clear() self.bookToolBar.tocBox.addItems(current_toc) if current_position: self.bookToolBar.tocBox.setCurrentIndex( current_position['current_chapter'] - 1) self.bookToolBar.tocBox.blockSignals(False) self.format_contentView() self.statusMessage.setText( current_author + ' - ' + current_title) def tab_close(self, tab_index=None): if not tab_index: tab_index = self.tabWidget.currentIndex() if tab_index == 0: return tab_metadata = self.tabWidget.widget(tab_index).metadata self.thread = BackGroundTabUpdate( self.database_path, [tab_metadata]) self.thread.start() self.tabWidget.removeTab(tab_index) def set_toc_position(self, event=None): current_tab = self.tabWidget.widget(self.tabWidget.currentIndex()) # We're updating the underlying models to have real-time # updates on the read status # Since there are 2 separate models, they will each have to # be updated individually # The listView model # Set a baseline model index in case the item gets deleted # E.g It's open in a tab and deleted from the library model_index = None start_index = self.lib_ref.view_model.index(0, 0) # Find index of the model item that corresponds to the tab matching_item = self.lib_ref.view_model.match( start_index, QtCore.Qt.UserRole + 6, current_tab.metadata['hash'], 1, QtCore.Qt.MatchExactly) if matching_item: model_row = matching_item[0].row() model_index = self.lib_ref.view_model.index(model_row, 0) current_tab.metadata[ 'position']['current_chapter'] = event + 1 if model_index: self.lib_ref.view_model.setData( model_index, current_tab.metadata['position'], QtCore.Qt.UserRole + 7) # The tableView model model_index = None start_index = self.lib_ref.table_model.index(0, 0) matching_item = self.lib_ref.table_model.match( start_index, QtCore.Qt.UserRole + 1, current_tab.metadata['hash'], 1, QtCore.Qt.MatchExactly) if matching_item: model_row = matching_item[0].row() self.lib_ref.table_model.display_data[model_row][5][ 'position'] = current_tab.metadata['position'] # Go on to change the value of the Table of Contents box current_tab.change_chapter_tocBox() def set_fullscreen(self): current_tab = self.tabWidget.currentIndex() current_tab_widget = self.tabWidget.widget(current_tab) current_tab_widget.go_fullscreen() def library_doubleclick(self, myindex): sender = self.sender().objectName() if sender == 'listView': index = self.lib_ref.proxy_model.index(myindex.row(), 0) metadata = self.lib_ref.proxy_model.data(index, QtCore.Qt.UserRole + 3) elif sender == 'tableView': index = self.lib_ref.table_proxy_model.index(myindex.row(), 0) metadata = self.lib_ref.table_proxy_model.data(index, QtCore.Qt.UserRole) # Shift focus to the tab that has the book open (if there is one) for i in range(1, self.tabWidget.count()): tab_metadata = self.tabWidget.widget(i).metadata if tab_metadata['hash'] == metadata['hash']: self.tabWidget.setCurrentIndex(i) return path = metadata['path'] self.open_files([path]) def open_files(self, file_paths): # file_paths is expected to be a list # This allows for threading file opening # Which should speed up multiple file opening # especially @ application start if not file_paths: return print('Attempting to open: ' + ', '.join(file_paths)) contents = sorter.BookSorter( file_paths, 'reading', self.database_path, self.temp_dir.path()).initiate_threads() found_a_focusable_tab = False for i in contents: file_data = contents[i] Tab(file_data, self.tabWidget) # New tabs are created here # Initial position adjustment # is carried out by the tab itself if file_data['path'] == self.last_open_tab: found_a_focusable_tab = True self.tabWidget.setCurrentIndex(self.tabWidget.count() - 1) if not found_a_focusable_tab: self.tabWidget.setCurrentIndex(self.tabWidget.count() - 1) self.format_contentView() def get_color(self): signal_sender = self.sender().objectName() profile_index = self.bookToolBar.profileBox.currentIndex() current_profile = self.bookToolBar.profileBox.itemData( profile_index, QtCore.Qt.UserRole) # Retain current values on opening a new dialog def open_color_dialog(current_color): color_dialog = QtWidgets.QColorDialog() new_color = color_dialog.getColor(current_color) if new_color.isValid(): # Returned in case cancel is pressed return new_color else: return current_color if signal_sender == 'fgColor': current_color = current_profile['foreground'] new_color = open_color_dialog(current_color) self.bookToolBar.colorBoxFG.setStyleSheet( 'background-color: %s' % new_color.name()) current_profile['foreground'] = new_color elif signal_sender == 'bgColor': current_color = current_profile['background'] new_color = open_color_dialog(current_color) self.bookToolBar.colorBoxBG.setStyleSheet( 'background-color: %s' % new_color.name()) current_profile['background'] = new_color elif signal_sender == 'comicBGColor': current_color = self.comic_profile['background'] new_color = open_color_dialog(current_color) self.bookToolBar.comicBGColor.setStyleSheet( 'background-color: %s' % new_color.name()) self.comic_profile['background'] = new_color self.bookToolBar.profileBox.setItemData( profile_index, current_profile, QtCore.Qt.UserRole) self.format_contentView() def modify_font(self): signal_sender = self.sender().objectName() profile_index = self.bookToolBar.profileBox.currentIndex() current_profile = self.bookToolBar.profileBox.itemData( profile_index, QtCore.Qt.UserRole) if signal_sender == 'fontBox': current_profile['font'] = self.bookToolBar.fontBox.currentFont().family() if signal_sender == 'fontSizeBox': old_size = current_profile['font_size'] new_size = self.bookToolBar.fontSizeBox.currentText() if new_size.isdigit(): current_profile['font_size'] = int(new_size) else: current_profile['font_size'] = old_size if signal_sender == 'lineSpacingUp': current_profile['line_spacing'] += .5 if signal_sender == 'lineSpacingDown': current_profile['line_spacing'] -= .5 if signal_sender == 'paddingUp': current_profile['padding'] += 5 if signal_sender == 'paddingDown': current_profile['padding'] -= 5 self.bookToolBar.profileBox.setItemData( profile_index, current_profile, QtCore.Qt.UserRole) self.format_contentView() def modify_comic_view(self): signal_sender = self.sender().objectName() current_tab = self.tabWidget.widget(self.tabWidget.currentIndex()) self.bookToolBar.fitWidth.setChecked(False) self.bookToolBar.bestFit.setChecked(False) self.bookToolBar.originalSize.setChecked(False) if signal_sender == 'zoomOut': self.comic_profile['zoom_mode'] = 'manualZoom' self.comic_profile['padding'] += 50 # This prevents infinite zoom out if self.comic_profile['padding'] * 2 > current_tab.contentView.viewport().width(): self.comic_profile['padding'] -= 50 if signal_sender == 'zoomIn': self.comic_profile['zoom_mode'] = 'manualZoom' self.comic_profile['padding'] -= 50 # This prevents infinite zoom in if self.comic_profile['padding'] < 0: self.comic_profile['padding'] = 0 if signal_sender == 'fitWidth': self.comic_profile['zoom_mode'] = 'fitWidth' self.comic_profile['padding'] = 0 self.bookToolBar.fitWidth.setChecked(True) # Padding in the following cases is decided by # the image pixmap loaded by the widget if signal_sender == 'bestFit': self.comic_profile['zoom_mode'] = 'bestFit' self.bookToolBar.bestFit.setChecked(True) if signal_sender == 'originalSize': self.comic_profile['zoom_mode'] = 'originalSize' self.bookToolBar.originalSize.setChecked(True) self.format_contentView() def format_contentView(self): # TODO # Implement line spacing # See what happens if a font isn't installed current_tab = self.tabWidget.widget(self.tabWidget.currentIndex()) try: current_metadata = current_tab.metadata except AttributeError: return if current_metadata['images_only']: background = self.comic_profile['background'] padding = self.comic_profile['padding'] zoom_mode = self.comic_profile['zoom_mode'] if zoom_mode == 'fitWidth': self.bookToolBar.fitWidth.setChecked(True) if zoom_mode == 'bestFit': self.bookToolBar.bestFit.setChecked(True) if zoom_mode == 'originalSize': self.bookToolBar.originalSize.setChecked(True) self.bookToolBar.comicBGColor.setStyleSheet( 'background-color: %s' % background.name()) current_tab.format_view( None, None, None, background, padding) else: profile_index = self.bookToolBar.profileBox.currentIndex() current_profile = self.bookToolBar.profileBox.itemData( profile_index, QtCore.Qt.UserRole) font = current_profile['font'] foreground = current_profile['foreground'] background = current_profile['background'] padding = current_profile['padding'] font_size = current_profile['font_size'] # Change toolbar widgets to match new settings self.bookToolBar.fontBox.blockSignals(True) self.bookToolBar.fontSizeBox.blockSignals(True) self.bookToolBar.fontBox.setCurrentText(font) self.bookToolBar.fontSizeBox.setCurrentText(str(font_size)) self.bookToolBar.fontBox.blockSignals(False) self.bookToolBar.fontSizeBox.blockSignals(False) self.bookToolBar.colorBoxFG.setStyleSheet( 'background-color: %s' % foreground.name()) self.bookToolBar.colorBoxBG.setStyleSheet( 'background-color: %s' % background.name()) current_tab.format_view( font, font_size, foreground, background, padding) def reset_profile(self): current_profile_index = self.bookToolBar.profileBox.currentIndex() current_profile_default = Settings(self).default_profiles[current_profile_index] self.bookToolBar.profileBox.setItemData( current_profile_index, current_profile_default, QtCore.Qt.UserRole) self.format_contentView() def show_settings(self): # TODO # The hiding of the settings dialog should uncheck the settings show action if not self.settings_dialog.isVisible(): self.settings_dialog.show() else: self.settings_dialog.hide() def closeEvent(self, event=None): # All tabs must be iterated upon here self.hide() self.settings_dialog.hide() self.temp_dir.remove() self.last_open_books = [] if self.tabWidget.count() > 1: all_metadata = [] for i in range(1, self.tabWidget.count()): tab_metadata = self.tabWidget.widget(i).metadata self.last_open_books.append(tab_metadata['path']) all_metadata.append(tab_metadata) Settings(self).save_settings() self.thread = BackGroundTabUpdate(self.database_path, all_metadata) self.thread.finished.connect(QtWidgets.qApp.exit) self.thread.start() else: Settings(self).save_settings() QtWidgets.qApp.exit() def main(): app = QtWidgets.QApplication(sys.argv) app.setApplicationName('Lector') # This is needed for QStandardPaths # and my own hubris form = MainUI() form.show() form.resizeEvent() app.exec_() if __name__ == '__main__': main()