Page MenuHomec4science
Files F111520668

zip.py
No OneTemporary
Actions

File Metadata

Created
Fri, May 2, 22:09

zip.py
View Options

## This file is part of Invenio.
## Copyright (C) 2013, 2014 CERN.
##
## Invenio is free software; you can redistribute it and/or
## modify it under the terms of the GNU General Public License as
## published by the Free Software Foundation; either version 2 of the
## License, or (at your option) any later version.
##
## Invenio is distributed in the hope that it will be useful, but
## WITHOUT ANY WARRANTY; without even the implied warranty of
## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
## General Public License for more details.
##
## You should have received a copy of the GNU General Public License
## along with Invenio; if not, write to the Free Software Foundation, Inc.,
## 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA.
"""Simple ZIP archive previewer.
Previewer needs to be enabled by setting following config variable.
.. code-block:: python
CFG_PREVIEW_PREFERENCE = {'.zip': ['zip']}
"""
import os
import zipfile
from flask import render_template, request
def make_tree(archive_name):
"""Create tree structure from ZIP archive."""
zf = zipfile.ZipFile(archive_name)
tree = {'type': 'folder', 'id': -1, 'children': {}}
for i, info in enumerate(zf.infolist()):
comps = info.filename.split(os.sep)
node = tree
for c in comps:
if c not in node['children']:
if c == '':
node['type'] = 'folder'
continue
node['children'][c] = {
'name': c, 'type': 'item', 'id': 'item%s' % i,
'children': {}}
node = node['children'][c]
node['size'] = info.file_size
return tree
def children_to_list(node):
"""Organize children structure."""
if node['type'] == 'item':
del node['children']
else:
node['children'] = list(node['children'].values())
node['children'].sort(key=lambda x: x['name'])
node['children'] = map(children_to_list, node['children'])
return node
def can_preview(f):
"""Return True if filetype can be previewed."""
return f.superformat == '.zip'
def preview(f):
"""Return appropiate template and pass the file and an embed flag."""
tree = children_to_list(make_tree(f.get_full_path()))['children']
return render_template("previewer/zip.html", f=f, tree=tree,
embed=request.args.get('embed', type=bool))

Event Timeline

c4science · Help