77 lines
2.6 KiB
Python
77 lines
2.6 KiB
Python
import logging
|
|
|
|
from django.contrib.auth.models import User
|
|
from django.core.files import File
|
|
|
|
from mayan.celery import app
|
|
|
|
from common.models import SharedUploadedFile
|
|
from converter.exceptions import ConvertError
|
|
|
|
from .models import Document, DocumentType, DocumentVersion
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@app.task(bind=True, max_retries=3, default_retry_delay=1, compression='zlib')
|
|
def task_get_document_image(self, document_id, *args, **kwargs):
|
|
document = Document.objects.get(pk=document_id)
|
|
try:
|
|
return document.get_image(*args, **kwargs)
|
|
except ConvertError as exception:
|
|
raise self.retry(exc=exception)
|
|
|
|
|
|
@app.task(ignore_result=True)
|
|
def task_clear_image_cache():
|
|
# TODO: Error logging / notification
|
|
Document.clear_image_cache()
|
|
|
|
|
|
@app.task(ignore_result=True)
|
|
def task_update_page_count(version_id):
|
|
document_version = DocumentVersion.objects.get(pk=version_id)
|
|
document_version.update_page_count()
|
|
|
|
|
|
@app.task(ignore_result=True)
|
|
def task_new_document(document_type_id, shared_uploaded_file_id, label, description=None, expand=False, language=None, user_id=None):
|
|
shared_uploaded_file = SharedUploadedFile.objects.get(pk=shared_uploaded_file_id)
|
|
document_type = DocumentType.objects.get(pk=document_type_id)
|
|
|
|
if user_id:
|
|
user = User.objects.get(pk=user_id)
|
|
else:
|
|
user = None
|
|
|
|
with File(file=shared_uploaded_file.file) as file_object:
|
|
Document.objects.new_document(document_type=document_type, expand=expand, file_object=file_object, label=label, description=description, language=language, user=user)
|
|
|
|
shared_uploaded_file.delete()
|
|
|
|
# TODO: Report/record how was file uploaded
|
|
# if result['is_compressed'] is None:
|
|
# messages.success(request, _(u'File uploaded successfully.'))
|
|
|
|
# if result['is_compressed'] is True:
|
|
# messages.success(request, _(u'File uncompressed successfully and uploaded as individual files.'))
|
|
|
|
# if result['is_compressed'] is False:
|
|
# messages.warning(request, _(u'File was not a compressed file, uploaded as it was.'))
|
|
|
|
|
|
@app.task(ignore_result=True)
|
|
def task_upload_new_version(document_id, shared_uploaded_file_id, user_id, comment=None, version_update=None):
|
|
shared_file = SharedUploadedFile.objects.get(pk=shared_uploaded_file_id)
|
|
document = Document.objects.get(pk=document_id)
|
|
|
|
if user_id:
|
|
user = User.objects.get(pk=user_id)
|
|
else:
|
|
user = None
|
|
|
|
with File(file=shared_file.file) as file_object:
|
|
document.new_version(comment=comment, file_object=file_object, user=user, version_update=version_update)
|
|
|
|
shared_file.delete()
|