class PluginWidget(QWidget): TITLE = _('CSV/XML options') HELP = _('Options specific to') + ' CSV/XML ' + _('output') sync_enabled = False formats = {'csv', 'xml'} handles_scrolling = True def __init__(self, parent=None): QWidget.__init__(self, parent) self.l = l = QVBoxLayout(self) self.la = la = QLabel(_('Fields to include in output:')) la.setWordWrap(True) l.addWidget(la) self.db_fields = QListWidget(self) l.addWidget(self.db_fields) self.la2 = la = QLabel(_('Drag and drop to re-arrange fields')) l.addWidget(la) self.db_fields.setDragEnabled(True) self.db_fields.setDragDropMode( QAbstractItemView.DragDropMode.InternalMove) self.db_fields.setDefaultDropAction( Qt.DropAction.CopyAction if ismacos else Qt.DropAction.MoveAction) self.db_fields.setAlternatingRowColors(True) self.db_fields.setObjectName("db_fields") def initialize(self, catalog_name, db): self.name = catalog_name from calibre.library.catalogs import FIELDS db = get_gui().current_db self.all_fields = {x for x in FIELDS if x != 'all'} | set(db.custom_field_keys()) sort_order, fields = get_saved_field_data(self.name, self.all_fields) fm = db.field_metadata def name(x): if x == 'isbn': return 'ISBN' if x == 'library_name': return _('Library name') if x.endswith('_index'): return name(x[:-len('_index')]) + ' ' + _('Number') return fm[x].get('name') or x def key(x): return (sort_order.get(x, 10000), name(x)) self.db_fields.clear() for x in sorted(self.all_fields, key=key): QListWidgetItem(name(x) + ' (%s)' % x, self.db_fields).setData(Qt.ItemDataRole.UserRole, x) if x.startswith('#') and fm[x]['datatype'] == 'series': x += '_index' QListWidgetItem(name(x) + ' (%s)' % x, self.db_fields).setData( Qt.ItemDataRole.UserRole, x) # Restore the activated fields from last use for x in range(self.db_fields.count()): item = self.db_fields.item(x) item.setCheckState(Qt.CheckState.Checked if str( item.data(Qt.ItemDataRole.UserRole)) in fields else Qt.CheckState.Unchecked) def options(self): # Save the currently activated fields fields, all_fields = [], [] for x in range(self.db_fields.count()): item = self.db_fields.item(x) all_fields.append(str(item.data(Qt.ItemDataRole.UserRole))) if item.checkState() == Qt.CheckState.Checked: fields.append(str(item.data(Qt.ItemDataRole.UserRole))) set_saved_field_data(self.name, fields, {x: i for i, x in enumerate(all_fields)}) # Return a dictionary with current options for this widget if len(fields): return {'fields': fields} else: return {'fields': ['all']}
class SavedSearchEditor(Dialog): def __init__(self, parent, initial_search=None): self.initial_search = initial_search Dialog.__init__(self, _('Manage Saved searches'), 'manage-saved-searches', parent) def setup_ui(self): from calibre.gui2.ui import get_gui db = get_gui().current_db self.l = l = QVBoxLayout(self) b = self.bb.addButton(_('&Add search'), QDialogButtonBox.ButtonRole.ActionRole) b.setIcon(QIcon(I('plus.png'))) b.clicked.connect(self.add_search) b = self.bb.addButton(_('&Remove search'), QDialogButtonBox.ButtonRole.ActionRole) b.setIcon(QIcon(I('minus.png'))) b.clicked.connect(self.del_search) b = self.bb.addButton(_('&Edit search'), QDialogButtonBox.ButtonRole.ActionRole) b.setIcon(QIcon(I('modified.png'))) b.clicked.connect(self.edit_search) self.slist = QListWidget(self) self.slist.setStyleSheet('QListView::item { padding: 3px }') self.slist.activated.connect(self.edit_search) self.slist.setAlternatingRowColors(True) self.searches = { name: db.saved_search_lookup(name) for name in db.saved_search_names() } self.populate_search_list() if self.initial_search is not None and self.initial_search in self.searches: self.select_search(self.initial_search) elif self.searches: self.slist.setCurrentRow(0) self.slist.currentItemChanged.connect(self.current_index_changed) l.addWidget(self.slist) self.desc = la = QLabel('\xa0') la.setWordWrap(True) l.addWidget(la) l.addWidget(self.bb) self.current_index_changed(self.slist.currentItem()) self.setMinimumHeight(500) self.setMinimumWidth(600) @property def current_search_name(self): i = self.slist.currentItem() if i is not None: ans = i.text() if ans in self.searches: return ans def keyPressEvent(self, ev): if ev.key() == Qt.Key.Key_Delete: self.del_search() return return Dialog.keyPressEvent(self, ev) def populate_search_list(self): self.slist.clear() for name in sorted(self.searches.keys(), key=sort_key): self.slist.addItem(name) def add_search(self): d = AddSavedSearch(parent=self, commit_changes=False, validate=self.validate_add) if d.exec_() != QDialog.DialogCode.Accepted: return name, expression = d.accepted_data self.searches[name] = expression self.populate_search_list() self.select_search(name) def del_search(self): n = self.current_search_name if n is not None: if not confirm( '<p>' + _('The current saved search will be ' '<b>permanently deleted</b>. Are you sure?') + '</p>', 'saved_search_editor_delete', self): return self.slist.takeItem(self.slist.currentRow()) del self.searches[n] def edit_search(self): n = self.current_search_name if not n: return d = AddSavedSearch(parent=self, commit_changes=False, label=_('Edit the name and/or expression below.'), validate=self.validate_edit) d.setWindowTitle(_('Edit saved search')) d.sname.setText(n) d.search.setPlainText(self.searches[n]) if d.exec_() != QDialog.DialogCode.Accepted: return name, expression = d.accepted_data self.slist.currentItem().setText(name) del self.searches[n] self.searches[name] = expression self.current_index_changed(self.slist.currentItem()) def duplicate_msg(self, name): return _( 'A saved search with the name {} already exists. Choose another name' ).format(name) def validate_edit(self, name, expression): q = self.current_search_name if icu_lower(name) in {icu_lower(n) for n in self.searches if n != q}: return self.duplicate_msg(name) def validate_add(self, name, expression): if icu_lower(name) in {icu_lower(n) for n in self.searches}: return self.duplicate_msg(name) def select_search(self, name): items = self.slist.findItems( name, Qt.MatchFlag.MatchFixedString | Qt.MatchFlag.MatchCaseSensitive) if items: self.slist.setCurrentItem(items[0]) def current_index_changed(self, item): n = self.current_search_name if n: t = self.searches[n] else: t = '' self.desc.setText('<p><b>{}</b>: '.format(_('Search expression')) + prepare_string_for_xml(t)) def accept(self): commit_searches(self.searches) Dialog.accept(self)