123 lines
3.9 KiB
Python
123 lines
3.9 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import hashlib
|
|
import logging
|
|
import os
|
|
|
|
from django.utils.encoding import smart_str
|
|
|
|
from common.settings import TEMPORARY_DIRECTORY
|
|
from common.utils import fs_cleanup
|
|
|
|
from .exceptions import OfficeConversionError, UnknownFileFormat
|
|
from .literals import (
|
|
DEFAULT_PAGE_NUMBER, DEFAULT_ZOOM_LEVEL, DEFAULT_ROTATION,
|
|
DEFAULT_FILE_FORMAT, TRANSFORMATION_CHOICES, TRANSFORMATION_RESIZE,
|
|
TRANSFORMATION_ROTATE, TRANSFORMATION_ZOOM, DIMENSION_SEPARATOR
|
|
)
|
|
|
|
HASH_FUNCTION = lambda x: hashlib.sha256(x).hexdigest()
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def cache_cleanup(input_filepath, *args, **kwargs):
|
|
try:
|
|
os.remove(create_image_cache_filename(input_filepath, *args, **kwargs))
|
|
except OSError:
|
|
pass
|
|
|
|
|
|
def create_image_cache_filename(input_filepath, *args, **kwargs):
|
|
if input_filepath:
|
|
hash_value = HASH_FUNCTION(''.join([HASH_FUNCTION(smart_str(input_filepath)), unicode(args), unicode(kwargs)]))
|
|
return os.path.join(TEMPORARY_DIRECTORY, hash_value)
|
|
else:
|
|
return None
|
|
|
|
|
|
def convert(input_filepath, output_filepath=None, cleanup_files=False, mimetype=None, *args, **kwargs):
|
|
size = kwargs.get('size')
|
|
file_format = kwargs.get('file_format', DEFAULT_FILE_FORMAT)
|
|
zoom = kwargs.get('zoom', DEFAULT_ZOOM_LEVEL)
|
|
rotation = kwargs.get('rotation', DEFAULT_ROTATION)
|
|
page = kwargs.get('page', DEFAULT_PAGE_NUMBER)
|
|
transformations = kwargs.get('transformations', [])
|
|
|
|
if transformations is None:
|
|
transformations = []
|
|
|
|
if output_filepath is None:
|
|
output_filepath = create_image_cache_filename(input_filepath, *args, **kwargs)
|
|
|
|
if os.path.exists(output_filepath):
|
|
return output_filepath
|
|
|
|
if office_converter:
|
|
try:
|
|
office_converter.convert(input_filepath, mimetype=mimetype)
|
|
if office_converter.exists:
|
|
input_filepath = office_converter.output_filepath
|
|
mimetype = 'application/pdf'
|
|
else:
|
|
# Recycle the already detected mimetype
|
|
mimetype = office_converter.mimetype
|
|
|
|
except OfficeConversionError:
|
|
raise UnknownFileFormat('office converter exception')
|
|
|
|
if size:
|
|
transformations.append(
|
|
{
|
|
'transformation': TRANSFORMATION_RESIZE,
|
|
'arguments': dict(zip(['width', 'height'], size.split(DIMENSION_SEPARATOR)))
|
|
}
|
|
)
|
|
|
|
if zoom != 100:
|
|
transformations.append(
|
|
{
|
|
'transformation': TRANSFORMATION_ZOOM,
|
|
'arguments': {'percent': zoom}
|
|
}
|
|
)
|
|
|
|
if rotation != 0 and rotation != 360:
|
|
transformations.append(
|
|
{
|
|
'transformation': TRANSFORMATION_ROTATE,
|
|
'arguments': {'degrees': rotation}
|
|
}
|
|
)
|
|
|
|
try:
|
|
backend.convert_file(input_filepath=input_filepath, output_filepath=output_filepath, transformations=transformations, page=page, file_format=file_format, mimetype=mimetype)
|
|
finally:
|
|
if cleanup_files:
|
|
fs_cleanup(input_filepath)
|
|
|
|
return output_filepath
|
|
|
|
|
|
def get_page_count(input_filepath):
|
|
logger.debug('office_converter: %s', office_converter)
|
|
if office_converter:
|
|
try:
|
|
office_converter.convert(input_filepath)
|
|
logger.debug('office_converter.exists: %s', office_converter.exists)
|
|
if office_converter.exists:
|
|
input_filepath = office_converter.output_filepath
|
|
|
|
except OfficeConversionError:
|
|
raise UnknownFileFormat('office converter exception')
|
|
|
|
return backend.get_page_count(input_filepath)
|
|
|
|
|
|
def get_available_transformations_choices():
|
|
result = []
|
|
for transformation in backend.get_available_transformations():
|
|
result.append((transformation, TRANSFORMATION_CHOICES[transformation]['label']))
|
|
|
|
return result
|