def export(export_file_type, filename, exporter_name=None):
""" Convert a file to another type and download that file.
:param export_file_type: the type to export a file as
:param filename: file to be exported
:param exporter_name: optional name of the specific exporter module
"""
try:
fp = open(os.path.join(current_app.config['FILES_DIR'], filename))
except IOError as err:
flash(err, 'error')
abort(404)
# If handler name is not specified, choose the first that will work
if exporter_name is None:
exporter = mfr.detect(fp, many=False)
exported_content = mfr.export(fp, exporter, exporter=export_file_type)
else:
handlers = get_registry()
for handler in handlers:
if mfr.get_namespace(handler) == exporter_name:
exported_content = mfr.export(fp, handler=handler, exporter=export_file_type)
if not exported_content:
raise NameError("A matching exporter was not found")
short_name, _ = os.path.splitext(filename)
export_name = short_name + '.' + export_file_type
return send_file(
StringIO(exported_content),
as_attachment=True,
attachment_filename=export_name,
)
def render(filename, renderer_name=None):
"""Make a file viewable in HTML
:param filename: file to be rendered
:param renderer_name: optional name of the specific renderer module
:return: html representation of the file
"""
try:
fp = open(os.path.join(current_app.config['FILES_DIR'], filename))
except IOError as err:
flash(err, 'error')
abort(404)
renderer = None
if renderer_name is None:
renderer = mfr.detect(fp, many=False) # return the first valid filehandler
else:
handlers = get_registry()
for available_handler in handlers:
if mfr.get_namespace(available_handler) == renderer_name:
renderer = available_handler()
break
if renderer is None:
raise IOError('Could not load a matching renderer')
src = url_for('render.serve_file', filename=filename)
if renderer is None:
return ("A matching renderer cannot be found", 400)
return render_template(
"main/view_file.html",
render_result=mfr.render(fp, handler=renderer, src=src)
)
def test_register_filehandlers():
core.register_filehandlers([FakeHandler, TestHandler])
assert FakeHandler in core.get_registry()
assert TestHandler in core.get_registry()
def test_register_filehandler():
core.register_filehandler(FakeHandler)
assert FakeHandler in core.get_registry()
def test_registering_handlers_with_config():
class FakeConfig:
HANDLERS = [FakeHandler, TestHandler]
core.config.from_object(FakeConfig)
assert FakeHandler in core.get_registry()
def test_get_registry():
core.register_filehandler(TestHandler)
assert TestHandler in core.get_registry()