view foodtruck/views/publish.py @ 629:40e897e2f11e

admin: Make the publish UI handle new kinds of target configurations.
author Ludovic Chabant <ludovic@chabant.com>
date Mon, 08 Feb 2016 23:29:21 -0800
parents e2e955a3bb25
children 3885421c29a3
line wrap: on
line source

import copy
import logging
from flask import request, g, url_for, render_template, Response
from flask.ext.login import login_required
from ..pubutil import PublishLogReader
from ..views import with_menu_context
from ..web import app


logger = logging.getLogger(__name__)


@app.route('/publish', methods=['GET', 'POST'])
@login_required
def publish():
    if request.method == 'POST':
        target = request.form.get('target')
        if not target:
            raise Exception("No target specified.")

        g.site.publish(target)

    site = g.site
    pub_cfg = copy.deepcopy(site.piecrust_app.config.get('publish', {}))
    if not pub_cfg:
        data = {'error': "There are not publish targets defined in your "
                         "configuration file."}
        return render_template('error.html', **data)

    data = {}
    data['url_run'] = url_for('publish')
    data['site_title'] = site.piecrust_app.config.get('site/title', site.name)
    data['targets'] = []
    for tn in sorted(pub_cfg.keys()):
        tc = pub_cfg[tn]
        desc = None
        if isinstance(tc, dict):
            desc = tc.get('description')
        data['targets'].append({
            'name': tn,
            'description': desc
            })

    with_menu_context(data)

    return render_template('publish.html', **data)


@app.route('/publish-log')
@login_required
def stream_publish_log():
    pid_path = g.site.publish_pid_file
    log_path = g.site.publish_log_file
    rdr = PublishLogReader(pid_path, log_path)

    response = Response(rdr.run(), mimetype='text/event-stream')
    response.headers['Cache-Control'] = 'no-cache'
    return response