Mercurial > piecrust2
diff piecrust/admin/views/edit.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/edit.py@3885421c29a3 |
children | 22c6f6a3d0a0 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/piecrust/admin/views/edit.py Sat Jul 16 15:02:24 2016 +0200 @@ -0,0 +1,76 @@ +import os.path +import logging +from flask import ( + g, request, abort, render_template, url_for, flash) +from flask.ext.login import login_required +from piecrust.rendering import ( + PageRenderingContext, render_page) +from piecrust.serving.util import get_requested_page +from ..blueprint import foodtruck_bp +from ..views import with_menu_context + + +logger = logging.getLogger(__name__) + + +@foodtruck_bp.route('/edit/', defaults={'slug': ''}, methods=['GET', 'POST']) +@foodtruck_bp.route('/edit/<path:slug>', methods=['GET', 'POST']) +@login_required +def edit_page(slug): + site = g.site + site_app = site.piecrust_app + rp = get_requested_page(site_app, + '/site/%s/%s' % (g.sites.current_site, slug)) + page = rp.qualified_page + if page is None: + abort(404) + + if request.method == 'POST': + page_text = request.form['page_text'] + if request.form['is_dos_nl'] == '0': + page_text = page_text.replace('\r\n', '\n') + + if 'do_preview' in request.form or 'do_save' in request.form or \ + 'do_save_and_commit' in request.form: + logger.debug("Writing page: %s" % page.path) + with open(page.path, 'w', encoding='utf8', newline='') as fp: + fp.write(page_text) + flash("%s was saved." % os.path.relpath( + page.path, site_app.root_dir)) + + if 'do_save_and_commit' in request.form: + message = request.form.get('commit_msg') + if not message: + message = "Edit %s" % os.path.relpath( + page.path, site_app.root_dir) + if site.scm: + site.scm.commit([page.path], message) + + if 'do_preview' in request.form: + return _preview_page(page) + + if 'do_save' in request.form or 'do_save_and_commit' in request.form: + return _edit_page_form(page) + + abort(400) + + return _edit_page_form(page) + + +def _preview_page(page): + render_ctx = PageRenderingContext(page, force_render=True) + rp = render_page(render_ctx) + return rp.content + + +def _edit_page_form(page): + data = {} + data['is_new_page'] = False + data['url_cancel'] = url_for('.list_source', source_name=page.source.name) + with open(page.path, 'r', encoding='utf8', newline='') as fp: + data['page_text'] = fp.read() + data['is_dos_nl'] = "1" if '\r\n' in data['page_text'] else "0" + + with_menu_context(data) + return render_template('edit_page.html', **data) +