Add backends arg

1 files changed, 15 insertions, 4 deletions

diff --git a/mkbinder.py b/mkbinder.py
index 2c02ef5..59bfccb 100644..100755
--- a/mkbinder.py
+++ b/mkbinder.py
@@ -1,7 +1,16 @@
+#!/usr/bin/python3
 import pdfkit
-# import weasyprint
+import weasyprint
 import re
 import os
+import argparse
+
+
+# CLI arguments
+parser = argparse.ArgumentParser()
+parser.add_argument('--backend', '-b', dest = 'backend', help = 'change the download backend; default: pdfkit', default = 'pdfkit', choices = ['pdfkit', 'weasyprint'])
+args = parser.parse_args()
+
 
 for filename in os.listdir("."):
     if not filename.endswith(".txt"): continue
@@ -22,9 +31,11 @@ for filename in os.listdir("."):
         # print(name)
         try:
             # weasyprint seems faster?
-            pdfkit.from_url(link, name)
-            # pdf = weasyprint.HTML(link).write_pdf()
-            # open(name, 'wb').write(pdf)
+            if args.backend == 'pdfkit':
+                pdfkit.from_url(link, name)
+            else:
+                pdf = weasyprint.HTML(link).write_pdf()
+                open(name, 'wb').write(pdf)
         except:
             # Ignore exceptions
             # Probably not a good idea