gcrkrause
/
mglib
mirror of https://github.com/papermerge/mglib
1
0
Fork 0
Code Issues Releases Wiki Activity

Replace extension checking with python-magic

pull/6/head
francesco.carzaniga 2020-11-27 19:22:43 +01:00
parent 9fbaaf7dfd
commit 0b80c1f446
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mglib/pdfinfo.py
View File

@ -2,6 +2,7 @@ import os
import re
import subprocess
import logging
from magic import from_file
from .conf import settings
from .exceptions import FileTypeNotSupported
@ -63,18 +64,18 @@ def get_pagecount(filepath):
if os.path.isdir(filepath):
raise ValueError("Filepath %s is a directory!" % filepath)
base, ext = os.path.splitext(filepath)
mime_type = from_file(filepath, mime=True)
# pure images (png, jpeg) have only one page :)
if ext and ext.lower() in ('.jpeg', '.png', '.jpg'):
if mime_type in ['image/png', 'image/jpeg', 'image/jpg']:
# whatever png/jpg image is there - it is
# considered by default one page document.
return 1
if ext and ext.lower() in ('.tiff', ):
if mime_type == 'image/tiff':
return get_tiff_pagecount(filepath)
if ext and ext.lower() not in ('.pdf', '.tiff'):
if mime_type != 'application/pdf' :
raise FileTypeNotSupported(
"Only jpeg, png, pdf and tiff are handled by this"
" method"