From 09a34cf7d92e8f4e029fdfc0922f7fb2ca1a444c Mon Sep 17 00:00:00 2001 From: NoDRM Date: Mon, 29 Nov 2021 16:33:45 +0100 Subject: [PATCH] Fix watermark stuff --- DeDRM_plugin/epubwatermark.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/DeDRM_plugin/epubwatermark.py b/DeDRM_plugin/epubwatermark.py index e6115b3..9545975 100644 --- a/DeDRM_plugin/epubwatermark.py +++ b/DeDRM_plugin/epubwatermark.py @@ -198,7 +198,9 @@ def removeOPFwatermarks(object, path_to_ebook): # This regex replaces all "idX_Y" IDs with "id_Y", removing the watermark IDs. pre_remove = container_str_new container_str_new = re.sub(r'((\r\n|\r|\n)\s*)?\<\!\-\-\s*Wygenerowane przez elibri dla zamówienia numer [0-9a-fA-F]+\s*\-\-\>', '', container_str_new) - container_str_new = re.sub(r'\=\"id[0-9]+_([0-9]+)\"', r'="id_\1"', container_str_new) + if pre_remove != container_str_new: + # To prevent this Regex from applying to books without that watermark, only do that if the watermark above was found. + container_str_new = re.sub(r'\=\"id[0-9]+_([0-9]+)\"', r'="id_\1"', container_str_new) if pre_remove != container_str_new: had_elibri = True