def index():
    files = []

    for f in os.listdir(current_app.config['FILES_DIR']):

        # ignore dotfiles
        if f.startswith('.'):
            continue

        fp = open(os.path.join(current_app.config['FILES_DIR'], f))
        # List of all modules for this file type
        handlers = mfr.detect(fp, many=True)

        renderer_handlers = [mfr.get_namespace(handler) for handler in handlers if handler.renderers]
        exporter_handlers = [handler for handler in handlers if handler.exporters]

        # Generate a list of modules and their file extensions
        export_options = []
        for handler in exporter_handlers:
            available_file_extensions = mfr.get_file_exporters(handler)
            for export_file_type in available_file_extensions:
                export_options.append((export_file_type, mfr.get_namespace(handler)))

        files.append((f, renderer_handlers, export_options))

    return render_template('main/index.html', files=files)
 def test_render_return_type(self):
     for filename in os.listdir(self.FILES_DIR):
         fp = open(os.path.join(self.FILES_DIR, filename))
         renderer = mfr.detect(fp, many=False)
         if renderer:
             # src = url_for('render.serve_file', filename=filename)
             result = mfr.render(fp)
             assert type(result) == mfr.RenderResult
 def test_render_request(self):
     for filename in os.listdir(self.FILES_DIR):
         fp = open(os.path.join(self.FILES_DIR, filename))
         status_code = self.app.get('/render/{filename}'.format(filename=filename)).status_code
         if mfr.detect(fp, many=False):
             assert status_code == 200
         else:
             assert status_code == 400
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)
    )