mirror of
https://github.com/kovidgoyal/calibre.git
synced 2025-06-06 07:04:20 -04:00
166 lines
6.7 KiB
Python
166 lines
6.7 KiB
Python
#!/usr/bin/env python
|
|
# vim:fileencoding=UTF-8:ts=4:sw=4:sta:et:sts=4:ai
|
|
from __future__ import (unicode_literals, division, absolute_import,
|
|
print_function)
|
|
|
|
__license__ = 'GPL v3'
|
|
__copyright__ = '2011, Kovid Goyal <kovid@kovidgoyal.net>'
|
|
__docformat__ = 'restructuredtext en'
|
|
|
|
if False:
|
|
# This is here to keep my python error checker from complaining about
|
|
# the builtin functions that will be defined by the plugin loading system
|
|
# You do not need this code in your plugins
|
|
get_icons = get_resources = None
|
|
|
|
from PyQt4.Qt import QDialog, QVBoxLayout, QPushButton, QMessageBox, QLabel
|
|
|
|
from calibre_plugins.interface_demo.config import prefs
|
|
|
|
class DemoDialog(QDialog):
|
|
|
|
def __init__(self, gui, icon, do_user_config):
|
|
QDialog.__init__(self, gui)
|
|
self.gui = gui
|
|
self.do_user_config = do_user_config
|
|
|
|
# The current database shown in the GUI
|
|
# db is an instance of the class LibraryDatabase2 from database.py
|
|
# This class has many, many methods that allow you to do a lot of
|
|
# things.
|
|
self.db = gui.current_db
|
|
|
|
self.l = QVBoxLayout()
|
|
self.setLayout(self.l)
|
|
|
|
self.label = QLabel(prefs['hello_world_msg'])
|
|
self.l.addWidget(self.label)
|
|
|
|
self.setWindowTitle('Interface Plugin Demo')
|
|
self.setWindowIcon(icon)
|
|
|
|
self.about_button = QPushButton('About', self)
|
|
self.about_button.clicked.connect(self.about)
|
|
self.l.addWidget(self.about_button)
|
|
|
|
self.marked_button = QPushButton(
|
|
'Show books with only one format in the calibre GUI', self)
|
|
self.marked_button.clicked.connect(self.marked)
|
|
self.l.addWidget(self.marked_button)
|
|
|
|
self.view_button = QPushButton(
|
|
'View the most recently added book', self)
|
|
self.view_button.clicked.connect(self.view)
|
|
self.l.addWidget(self.view_button)
|
|
|
|
self.update_metadata_button = QPushButton(
|
|
'Update metadata in a book\'s files', self)
|
|
self.update_metadata_button.clicked.connect(self.update_metadata)
|
|
self.l.addWidget(self.update_metadata_button)
|
|
|
|
self.conf_button = QPushButton(
|
|
'Configure this plugin', self)
|
|
self.conf_button.clicked.connect(self.config)
|
|
self.l.addWidget(self.conf_button)
|
|
|
|
self.resize(self.sizeHint())
|
|
|
|
def about(self):
|
|
# Get the about text from a file inside the plugin zip file
|
|
# The get_resources function is a builtin function defined for all your
|
|
# plugin code. It loads files from the plugin zip file. It returns
|
|
# the bytes from the specified file.
|
|
#
|
|
# Note that if you are loading more than one file, for performance, you
|
|
# should pass a list of names to get_resources. In this case,
|
|
# get_resources will return a dictionary mapping names to bytes. Names that
|
|
# are not found in the zip file will not be in the returned dictionary.
|
|
text = get_resources('about.txt')
|
|
QMessageBox.about(self, 'About the Interface Plugin Demo',
|
|
text.decode('utf-8'))
|
|
|
|
def marked(self):
|
|
''' Show books with only one format '''
|
|
fmt_idx = self.db.FIELD_MAP['formats']
|
|
matched_ids = set()
|
|
for record in self.db.data.iterall():
|
|
# Iterate over all records
|
|
fmts = record[fmt_idx]
|
|
# fmts is either None or a comma separated list of formats
|
|
if fmts and ',' not in fmts:
|
|
matched_ids.add(record[0])
|
|
# Mark the records with the matching ids
|
|
self.db.set_marked_ids(matched_ids)
|
|
|
|
# Tell the GUI to search for all marked records
|
|
self.gui.search.setEditText('marked:true')
|
|
self.gui.search.do_search()
|
|
|
|
def view(self):
|
|
''' View the most recently added book '''
|
|
most_recent = most_recent_id = None
|
|
timestamp_idx = self.db.FIELD_MAP['timestamp']
|
|
|
|
for record in self.db.data:
|
|
# Iterate over all currently showing records
|
|
timestamp = record[timestamp_idx]
|
|
if most_recent is None or timestamp > most_recent:
|
|
most_recent = timestamp
|
|
most_recent_id = record[0]
|
|
|
|
if most_recent_id is not None:
|
|
# Get the row number of the id as shown in the GUI
|
|
row_number = self.db.row(most_recent_id)
|
|
# Get a reference to the View plugin
|
|
view_plugin = self.gui.iactions['View']
|
|
# Ask the view plugin to launch the viewer for row_number
|
|
view_plugin._view_books([row_number])
|
|
|
|
def update_metadata(self):
|
|
'''
|
|
Set the metadata in the files in the selected book's record to
|
|
match the current metadata in the database.
|
|
'''
|
|
from calibre.ebooks.metadata.meta import set_metadata
|
|
from calibre.gui2 import error_dialog, info_dialog
|
|
|
|
# Get currently selected books
|
|
rows = self.gui.library_view.selectionModel().selectedRows()
|
|
if not rows or len(rows) == 0:
|
|
return error_dialog(self.gui, 'Cannot update metadata',
|
|
'No books selected', show=True)
|
|
# Map the rows to book ids
|
|
ids = list(map(self.gui.library_view.model().id, rows))
|
|
for book_id in ids:
|
|
# Get the current metadata for this book from the db
|
|
mi = self.db.get_metadata(book_id, index_is_id=True,
|
|
get_cover=True, cover_as_data=True)
|
|
fmts = self.db.formats(book_id, index_is_id=True)
|
|
if not fmts: continue
|
|
for fmt in fmts.split(','):
|
|
fmt = fmt.lower()
|
|
# Get a python file object for the format. This will be either
|
|
# an in memory file or a temporary on disk file
|
|
ffile = self.db.format(book_id, fmt, index_is_id=True,
|
|
as_file=True)
|
|
# Set metadata in the format
|
|
set_metadata(ffile, mi, fmt)
|
|
ffile.seek(0)
|
|
# Now replace the file in the calibre library with the updated
|
|
# file. We dont use add_format_with_hooks as the hooks were
|
|
# already run when the file was first added to calibre.
|
|
ffile.name = 'xxx' # add_format() will not work if the file
|
|
# path of the file being added is the same
|
|
# as the path of the file being replaced
|
|
self.db.add_format(book_id, fmt, ffile, index_is_id=True)
|
|
|
|
info_dialog(self, 'Updated files',
|
|
'Updated the metadata in the files of %d book(s)'%len(ids),
|
|
show=True)
|
|
|
|
def config(self):
|
|
self.do_user_config(parent=self)
|
|
# Apply the changes
|
|
self.label.setText(prefs['hello_world_msg'])
|
|
|