view piecrust/admin/views/publish.py @ 778:5e91bc0e3b4d

internal: Move admin panel code into the piecrust package.
author Ludovic Chabant <ludovic@chabant.com>
date Sat, 16 Jul 2016 15:02:24 +0200
parents foodtruck/views/publish.py@3885421c29a3
children 82509bce94ca
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 ..blueprint import foodtruck_bp
from ..pubutil import PublishLogReader
from ..views import with_menu_context


logger = logging.getLogger(__name__)


@foodtruck_bp.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)


@foodtruck_bp.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