From b4fc57d137c78797719f02e64b544983c6b3bec0 Mon Sep 17 00:00:00 2001
From: jiaminxin <33108686+jiandandaoxingfu@users.noreply.github.com>
Date: Mon, 18 Jan 2021 15:44:36 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8B=86=E5=88=86=E6=94=AF=E6=8C=81=E5=A4=9A?=
 =?UTF-8?q?=E4=B8=AA=E6=96=87=E4=BB=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 client.py |  2 +-
 pdfdo.py  | 67 +++++++++++++++++++++++++++++--------------------------
 2 files changed, 36 insertions(+), 33 deletions(-)

diff --git a/client.py b/client.py
index d29ce72..431dee8 100644
--- a/client.py
+++ b/client.py
@@ -78,7 +78,7 @@ def update_state():
 top = 20;
 margin = 36;
 labels = ['选择文件', '拆分每页', '部分拆分', '文件合并', '文件剪切', '文件旋转', '添加页码'];
-default_values = ['', '支持单个文件', '支持单个文件， 如：[(1, 3), (20, 25), (30, 40)]', '合并后文件名.pdf', '支持单个文件，如：[10, 20, 10, 20, "even", 1] (注：左, 右, 下, 上, odd/even/all, 0/1: 0为全部, 1为测试10张)', '支持单个文件，如：[90, 1] (注：旋转度数是90的整数倍, 0/1: 1为测试一张, 0为全部)', '支持多个文件']
+default_values = ['', '支持多个文件', '支持多个文件， 如：[(1, 3), (20, 25), (30, 40)]', '合并后文件名.pdf', '支持单个文件，如：[10, 20, 10, 20, "even", 1] (注：左, 右, 下, 上, odd/even/all, 0/1: 0为全部, 1为测试10张)', '支持单个文件，如：[90, 1] (注：旋转度数是90的整数倍, 0/1: 1为测试一张, 0为全部)', '支持多个文件']
 length = len(labels)
 
 for i in range(length):
diff --git a/pdfdo.py b/pdfdo.py
index 22b7358..b84daea 100644
--- a/pdfdo.py
+++ b/pdfdo.py
@@ -12,32 +12,34 @@ def pdf_info(self):
 
 
     def split_pdf_each(self): 
-        try:
-            pdf_input = PdfFileReader(open(self.infn[0], 'rb')) 
-            pages = pdf_input.getNumPages()  
-            self.message = '正在拆分...'
-            for i in range(pages): 
-                pdf_output = PdfFileWriter()
-                pdf_output.addPage(pdf_input.getPage(i)) 
-                pdf_output.write(open(self.infn[0][:-4] + '-' + str(i + 1) + '.pdf', 'wb')) 
-                self.message = '已拆分%.2f' %((i + 1)/pages * 100) + '%'
-            self.message = '拆分完成'
-        except:
-            self.message = '出错了，请检查输入格式是否正确'
+        for infn in self.infn:
+            try:
+                pdf_input = PdfFileReader(open(infn, 'rb')) 
+                pages = pdf_input.getNumPages()  
+                self.message = '正在拆分...'
+                for i in range(pages): 
+                    pdf_output = PdfFileWriter()
+                    pdf_output.addPage(pdf_input.getPage(i)) 
+                    pdf_output.write(open(infn[:-4] + '-' + str(i + 1) + '.pdf', 'wb')) 
+                    self.message = '已拆分%.2f' %((i + 1)/pages * 100) + '%'
+                self.message = '拆分完成'
+            except:
+                self.message = '出错了，请检查输入格式是否正确'
 
     def split_pdf_parts(self): 
-        try:
-            pdf_input = PdfFileReader(open(self.infn[0], 'rb'))
-            self.message = '正在拆分...'
-            for part in self.params:
-                pdf_output = PdfFileWriter()
-                for i in range(part[0] - 1, part[1]):
-                    pdf_output.addPage(pdf_input.getPage(i));
-                pdf_output.write(open(self.infn[0][:-4] + '-' + str(part[0]) + '-' + str(part[1]) + '.pdf', 'wb'))
-                self.message = '第%d部分已拆分'%(self.params.index(part) + 1)
-            self.message = '拆分完成'
-        except:
-            self.message = '出错了，请检查输入格式是否正确'
+        for infn in self.infn:
+            try:
+                pdf_input = PdfFileReader(open(infn, 'rb'))
+                self.message = '正在拆分...'
+                for part in self.params:
+                    pdf_output = PdfFileWriter()
+                    for i in range(part[0] - 1, part[1]):
+                        pdf_output.addPage(pdf_input.getPage(i));
+                    pdf_output.write(open(infn[:-4] + '-' + str(part[0]) + '-' + str(part[1]) + '.pdf', 'wb'))
+                    self.message = '第%d部分已拆分'%(self.params.index(part) + 1)
+                self.message = '拆分完成'
+            except:
+                self.message = '出错了，请检查输入格式是否正确'
 
     def merge_pdf(self): 
         try:
@@ -96,11 +98,11 @@ def rotate_pdf(self):
             self.message = '出错了，请检查输入格式是否正确(旋转角度为90的倍数)'
 
     def add_watermark(self):
-        try:
-            self.message = '正在添加页码'
-            water_pdf = PdfFileReader(open('page-number.pdf', 'rb'));
-            water_pages = water_pdf.getNumPages()
-            for infn in self.infn:
+        self.message = '正在添加页码'
+        water_pdf = PdfFileReader(open('page-number.pdf', 'rb'));
+        water_pages = water_pdf.getNumPages()
+        for infn in self.infn:
+            try:
                 pdf_input = PdfFileReader(open(infn, 'rb'));
                 pages = min(pdf_input.getNumPages(), water_pages);
                 pdf_output = PdfFileWriter();
@@ -111,6 +113,7 @@ def add_watermark(self):
                     pdf_output.addPage(page);
                     self.message = str(i) + '/' + str(pages) + ': ' + infn
                 pdf_output.write(open(infn[:-4] + '-number2.pdf', 'wb'));
-            self.message = '页码添加完成'
-        except:
-            self.message = '出错了，请检查输入格式是否正确(page-number.pdf文件要求和程序在同一目录)'
\ No newline at end of file
+                self.message = '页码添加完成'
+            except: 
+                self.message = '出错了，请检查输入格式是否正确(page-number.pdf文件要求和程序在同一目录)'
+