Add support for stapler (lighter all-python alternative to pdftk)

hannoL · Jun 7, 2010 · a13941e · a13941e
1 parent c5912e1
commit a13941e
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/springer_download.py b/springer_download.py
@@ -17,8 +17,12 @@ class SpringerURLopener(urllib.FancyURLopener):
 
 # validate CLI arguments and start downloading
 def main(argv):
-    if not findInPath("pdftk"):
-        error("You have to install pdftk.")
+    if findInPath("pdftk"):
+        pdfcat = lambda fileList, bookTitlePath: "pdftk %s cat output '%s'" % (" ".join(fileList), bookTitlePath)
+    elif findInPath("stapler"):
+        pdfcat = lambda fileList, bookTitlePath: "stapler cat %s '%s'" % (" ".join(fileList), bookTitlePath)
+    else:
+        error("You have to install pdftk (http://www.accesspdf.com/pdftk/) or stapler (http://github.com/hellerbarde/stapler).")
     if not findInPath("iconv"):
         error("You have to install iconv.")
 
@@ -177,7 +181,7 @@ def main(argv):
     if len(fileList) == 1:
       shutil.move(fileList[0], bookTitlePath)
     else:
-      os.system("pdftk %s cat output '%s'" % (" ".join(fileList), bookTitlePath))
+      os.system(pdfcat(fileList, bookTitlePath))
 
     # cleanup
     os.chdir(curDir)