blob: 36fdcd3930f37ee7a77c170696de972ef199c865 (
plain) (
blame)
| 1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
 | # coding=utf-8
from flask import redirect, url_for
from werkzeug.exceptions import NotFound
from pysite.base_route import RouteView
from pysite.constants import DEBUG_MODE, EDITOR_ROLES
from pysite.mixins import DBMixin
class PageView(RouteView, DBMixin):
    path = "/wiki/<path:page>"  # "path" means that it accepts slashes
    name = "page"
    table_name = "wiki"
    table_primary_key = "slug"
    def get(self, page):
        obj = self.db.get(self.table_name, page)
        if obj is None:
            if self.is_staff():
                return redirect(url_for("wiki.edit", page=page))
            raise NotFound()
        return self.render("wiki/page_view.html", page=page, data=obj, can_edit=self.is_staff())
    def is_staff(self):
        if DEBUG_MODE:
            return True
        if not self.logged_in:
            return False
        roles = self.user_data.get("roles", [])
        for role in roles:
            if role in EDITOR_ROLES:
                return True
        return False
 |