first commit

modules/account_invoice_watermark/tests/tools.py

@@ -0,0 +1,20 @@
+# This file is part of Tryton.  The COPYRIGHT file at the top level of
+# this repository contains the full copyright notices and license terms.
+import os
+import subprocess
+from tempfile import TemporaryDirectory
+
+__all__ = ['pdf_contains']
+
+
+def pdf_contains(pdf, text):
+    with TemporaryDirectory() as dirname:
+        input_file = os.path.join(dirname, 'input.pdf')
+        output_file = os.path.join(dirname, 'output.text')
+        with open(input_file, 'wb') as fp:
+            fp.write(pdf)
+        subprocess.check_call([
+                'mutool', 'convert', '-F', 'text',
+                '-o', output_file, input_file])
+        with open(output_file, 'r') as fp:
+            return text in fp.read()