Add is_pdf from scratch

papis/filetype.py

@@ -0,0 +1,15 @@
+import sys
+import os
+import re
+
+
+def is_pdf(filepath):
+
+    if not os.path.exists(filepath):
+        return False
+
+    with open(filepath, 'rb') as fd:
+        magic = fd.read(8)
+    return re.match(r'%PDF-.\..', magic.decode()) is not None
+
+

tests/filetype.py

@@ -0,0 +1,13 @@
+from papis.filetype import is_pdf
+import tempfile
+import os
+
+def test_is_pdf():
+    tmp = tempfile.mktemp()
+
+    assert not os.path.exists(tmp)
+    assert not is_pdf(tmp)
+
+    with open(tmp, 'w+') as fd:
+        fd.write('%PDF-1.8\n')
+    assert is_pdf(tmp)