Made sure that parts are below the part_length even after adding the checksum.

2017-05-05 03:33:20 +02:00 · 2017-05-05 03:33:20 +02:00 · a0ab29bb8d
commit a0ab29bb8d
parent 7e0734b0cc
1 changed files with 6 additions and 1 deletions
--- a/paperize/main.py
+++ b/paperize/main.py
@ -199,7 +199,12 @@ def prepare_data(data, no_parts, part_length, file_name, hashsum):
        ret.append(part_header + part + PART_TRAILER)
        data = fill(data, 80)[length:]
        i += 1
-    ret[-1] = ret[-1] + CHECKSUM.format(sum=hashsum)
+    checksum = CHECKSUM.format(sum=hashsum)
+    if (len(ret[-1]) + len(checksum) < part_length):
+        ret[-1] = ret[-1] + checksum
+    else:
+        ret.append(PART_HEADER.format(part=len(ret)+1, parts=no_parts)[:-1]
+                + PART_TRAILER + checksum)
    return ret