add one optional arg to default file storage .delete_pages

pull/3/head
Eugen Ciur 2020-06-04 19:16:51 +02:00
parent 4ea658a1b9
commit f1ddc389dc
2 changed files with 13 additions and 12 deletions

View File

@ -1,15 +1,13 @@
import logging
import os import os
import shutil
from os import listdir from os import listdir
from os.path import isdir, join from os.path import isdir, join
import logging
import shutil
from mglib.step import Steps
from mglib.utils import (
safe_to_delete,
get_assigns_after_delete
)
from mglib import pdftk from mglib import pdftk
from mglib.path import PagePath, DocumentPath from mglib.path import DocumentPath, PagePath
from mglib.step import Steps
from mglib.utils import get_assigns_after_delete, safe_to_delete
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -246,7 +244,7 @@ class Storage:
return doc_path.version + 1 return doc_path.version + 1
def delete_pages(self, doc_path, page_numbers): def delete_pages(self, doc_path, page_numbers, total_page_count=None):
""" """
Delets pages in the document pointed by doc_path. Delets pages in the document pointed by doc_path.
doc_path is an instance of mglib.path.DocumentPath doc_path is an instance of mglib.path.DocumentPath
@ -272,7 +270,10 @@ class Storage:
page_numbers page_numbers
) )
page_count = self.get_pagecount(doc_path) if total_page_count:
page_count = total_page_count
else:
page_count = self.get_pagecount(doc_path)
if len(page_numbers) > page_count: if len(page_numbers) > page_count:
logger.error( logger.error(
f"deleted_pages({page_numbers}) > page_count({page_count})" f"deleted_pages({page_numbers}) > page_count({page_count})"

View File

@ -1,4 +1,4 @@
from setuptools import setup, find_packages from setuptools import find_packages, setup
with open("README.md", "r") as fh: with open("README.md", "r") as fh:
long_description = fh.read() long_description = fh.read()
@ -6,7 +6,7 @@ with open("README.md", "r") as fh:
setup( setup(
name="mglib", name="mglib",
version="1.0.1", version="1.0.3",
author="Eugen Ciur", author="Eugen Ciur",
author_email="eugen@papermerge.com", author_email="eugen@papermerge.com",
url="https://github.com/papermerge/mglib", url="https://github.com/papermerge/mglib",