blob: e88056c552116fab37325a383c618d413e8c2bf0 (
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
40
41
42
43
44
45
46
47
48
49
|
from flask import redirect, request, url_for
from werkzeug.exceptions import BadRequest
from pysite.base_route import RouteView
from pysite.constants import ALL_STAFF_ROLES
from pysite.decorators import csrf, require_roles
from pysite.mixins import DBMixin
REQUIRED_KEYS = ["title", "date_start", "date_end"]
class StaffView(RouteView, DBMixin):
path = "/jams/create"
name = "jams.create"
table_name = "code_jams"
@require_roles(*ALL_STAFF_ROLES)
def get(self):
number = self.get_next_number()
return self.render("staff/jams/create.html", number=number)
@require_roles(*ALL_STAFF_ROLES)
@csrf
def post(self):
data = {}
for key in REQUIRED_KEYS:
arg = request.form.get(key)
if not arg:
return BadRequest()
data[key] = arg
data["state"] = "planning"
data["number"] = self.get_next_number()
self.db.insert(self.table_name, data)
return redirect(url_for("staff.jams.index"))
def get_next_number(self) -> int:
count = self.db.run(self.table.count(), coerce=int)
if count:
max_num = self.db.run(self.table.max("number"))["number"]
return max_num + 1
return 1
|
