大蟒蛇python教程共享Python自动化办公之编写PDF拆分工具

from pypdf2 import pdffilewriter, pdffilereader    infile = "input.pdf"  outfile = "output.pdf"    page_range = "1-2,6"  

output = pdffilewriter()  input_pdf = pdffilereader(open(infile, "rb"))  output_file = open(outfile, "wb")  

page_ranges = (x.split("-") for x in page_range.split(","))  range_list = [i for r in page_ranges for i in range(int(r[0]), int(r[-1]) + 1)]  

for p in range_list:      output.addpage(input_pdf.getpage(p - 1))  output.write(output_file)  

from appjar import gui  from pypdf2 import pdffilewriter, pdffilereader  from pathlib import path  

app = gui("pdf splitter", usettk=true)  app.setttktheme("default")  app.setsize(500, 200)  

app.addlabel("choose source pdf file")  app.addfileentry("input_file")    app.addlabel("select output directory")  app.adddirectoryentry("output_directory")    app.addlabel("output file name")  app.addentry("output_name")    app.addlabel("page ranges: 1,3,4-10")  app.addentry("page_ranges")  

app.addbuttons(["process", "quit"], press)  

# start the gui  app.go()  

def press(button):      if button == "process":          src_file = app.getentry("input_file")          dest_dir = app.getentry("output_directory")          page_range = app.getentry("page_ranges")          out_file = app.getentry("output_name")          errors, error_msg = validate_inputs(src_file, dest_dir, page_range, out_file)          if errors:              app.errorbox("error", "n".join(error_msg), parent=none)          else:              split_pages(src_file, page_range, path(dest_dir, out_file))      else:          app.stop()  

def validate_inputs(input_file, output_dir, range, file_name):      errors = false      error_msgs = []        # make sure a pdf is selected      if path(input_file).suffix.upper() != ".pdf":          errors = true          error_msgs.append("please select a pdf input file")        # make sure a range is selected      if len(range) < 1:          errors = true          error_msgs.append("please enter a valid page range")        # check for a valid directory      if not(path(output_dir)).exists():          errors = true          error_msgs.append("please select a valid output directory")        # check for a file name      if len(file_name) < 1:          errors = true          error_msgs.append("please enter a file name")        return(errors, error_msgs)  

def split_pages(input_file, page_range, out_file):      output = pdffilewriter()      input_pdf = pdffilereader(open(input_file, "rb"))      output_file = open(out_file, "wb")        page_ranges = (x.split("-") for x in page_range.split(","))      range_list = [i for r in page_ranges for i in range(int(r[0]), int(r[-1]) + 1)]        for p in range_list:          # need to subtract 1 because pages are 0 indexed          try:              output.addpage(input_pdf.getpage(p - 1))          except indexerror:              # alert the user and stop adding pages              app.infobox("info", "range exceeded number of pages in input.nfile will still be saved.")              break      output.write(output_file)        if(app.questionbox("file save", "output pdf saved. do you want to quit?")):          app.stop()