diff --git a/src/calibre/ebooks/metadata/book/merge_book_metadata.py b/src/calibre/ebooks/metadata/book/merge_book_metadata.py new file mode 100644 index 0000000000..c72a4ea8d1 --- /dev/null +++ b/src/calibre/ebooks/metadata/book/merge_book_metadata.py @@ -0,0 +1,97 @@ +#!/usr/bin/env python + + +__license__ = 'GPL v3' +__copyright__ = '2010, Kovid Goyal ' +__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) \ No newline at end of file diff --git a/src/calibre/gui2/actions/edit_metadata.py b/src/calibre/gui2/actions/edit_metadata.py index 53a3662dc3..099024e697 100644 --- a/src/calibre/gui2/actions/edit_metadata.py +++ b/src/calibre/gui2/actions/edit_metadata.py @@ -18,6 +18,7 @@ from qt.core import ( from calibre.db.errors import NoSuchFormat from calibre.ebooks.metadata import authors_to_string 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.sources.prefs import msprefs 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.library.comments import merge_comments from calibre.utils.config import tweaks -from calibre.utils.date import is_date_undefined from calibre.utils.icu import sort_key from calibre.utils.localization import ngettext from polyglot.builtins import iteritems @@ -733,92 +733,7 @@ class EditMetadataAction(InterfaceAction): def merge_metadata(self, dest_id, src_ids, replace_cover=False): db = self.gui.library_view.model().db - 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) + merge_book_metadata(db, dest_id, src_ids, replace_cover) # }}} def edit_device_collections(self, view, oncard=None):