Consolidate the page thumbnail and staging folder thumbnail widgets.

Update the staging folder file image API end point to work
in the same way the document page image API works.

mayan/apps/sources/widgets.py

@@ -1,105 +1,32 @@
 from __future__ import unicode_literals
 
-from django.contrib.staticfiles.templatetags.staticfiles import static
-from django.core.urlresolvers import reverse
-from django.utils.html import strip_tags
-from django.utils.http import urlencode
-from django.utils.safestring import mark_safe
-from django.utils.translation import ugettext_lazy as _
-
-from converter.literals import (
-    DEFAULT_PAGE_NUMBER, DEFAULT_ROTATION, DEFAULT_ZOOM_LEVEL
-)
 from documents.settings import setting_preview_size, setting_thumbnail_size
+from documents.widgets import BaseDocumentThumbnailWidget
 
 
-def staging_file_thumbnail(staging_file, **kwargs):
-    return staging_file_html_widget(
-        staging_file, click_view='rest_api:stagingfolderfile-image-view',
-        **kwargs
-    )
-
-
-def staging_file_html_widget(staging_file, click_view=None, page=DEFAULT_PAGE_NUMBER, zoom=DEFAULT_ZOOM_LEVEL, rotation=DEFAULT_ROTATION, gallery_name=None, fancybox_class='fancybox-staging', image_class='lazy-load', title=None, size=setting_thumbnail_size.value, nolazyload=False):
-    result = []
-
-    alt_text = _('Staging file page image')
-
-    query_dict = {
-        'page': page,
-        'zoom': zoom,
-        'rotation': rotation,
-        'size': size,
+class StagingFileThumbnailWidget(BaseDocumentThumbnailWidget):
+    disable_title_link = True
+    gallery_name = 'sources:staging_list'
+    click_view_name = 'rest_api:stagingfolderfile-image-view'
+    click_view_query_dict = {
+        'size': setting_preview_size.value
+    }
+    preview_view_name = 'rest_api:stagingfolderfile-image-view'
+    preview_view_query_dict = {
+        'size': setting_thumbnail_size.value
     }
 
-    if gallery_name:
-        gallery_template = 'rel="%s"' % gallery_name
-    else:
-        gallery_template = ''
+    def get_click_view_kwargs(self, instance):
+        return {
+            'staging_folder_pk': instance.staging_folder.pk,
+            'encoded_filename': instance.encoded_filename
+        }
 
-    query_string = urlencode(query_dict)
+    def get_preview_view_kwargs(self, instance):
+        return {
+            'staging_folder_pk': instance.staging_folder.pk,
+            'encoded_filename': instance.encoded_filename
+        }
 
-    preview_view = '%s?%s' % (
-        reverse(
-            'rest_api:stagingfolderfile-image-view',
-            args=(
-                staging_file.staging_folder.pk, staging_file.encoded_filename,
-            )
-        ), query_string
-    )
-
-    plain_template = []
-    plain_template.append(
-        '<img src="%s" alt="%s" />' % (preview_view, alt_text)
-    )
-
-    result.append(
-        '<div class="tc" id="staging_file-%s-%d">' % (
-            staging_file.filename, page if page else DEFAULT_PAGE_NUMBER
-        )
-    )
-
-    if title:
-        title_template = 'title="%s"' % strip_tags(title)
-    else:
-        title_template = ''
-
-    if click_view:
-        # TODO: fix this hack
-        query_dict['size'] = setting_preview_size.value
-        query_string = urlencode(query_dict)
-        result.append(
-            '<a %s class="%s" href="%s" %s>' % (
-                gallery_template, fancybox_class,
-                '%s?%s' % (
-                    reverse(
-                        click_view,
-                        args=(
-                            staging_file.staging_folder.pk,
-                            staging_file.encoded_filename,
-                        )
-                    ),
-                    query_string
-                ), title_template
-            )
-        )
-
-    if nolazyload:
-        result.append(
-            '<img style="border: 1px solid black;" src="%s" alt="%s" />' % (
-                preview_view, alt_text
-            )
-        )
-    else:
-        result.append(
-            '<img class="thin_border %s" data-src="%s" src="%s" alt="%s" />' % (
-                image_class, preview_view,
-                static('appearance/images/loading.png'), alt_text
-            )
-        )
-
-    if click_view:
-        result.append('</a>')
-    result.append('</div>')
-
-    return mark_safe(''.join(result))
+    def get_title(self, instance):
+        return instance.filename