From 8dcb903a7b9302d58c9d7ba53f04befaf7f43117 Mon Sep 17 00:00:00 2001
From: Eugen Ciur <eugen@django-lessons.com>
Date: Sun, 17 May 2020 12:17:27 +0200
Subject: [PATCH] refactoring

---
 mglib/pdftk.py   | 7 +++++--
 mglib/storage.py | 1 +
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/mglib/pdftk.py b/mglib/pdftk.py
index 6f7e90f..6384bc5 100644
--- a/mglib/pdftk.py
+++ b/mglib/pdftk.py
@@ -134,12 +134,13 @@ def split_ranges(total, after=False, before=False):
 
 
 def paste_pages_into_existing_doc(
+    src,
     dst,
     data_list,
     after_page_number=False,
     before_page_number=False
 ):
-    page_count = get_pagecount(dst)
+    page_count = get_pagecount(src)
     list1, list2 = split_ranges(
         total=page_count,
         after=after_page_number,
@@ -155,7 +156,7 @@ def paste_pages_into_existing_doc(
     letters_pages_after = []
 
     letters_2_doc_map.append(
-        f"A={dst.url()}"
+        f"A={src}"
     )
 
     for idx in range(0, len(data_list)):
@@ -204,6 +205,7 @@ def paste_pages_into_existing_doc(
 
 
 def paste_pages(
+    src,
     dst,
     data_list,
     dst_doc_is_new=True,
@@ -246,6 +248,7 @@ def paste_pages(
     """
     if not dst_doc_is_new:
         return paste_pages_into_existing_doc(
+            src=src,
             dst=dst,
             data_list=data_list,
             after_page_number=after_page_number,
diff --git a/mglib/storage.py b/mglib/storage.py
index 764740a..7f9367e 100644
--- a/mglib/storage.py
+++ b/mglib/storage.py
@@ -326,6 +326,7 @@ class Storage:
         )
 
         pdftk.paste_pages(
+            src=self.abspath(dest_doc_path),
             dst=self.abspath(next_ver_dp),
             data_list=data_list,
             dst_doc_is_new=dest_doc_is_new,