[FIX] look for 'Done' in stderr

MichaelKim0407 · MichaelKim0407 · commit 440596e0772c · 2018-08-04T16:17:25.000-07:00
Sometimes wkhtmltopdf will exit with non-zero even if
the generation is successful.
diff --git a/pdfkit/pdfkit.py b/pdfkit/pdfkit.py
@@ -130,6 +130,12 @@ def handle_error(exit_code, stderr):
         if exit_code == 0:
             return
 
+        # Sometimes wkhtmltopdf will exit with non-zero
+        # even if it finishes generation.
+        # If will display 'Done' in the second last line
+        if stderr.splitlines()[-2].strip() == 'Done':
+            return
+
         if 'cannot connect to X server' in stderr:
             raise IOError('%s\n'
                           'You will need to run wkhtmltopdf within a "virtual" X server.\n'
