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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
import datetime
import requests
from flask import redirect, request, url_for
from werkzeug.exceptions import BadRequest, NotFound
from pysite.base_route import RouteView
from pysite.constants import EDITOR_ROLES, WIKI_AUDIT_WEBHOOK
from pysite.decorators import csrf, require_roles
from pysite.mixins import DBMixin
class MoveView(RouteView, DBMixin):
path = "/move/<path:page>" # "path" means that it accepts slashes
name = "move"
table_name = "wiki"
revision_table_name = "wiki_revisions"
@require_roles(*EDITOR_ROLES)
def get(self, page):
obj = self.db.get(self.table_name, page)
if obj:
title = obj.get("title", "")
if obj.get("lock_expiry") and obj.get("lock_user") != self.user_data.get("user_id"):
lock_time = datetime.datetime.fromtimestamp(obj["lock_expiry"])
if datetime.datetime.utcnow() < lock_time:
return self.render("wiki/page_in_use.html", page=page)
return self.render("wiki/page_move.html", page=page, title=title)
else:
raise NotFound()
@require_roles(*EDITOR_ROLES)
@csrf
def post(self, page):
location = request.form.get("location")
if not location or not location.strip():
raise BadRequest()
obj = self.db.get(self.table_name, page)
if not obj:
raise NotFound()
title = obj.get("title", "")
other_obj = self.db.get(self.table_name, location)
if other_obj:
return self.render(
"wiki/page_move.html", page=page, title=title,
message=f"There's already a page at {location} - please pick a different location"
)
self.db.delete(self.table_name, page)
# Move all revisions for the old slug to the new slug.
revisions = self.db.filter(self.revision_table_name, lambda revision: revision["slug"] == obj["slug"])
for revision in revisions:
revision["slug"] = location
self.db.insert(self.revision_table_name, revision, conflict="update")
obj["slug"] = location
self.db.insert(self.table_name, obj, conflict="update")
self.audit_log(obj)
return redirect(url_for("wiki.page", page=location), code=303) # Redirect, ensuring a GET
def audit_log(self, obj):
if WIKI_AUDIT_WEBHOOK: # If the audit webhook is not configured there is no point processing it
audit_payload = {
"username": "Wiki Updates",
"embeds": [
{
"title": "Page Move",
"description": f"**{obj['title']}** was moved by "
f"**{self.user_data.get('username')}** to "
f"**{obj['slug']}**",
"color": 4165079,
"timestamp": datetime.datetime.utcnow().isoformat(),
"thumbnail": {
"url": "https://pythondiscord.com/static/logos/logo_discord.png"
}
}
]
}
requests.post(WIKI_AUDIT_WEBHOOK, json=audit_payload)
|
