美文网首页
一段统计 PDF 页数的代码

一段统计 PDF 页数的代码

作者: bitstring | 来源:发表于2016-12-02 12:16 被阅读0次
 import re, os, glob, sys

 rxcountpages = re.compile(r"/Type\s*/Page([^s]|$)", re.MULTILINE|re.DOTALL)

 def count_pages(filename):
     data = file(filename,"rb").read()
     return len(rxcountpages.findall(data))

 def sum_pages(args):
     if len(args) > 1:
         if args[1].endswith(".pdf"):
             return count_pages(args[1])
         else:
             os.chdir(args[1])
     total_pages = 0
     fnames = glob.glob("./*.pdf")
     for fname in fnames:
         total_pages = total_pages + count_pages(fname)
     return total_pages

 if __name__=="__main__":
     print(sum_pages(sys.argv))

相关文章

网友评论

      本文标题:一段统计 PDF 页数的代码

      本文链接:https://www.haomeiwen.com/subject/joafmttx.html