Ajout de preview-generator

tracim/development.ini.base

@@ -35,6 +35,7 @@ full_stack = true
 # You can set french as default language by uncommenting next line
 # lang = fr
 cache_dir = %(here)s/data
+preview_cache_dir = '/tmp/tracim/cache/previews/'
 beaker.session.key = tracim
 beaker.session.secret = 3283411b-1904-4554-b0e1-883863b53080
 
@@ -96,7 +97,7 @@ beaker.session.validate_key = 3283411b-1904-4554-b0e1-883863b53080
 # sqlalchemy.url=mysql://username:password@hostname:port/databasename
 # sqlalchemy.url = postgresql://tracimuser:tracimpassword@127.0.0.1:5432/tracimdb?client_encoding=utf8
 # sqlalchemy.url = mysql+pymysql://tracimuser:tracimpassword@127.0.0.1/tracimdb
-sqlalchemy.url = <replace_database_uri_here>
+sqlalchemy.url = sqlite:///tracimdb.sqlite
 
 #echo shouldn't be used together with the logging module.
 sqlalchemy.echo = false

tracim/tracim/config/app_cfg.py

@@ -194,6 +194,8 @@ class CFG(object):
 
     def __init__(self):
 
+        self.PREVIEW_CACHE = str(tg.config.get('preview_cache_dir'))
+
         self.DATA_UPDATE_ALLOWED_DURATION = int(tg.config.get('content.update.allowed.duration', 0))
 
         self.WEBSITE_TITLE = tg.config.get('website.title', 'TRACIM')

tracim/tracim/controllers/content.py

@@ -1,4 +1,6 @@
 # -*- coding: utf-8 -*-
+from tracim.config.app_cfg import CFG
+
 __author__ = 'damien'
 
 import sys
@@ -45,6 +47,8 @@ from tracim.model.data import UserRoleInWorkspace
 from tracim.model.data import Workspace
 from tracim.lib.integrity import render_invalid_integrity_chosen_path
 
+from preview_generator.manager import PreviewManager
+
 
 class UserWorkspaceFolderThreadCommentRestController(TIMRestController):
 
@@ -199,19 +203,56 @@ class UserWorkspaceFolderFileRestController(TIMWorkspaceContentRestController):
             show_archived=True,
             show_deleted=True,
         )
+        cfg = CFG.get_instance()
+        cache_path = cfg.PREVIEW_CACHE
+        preview_manager = PreviewManager(cache_path, create_folder=True)
+        if revision_id:
+            file_path = content_api.get_one_revision_filepath(revision_id)
+        else:
+            file = content_api.get_one(file_id, self._item_type)
+            file_path = content_api.get_one_revision_filepath(file.revision_id)
+        nb_page = preview_manager.get_nb_page(
+            file_path=file_path,
+        )
         if revision_id:
-            file = content_api.get_one_from_revision(file_id,  self._item_type, workspace, revision_id)
+            file = content_api.get_one_from_revision(
+                file_id,
+                self._item_type,
+                workspace,
+                revision_id
+            )
         else:
             file = content_api.get_one(file_id, self._item_type, workspace)
 
         fake_api_breadcrumb = self.get_breadcrumb(file_id)
-        fake_api_content = DictLikeClass(breadcrumb=fake_api_breadcrumb, current_user=current_user_content)
+        fake_api_content = DictLikeClass(
+            breadcrumb=fake_api_breadcrumb,
+            current_user=current_user_content
+        )
         fake_api = Context(CTX.FOLDER,
                            current_user=user).toDict(fake_api_content)
 
         dictified_file = Context(self._get_one_context,
                                  current_user=user).toDict(file, 'file')
-        return DictLikeClass(result = dictified_file, fake_api=fake_api)
+
+        url = []
+        if revision_id:
+            for i in range(int(nb_page)):
+                url.append('/previews/{}/pages/{}?revision_id={}'.format(
+                    file_id,
+                    i,
+                    revision_id)
+                )
+        else:
+            for i in range(int(nb_page)):
+                url.append('/previews/{}/pages/{}'.format(file_id, i))
+
+        return DictLikeClass(
+            result=dictified_file,
+            fake_api=fake_api,
+            nb_page=nb_page,
+            url=url,
+        )
 
     @tg.require(current_user_is_reader())
     @tg.expose()

tracim/tracim/controllers/page.py

@@ -0,0 +1,140 @@
+import tg
+from preview_generator.manager import PreviewManager
+from tg import expose, tmpl_context
+from tracim.config.app_cfg import CFG
+from tracim.controllers import TIMRestController
+from tracim.lib.content import ContentApi
+from tracim.model.data import ContentType
+
+__all__ = ['PagesController']
+
+class PagesController(TIMRestController):
+
+    @expose()
+    def _default(self):
+        return '<h2> Error Loading Page</h2>'
+
+    @expose()
+    def get_all(self, *args, **kwargs):
+        file_id = int(tg.request.controller_state.routing_args.get('file_id'))
+        return 'all the pages of document {}'.format(file_id)
+
+    @expose(content_type='image/jpeg')
+    def get_one(self, page_id: int, revision_id: int=None, *args, **kwargs):
+        file_id = int(tg.request.controller_state.routing_args.get('file_id'))
+
+        # For now it's done through database content
+        # but soon it'll be with disk access
+
+        user = tmpl_context.current_user
+        content_api = ContentApi(
+            user,
+            show_archived=True,
+            show_deleted=True,
+        )
+        if revision_id:
+            file_path = content_api.get_one_revision_filepath(revision_id)
+        else:
+            file = content_api.get_one(file_id, self._item_type)
+            file_path = content_api.get_one_revision_filepath(file.revision_id)
+
+        cfg = CFG.get_instance()
+        cache_path = cfg.PREVIEW_CACHE
+
+        preview_manager = PreviewManager(cache_path, create_folder=True)
+        path = preview_manager.get_jpeg_preview(
+            file_path=file_path,
+            page=page_id,
+            height=500,
+            width=500
+        )
+
+        with open(path, 'rb') as large:
+            return large.read()
+
+    @expose(content_type='image/jpeg')
+    def high_quality(self, page_id: int, *args, **kwargs):
+        file_id = int(tg.request.controller_state.routing_args.get('file_id'))
+
+        # For now it's done through database content
+        # but soon it'll be with disk access
+
+        user = tmpl_context.current_user
+        content_api = ContentApi(
+            user,
+            show_archived=True,
+            show_deleted=True,
+        )
+        file_name = content_api.get_one(file_id, self._item_type).file_name
+        cache_path = '/home/alexis/Pictures/cache/'
+
+        preview_manager = PreviewManager(cache_path, create_folder=True)
+        path = preview_manager.get_jpeg_preview(
+            file_path='/home/alexis/Pictures/cache/{}'.format(file_name),
+            page=page_id,
+            height=5000,
+            width=5000
+        )
+
+        with open(path, 'rb') as large:
+            return large.read()
+
+    @expose(content_type='application/pdf')
+    def download_pdf_full(self, *args, **kwargs):
+        file_id = int(tg.request.controller_state.routing_args.get('file_id'))
+
+        # For now it's done through database content
+        # but soon it'll be with disk access
+
+        user = tmpl_context.current_user
+        content_api = ContentApi(
+            user,
+            show_archived=True,
+            show_deleted=True,
+        )
+        file_name = content_api.get_one(file_id, self._item_type).file_name
+        cache_path = '/home/alexis/Pictures/cache/'
+
+        preview_manager = PreviewManager(cache_path, create_folder=True)
+        path = preview_manager.get_pdf_preview(
+            file_path='/home/alexis/Pictures/cache/{}'.format(file_name),
+        )
+
+        tg.response.headers['Content-Disposition'] = \
+            str('attachment; filename="{}"'.format(file_name))
+        with open(path, 'rb') as pdf:
+            return pdf.read()
+
+    @expose(content_type='application/pdf')
+    def download_pdf_one(self, page_id: int, *args, **kwargs):
+        file_id = int(tg.request.controller_state.routing_args.get('file_id'))
+        page_id = int(page_id)
+        # page_id = int(tg.request.controller_state.routing_args.get('page_id'))
+
+        # For now it's done through database content
+        # but soon it'll be with disk access
+
+        user = tmpl_context.current_user
+        content_api = ContentApi(
+            user,
+            show_archived=True,
+            show_deleted=True,
+        )
+        file_name = content_api.get_one(file_id, self._item_type).file_name
+
+        cache_path = '/home/alexis/Pictures/cache/'
+
+        preview_manager = PreviewManager(cache_path, create_folder=True)
+        path = preview_manager.get_pdf_preview(
+            file_path='/home/alexis/Pictures/cache/{}'.format(file_name),
+            page=page_id,
+        )
+
+        tg.response.headers['Content-Disposition'] = \
+            str('attachment; filename="{}"'.format(file_name))
+        with open(path, 'rb') as pdf:
+            return pdf.read()
+
+    @property
+    def _item_type(self):
+        return ContentType.File

tracim/tracim/controllers/previews.py

@@ -0,0 +1,24 @@
+from tracim.controllers import TIMRestController
+from tg import expose, tmpl_context
+from tracim.controllers.page import PagesController
+
+__all__ = ['PreviewsController']
+
+class PreviewsController(TIMRestController):
+
+    pages = PagesController()
+
+    @expose()
+    def _default(self, *args, **kwargs):
+        return '<h2> Error Loading Page</h2>'
+
+    @expose()
+    def get_all(self, *args, **kwargs):
+        print('getall _ document')
+        return 'all the files'
+
+    @expose()
+    def get_one(self, file_id, *args, **kwargs):
+        print('getone _ document')
+        tmpl_context.file = file_id
+        return 'File n°{}'.format(file_id)

tracim/tracim/controllers/root.py

@@ -13,6 +13,7 @@ from tg import url
 from tg.i18n import ugettext as _
 from tracim.controllers.api import APIController
 from tracim.controllers.content import ContentController
+from tracim.controllers.previews import PreviewsController
 
 from tracim.lib import CST
 from tracim.lib.base import logger
@@ -62,6 +63,7 @@ class RootController(StandardController):
     # Rest controllers
     workspaces = UserWorkspaceRestController()
     user = UserRestController()
+    previews = PreviewsController()
 
     content = ContentController()
 

tracim/tracim/lib/content.py

@@ -475,6 +475,7 @@ class ContentApi(object):
         # python 3.6 PEP 526 -- Syntax for Variable Annotations
         # https://www.python.org/dev/peps/pep-0526/
         # dpt_file_path: str = dpt.get(dpt_stored_file)._file_path
+        dpt_stored_file = dpt.get(revision.depot_file_uid)
         dpt_file_path = dpt.get(dpt_stored_file)._file_path
 
         return dpt_file_path

tracim/tracim/templates/file/getone.mak

@@ -100,6 +100,71 @@
     % endif
 
     <div class="content__detail file">
+        <div style="width: 15%;height: 100%;border:5px solid #606060;">
+            <a id="preview_link"><img id='preview' alt="Preview" style="width:100%;height:100%"></a>
+            <table style="width:100%;height:20%;">
+                <tr>
+                    <td rowspan="2">
+                        <button type="button" id="prev" onclick="previous_page()" style="width:100%;height:100%;"> - </button>
+                    </td>
+                    <td>
+                        <a id="dl_one_pdf" style=""> Download the page </a>
+                    </td>
+                    <td rowspan="2">
+                        <button type="button" id="next" onclick="next_page()" style="width:100%;height:100%;"> + </button>
+                    </td>
+                </tr>
+                <tr>
+                    <td >
+                        <a id="dl_full_pdf" style=""> Download all </a>
+                    </td>
+                </tr>
+            </table>
+
+            <script type="text/javascript">
+                var nb_page = parseInt(${nb_page});
+                console.log(nb_page);
+                var page = 0;
+                var urls = [];
+                % for one_url in url:
+                urls.push('${one_url}');
+                % endfor
+                console.log(urls);
+                document.getElementById('preview').src = urls[page];
+                refresh_button();
+
+                function next_page(){
+                    page = page+1;
+                    console.log('page next');
+                    console.log(urls[page]);
+                    document.getElementById('preview').src = urls[page];
+                    refresh_button();
+                }
+
+                function previous_page(){
+                    page = page-1;
+                    console.log('page previous');
+                    console.log(urls[page]);
+                    document.getElementById('preview').src = urls[page];
+                    refresh_button();
+                }
+
+                function refresh_button(){
+                    console.log(page);
+                    document.getElementById('prev').disabled = false;
+                    document.getElementById('next').disabled = false;
+                    document.getElementById('dl_one_pdf').href = "/previews/${result.file.id}/pages/" + page + "/download_pdf_one";
+                    document.getElementById('dl_full_pdf').href = "/previews/${result.file.id}/pages/" + page + "/download_pdf_full";
+                    document.getElementById('preview_link').href = "/previews/${result.file.id}/pages/" + page + "/high_quality";
+                    if(page >= nb_page-1){
+                        document.getElementById('next').disabled = true;
+                    }
+                    if(page <= 0){
+                        document.getElementById('prev').disabled = true;
+                    }
+                }
+            </script>
+        </div>
         <% download_url = tg.url('/workspaces/{}/folders/{}/files/{}/download?revision_id={}'.format(result.file.workspace.id, result.file.parent.id,result.file.id,result.file.selected_revision)) %>
         <div class="t-half-spacer-above download-file-button">
             <a style="" class="btn btn-default" tittle="${_('Download the file')}"