Merge pull request #52 from pitkley/fix/detect-orientation-errors

Ignore error if orientation detection fails
This commit is contained in:
Daniel Quinn 2016-02-19 09:13:14 +00:00
commit d9602312b1

View File

@ -17,6 +17,7 @@ from PIL import Image
from django.conf import settings from django.conf import settings
from django.utils import timezone from django.utils import timezone
from django.template.defaultfilters import slugify from django.template.defaultfilters import slugify
from pyocr.tesseract import TesseractError
from logger.models import Log from logger.models import Log
from paperless.db import GnuPG from paperless.db import GnuPG
@ -29,8 +30,11 @@ def image_to_string(args):
self, png, lang = args self, png, lang = args
with Image.open(os.path.join(self.SCRATCH, png)) as f: with Image.open(os.path.join(self.SCRATCH, png)) as f:
if self.OCR.can_detect_orientation(): if self.OCR.can_detect_orientation():
orientation = self.OCR.detect_orientation(f, lang=lang) try:
f = f.rotate(orientation["angle"], expand=1) orientation = self.OCR.detect_orientation(f, lang=lang)
f = f.rotate(orientation["angle"], expand=1)
except TesseractError:
pass
return self.OCR.image_to_string(f, lang=lang) return self.OCR.image_to_string(f, lang=lang)