From 6a5de7cfdf5b419acd8258d5dbbd0b164662c140 Mon Sep 17 00:00:00 2001
From: Gareth Coles <gdude2002@gmail.com>
Date: Thu, 5 Apr 2018 14:14:05 +0100
Subject: More work towards the wiki

---
 pysite/views/wiki/edit.py | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

(limited to 'pysite/views/wiki/edit.py')

diff --git a/pysite/views/wiki/edit.py b/pysite/views/wiki/edit.py
index 8f6c22e3..23a281dc 100644
--- a/pysite/views/wiki/edit.py
+++ b/pysite/views/wiki/edit.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-from flask import url_for
+from flask import url_for, request
 from werkzeug.utils import redirect
 
 from pysite.base_route import RouteView
@@ -17,9 +17,19 @@ class EditView(RouteView, DBMixin):
 
     @require_roles(*ALL_STAFF_ROLES)
     def get(self, page):
-        return self.render("wiki/page_edit.html", page=page)
+        rst = ""
+        title = ""
+
+        obj = self.db.get(self.table_name, page)
+
+        if obj:
+            rst = obj["rst"]
+            title = obj["title"]
+
+        return self.render("wiki/page_edit.html", page=page, rst=rst, title=title)
 
     @require_roles(*ALL_STAFF_ROLES)
     @csrf
     def post(self, page):
+        rst = request.form["rst"]
         return redirect(url_for("wiki.page", page=page), code=303)  # Redirect, ensuring a GET
-- 
cgit v1.2.3