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)
+