mirror of
https://github.com/kovidgoyal/calibre.git
synced 2025-07-07 18:24:30 -04:00
Merge branch 'master' of https://github.com/cbhaley/calibre
This commit is contained in:
Kovid Goyal
commit
e9ff107b1e
97
src/calibre/ebooks/metadata/book/merge_book_metadata.py
Normal file
97
src/calibre/ebooks/metadata/book/merge_book_metadata.py
Normal file
@ -0,0 +1,97 @@
|
|||||||
|
#!/usr/bin/env python
|
||||||
|
|
||||||
|
|
||||||
|
__license__ = 'GPL v3'
|
||||||
|
__copyright__ = '2010, Kovid Goyal <kovid@kovidgoyal.net>'
|
||||||
|
__docformat__ = 'restructuredtext en'
|
||||||
|
|
||||||
|
from calibre.utils.date import is_date_undefined
|
||||||
|
|
||||||
|
|
||||||
|
def merge_book_metadata(db, dest_id, src_ids, replace_cover=False):
|
||||||
|
dest_mi = db.get_metadata(dest_id, index_is_id=True)
|
||||||
|
merged_identifiers = db.get_identifiers(dest_id, index_is_id=True)
|
||||||
|
orig_dest_comments = dest_mi.comments
|
||||||
|
dest_cover = orig_dest_cover = db.cover(dest_id, index_is_id=True)
|
||||||
|
had_orig_cover = bool(dest_cover)
|
||||||
|
|
||||||
|
def is_null_date(x):
|
||||||
|
return x is None or is_date_undefined(x)
|
||||||
|
|
||||||
|
for src_id in src_ids:
|
||||||
|
src_mi = db.get_metadata(src_id, index_is_id=True)
|
||||||
|
|
||||||
|
if src_mi.comments and orig_dest_comments != src_mi.comments:
|
||||||
|
if not dest_mi.comments:
|
||||||
|
dest_mi.comments = src_mi.comments
|
||||||
|
else:
|
||||||
|
dest_mi.comments = str(dest_mi.comments) + '\n\n' + str(src_mi.comments)
|
||||||
|
if src_mi.title and dest_mi.is_null('title'):
|
||||||
|
dest_mi.title = src_mi.title
|
||||||
|
dest_mi.title_sort = src_mi.title_sort
|
||||||
|
if (src_mi.authors and src_mi.authors[0] != _('Unknown')) and (not dest_mi.authors or dest_mi.authors[0] == _('Unknown')):
|
||||||
|
dest_mi.authors = src_mi.authors
|
||||||
|
dest_mi.author_sort = src_mi.author_sort
|
||||||
|
if src_mi.tags:
|
||||||
|
if not dest_mi.tags:
|
||||||
|
dest_mi.tags = src_mi.tags
|
||||||
|
else:
|
||||||
|
dest_mi.tags.extend(src_mi.tags)
|
||||||
|
if not dest_cover or replace_cover:
|
||||||
|
src_cover = db.cover(src_id, index_is_id=True)
|
||||||
|
if src_cover:
|
||||||
|
dest_cover = src_cover
|
||||||
|
replace_cover = False
|
||||||
|
if not dest_mi.publisher:
|
||||||
|
dest_mi.publisher = src_mi.publisher
|
||||||
|
if not dest_mi.rating:
|
||||||
|
dest_mi.rating = src_mi.rating
|
||||||
|
if not dest_mi.series:
|
||||||
|
dest_mi.series = src_mi.series
|
||||||
|
dest_mi.series_index = src_mi.series_index
|
||||||
|
if is_null_date(dest_mi.pubdate) and not is_null_date(src_mi.pubdate):
|
||||||
|
dest_mi.pubdate = src_mi.pubdate
|
||||||
|
|
||||||
|
src_identifiers = db.get_identifiers(src_id, index_is_id=True)
|
||||||
|
src_identifiers.update(merged_identifiers)
|
||||||
|
merged_identifiers = src_identifiers.copy()
|
||||||
|
|
||||||
|
if merged_identifiers:
|
||||||
|
dest_mi.set_identifiers(merged_identifiers)
|
||||||
|
db.set_metadata(dest_id, dest_mi, ignore_errors=False)
|
||||||
|
|
||||||
|
if dest_cover and (not had_orig_cover or dest_cover is not orig_dest_cover):
|
||||||
|
db.set_cover(dest_id, dest_cover)
|
||||||
|
|
||||||
|
for key in db.field_metadata: # loop thru all defined fields
|
||||||
|
fm = db.field_metadata[key]
|
||||||
|
if not fm['is_custom']:
|
||||||
|
continue
|
||||||
|
dt = fm['datatype']
|
||||||
|
colnum = fm['colnum']
|
||||||
|
# Get orig_dest_comments before it gets changed
|
||||||
|
if dt == 'comments':
|
||||||
|
orig_dest_value = db.get_custom(dest_id, num=colnum, index_is_id=True)
|
||||||
|
|
||||||
|
for src_id in src_ids:
|
||||||
|
dest_value = db.get_custom(dest_id, num=colnum, index_is_id=True)
|
||||||
|
src_value = db.get_custom(src_id, num=colnum, index_is_id=True)
|
||||||
|
if (dt == 'comments' and src_value and src_value != orig_dest_value):
|
||||||
|
if not dest_value:
|
||||||
|
db.set_custom(dest_id, src_value, num=colnum)
|
||||||
|
else:
|
||||||
|
dest_value = str(dest_value) + '\n\n' + str(src_value)
|
||||||
|
db.set_custom(dest_id, dest_value, num=colnum)
|
||||||
|
if (dt in {'bool', 'int', 'float', 'rating', 'datetime'} and dest_value is None):
|
||||||
|
db.set_custom(dest_id, src_value, num=colnum)
|
||||||
|
if (dt == 'series' and not dest_value and src_value):
|
||||||
|
src_index = db.get_custom_extra(src_id, num=colnum, index_is_id=True)
|
||||||
|
db.set_custom(dest_id, src_value, num=colnum, extra=src_index)
|
||||||
|
if ((dt == 'enumeration' or (dt == 'text' and not fm['is_multiple'])) and not dest_value):
|
||||||
|
db.set_custom(dest_id, src_value, num=colnum)
|
||||||
|
if (dt == 'text' and fm['is_multiple'] and src_value):
|
||||||
|
if not dest_value:
|
||||||
|
dest_value = src_value
|
||||||
|
else:
|
||||||
|
dest_value.extend(src_value)
|
||||||
|
db.set_custom(dest_id, dest_value, num=colnum)
|
||||||
@ -18,6 +18,7 @@ from qt.core import (
|
|||||||
from calibre.db.errors import NoSuchFormat
|
from calibre.db.errors import NoSuchFormat
|
||||||
from calibre.ebooks.metadata import authors_to_string
|
from calibre.ebooks.metadata import authors_to_string
|
||||||
from calibre.ebooks.metadata.book.base import Metadata
|
from calibre.ebooks.metadata.book.base import Metadata
|
||||||
|
from calibre.ebooks.metadata.book.merge_book_metadata import merge_book_metadata
|
||||||
from calibre.ebooks.metadata.opf2 import OPF, metadata_to_opf
|
from calibre.ebooks.metadata.opf2 import OPF, metadata_to_opf
|
||||||
from calibre.ebooks.metadata.sources.prefs import msprefs
|
from calibre.ebooks.metadata.sources.prefs import msprefs
|
||||||
from calibre.gui2 import Dispatcher, error_dialog, gprefs, question_dialog
|
from calibre.gui2 import Dispatcher, error_dialog, gprefs, question_dialog
|
||||||
@ -28,7 +29,6 @@ from calibre.gui2.dialogs.device_category_editor import DeviceCategoryEditor
|
|||||||
from calibre.gui2.dialogs.metadata_bulk import MetadataBulkDialog
|
from calibre.gui2.dialogs.metadata_bulk import MetadataBulkDialog
|
||||||
from calibre.library.comments import merge_comments
|
from calibre.library.comments import merge_comments
|
||||||
from calibre.utils.config import tweaks
|
from calibre.utils.config import tweaks
|
||||||
from calibre.utils.date import is_date_undefined
|
|
||||||
from calibre.utils.icu import sort_key
|
from calibre.utils.icu import sort_key
|
||||||
from calibre.utils.localization import ngettext
|
from calibre.utils.localization import ngettext
|
||||||
from polyglot.builtins import iteritems
|
from polyglot.builtins import iteritems
|
||||||
@ -733,92 +733,7 @@ class EditMetadataAction(InterfaceAction):
|
|||||||
|
|
||||||
def merge_metadata(self, dest_id, src_ids, replace_cover=False):
|
def merge_metadata(self, dest_id, src_ids, replace_cover=False):
|
||||||
db = self.gui.library_view.model().db
|
db = self.gui.library_view.model().db
|
||||||
dest_mi = db.get_metadata(dest_id, index_is_id=True)
|
merge_book_metadata(db, dest_id, src_ids, replace_cover)
|
||||||
merged_identifiers = db.get_identifiers(dest_id, index_is_id=True)
|
|
||||||
orig_dest_comments = dest_mi.comments
|
|
||||||
dest_cover = orig_dest_cover = db.cover(dest_id, index_is_id=True)
|
|
||||||
had_orig_cover = bool(dest_cover)
|
|
||||||
|
|
||||||
def is_null_date(x):
|
|
||||||
return x is None or is_date_undefined(x)
|
|
||||||
|
|
||||||
for src_id in src_ids:
|
|
||||||
src_mi = db.get_metadata(src_id, index_is_id=True)
|
|
||||||
|
|
||||||
if src_mi.comments and orig_dest_comments != src_mi.comments:
|
|
||||||
if not dest_mi.comments:
|
|
||||||
dest_mi.comments = src_mi.comments
|
|
||||||
else:
|
|
||||||
dest_mi.comments = str(dest_mi.comments) + '\n\n' + str(src_mi.comments)
|
|
||||||
if src_mi.title and dest_mi.is_null('title'):
|
|
||||||
dest_mi.title = src_mi.title
|
|
||||||
dest_mi.title_sort = src_mi.title_sort
|
|
||||||
if (src_mi.authors and src_mi.authors[0] != _('Unknown')) and (not dest_mi.authors or dest_mi.authors[0] == _('Unknown')):
|
|
||||||
dest_mi.authors = src_mi.authors
|
|
||||||
dest_mi.author_sort = src_mi.author_sort
|
|
||||||
if src_mi.tags:
|
|
||||||
if not dest_mi.tags:
|
|
||||||
dest_mi.tags = src_mi.tags
|
|
||||||
else:
|
|
||||||
dest_mi.tags.extend(src_mi.tags)
|
|
||||||
if not dest_cover or replace_cover:
|
|
||||||
src_cover = db.cover(src_id, index_is_id=True)
|
|
||||||
if src_cover:
|
|
||||||
dest_cover = src_cover
|
|
||||||
replace_cover = False
|
|
||||||
if not dest_mi.publisher:
|
|
||||||
dest_mi.publisher = src_mi.publisher
|
|
||||||
if not dest_mi.rating:
|
|
||||||
dest_mi.rating = src_mi.rating
|
|
||||||
if not dest_mi.series:
|
|
||||||
dest_mi.series = src_mi.series
|
|
||||||
dest_mi.series_index = src_mi.series_index
|
|
||||||
if is_null_date(dest_mi.pubdate) and not is_null_date(src_mi.pubdate):
|
|
||||||
dest_mi.pubdate = src_mi.pubdate
|
|
||||||
|
|
||||||
src_identifiers = db.get_identifiers(src_id, index_is_id=True)
|
|
||||||
src_identifiers.update(merged_identifiers)
|
|
||||||
merged_identifiers = src_identifiers.copy()
|
|
||||||
|
|
||||||
if merged_identifiers:
|
|
||||||
dest_mi.set_identifiers(merged_identifiers)
|
|
||||||
db.set_metadata(dest_id, dest_mi, ignore_errors=False)
|
|
||||||
|
|
||||||
if dest_cover and (not had_orig_cover or dest_cover is not orig_dest_cover):
|
|
||||||
db.set_cover(dest_id, dest_cover)
|
|
||||||
|
|
||||||
for key in db.field_metadata: # loop thru all defined fields
|
|
||||||
fm = db.field_metadata[key]
|
|
||||||
if not fm['is_custom']:
|
|
||||||
continue
|
|
||||||
dt = fm['datatype']
|
|
||||||
colnum = fm['colnum']
|
|
||||||
# Get orig_dest_comments before it gets changed
|
|
||||||
if dt == 'comments':
|
|
||||||
orig_dest_value = db.get_custom(dest_id, num=colnum, index_is_id=True)
|
|
||||||
|
|
||||||
for src_id in src_ids:
|
|
||||||
dest_value = db.get_custom(dest_id, num=colnum, index_is_id=True)
|
|
||||||
src_value = db.get_custom(src_id, num=colnum, index_is_id=True)
|
|
||||||
if (dt == 'comments' and src_value and src_value != orig_dest_value):
|
|
||||||
if not dest_value:
|
|
||||||
db.set_custom(dest_id, src_value, num=colnum)
|
|
||||||
else:
|
|
||||||
dest_value = str(dest_value) + '\n\n' + str(src_value)
|
|
||||||
db.set_custom(dest_id, dest_value, num=colnum)
|
|
||||||
if (dt in {'bool', 'int', 'float', 'rating', 'datetime'} and dest_value is None):
|
|
||||||
db.set_custom(dest_id, src_value, num=colnum)
|
|
||||||
if (dt == 'series' and not dest_value and src_value):
|
|
||||||
src_index = db.get_custom_extra(src_id, num=colnum, index_is_id=True)
|
|
||||||
db.set_custom(dest_id, src_value, num=colnum, extra=src_index)
|
|
||||||
if ((dt == 'enumeration' or (dt == 'text' and not fm['is_multiple'])) and not dest_value):
|
|
||||||
db.set_custom(dest_id, src_value, num=colnum)
|
|
||||||
if (dt == 'text' and fm['is_multiple'] and src_value):
|
|
||||||
if not dest_value:
|
|
||||||
dest_value = src_value
|
|
||||||
else:
|
|
||||||
dest_value.extend(src_value)
|
|
||||||
db.set_custom(dest_id, dest_value, num=colnum)
|
|
||||||
# }}}
|
# }}}
|
||||||
|
|
||||||
def edit_device_collections(self, view, oncard=None):
|
def edit_device_collections(self, view, oncard=None):
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user