from __future__ import unicode_literals

from django.utils.encoding import force_text
from django.utils.html import conditional_escape

from .models import DocumentPageOCRContent


def get_document_ocr_content(document):
    for page in document.pages.all():
        try:
            page_content = page.ocr_content.content
        except DocumentPageOCRContent.DoesNotExist:
            pass
        else:
            yield conditional_escape(force_text(page_content))