コード例 #1
0
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,
    )
コード例 #2
0
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)
    )
コード例 #3
0
def test_register_filehandlers():
    core.register_filehandlers([FakeHandler, TestHandler])
    assert FakeHandler in core.get_registry()
    assert TestHandler in core.get_registry()
コード例 #4
0
def test_register_filehandler():
    core.register_filehandler(FakeHandler)
    assert FakeHandler in core.get_registry()
コード例 #5
0
def test_registering_handlers_with_config():
    class FakeConfig:
        HANDLERS = [FakeHandler, TestHandler]
    core.config.from_object(FakeConfig)
    assert FakeHandler in core.get_registry()
コード例 #6
0
def test_get_registry():
    core.register_filehandler(TestHandler)
    assert TestHandler in core.get_registry()