2 from flask import Flask
3 from flask import render_template, redirect
4 from flask import request
7 from flaskext.markdown import Markdown
8 from markdown.extensions import Extension
11 db = getattr(g, '_database', None)
13 db = g._database = postgresql.open(app.config.get("DATABASE"), user=app.config.get("USER"), password=app.config.get("PASSWORD"))
14 #db.row_factory = sqlite3.Row
18 app.register_blueprint(filters.blueprint)
20 class EscapeHtml(Extension):
21 def extendMarkdown(self, md, md_globals):
22 del md.preprocessors['html_block']
23 del md.inlinePatterns['html']
25 md = Markdown(app, extensions=[EscapeHtml()])
28 app.config.from_pyfile('config.py')
30 prefix=app.config.get("GROUP_PREFIX")
32 times=app.config.get("DURATION")
34 debuguser=app.config.get("DEBUGUSER")
42 if debuguser is not None:
43 parts =debuguser[request.host].split("/", 1)
47 if "USER_ROLES" in env:
48 parts = env.get("USER_ROLES").split("/", 1)
52 if "USER" in env and "ROLES" in env:
53 user = env.get("USER")
54 roles = env.get("ROLES")
58 return "Server misconfigured", 500
59 roles = roles.split(" ")
61 if user == "<invalid>":
62 return "Access denied", 403;
66 rv = db.prepare("SELECT id FROM voter WHERE email=$1")(user)
68 db.prepare("INSERT INTO voter(\"email\") VALUES($1)")(user)
69 rv = db.prepare("SELECT id FROM voter WHERE email=$1")(user)
70 g.voter = rv[0].get("id");
78 if a[0] not in g.roles:
81 g.roles[a[0]] = [group for group in prefix[request.host]]
83 g.roles[a[0]].append(val)
86 def get_allowed_cats(action):
87 return g.roles.get(action, []);
89 def may(action, motion):
90 return motion in get_allowed_cats(action)
92 @app.teardown_appcontext
93 def close_connection(exception):
94 db = getattr(g, '_database', None)
99 with app.app_context():
102 ver = db.prepare("SELECT version FROM schema_version")()[0][0];
103 print("Database Schema version: ", ver)
104 except postgresql.exceptions.UndefinedTableError:
108 with app.open_resource('sql/schema.sql', mode='r') as f:
113 with app.open_resource('sql/from_1.sql', mode='r') as f:
116 for g in [group for group in prefix[app.config.get("DEFAULT_HOST")]]:
117 ct[g] = {"dt": "", "c": 0}
119 p = db.prepare("UPDATE \"motion\" SET \"identifier\"=$1 WHERE \"id\"=$2")
120 for row in db.prepare("SELECT id, \"type\", \"posed\" FROM \"motion\" ORDER BY \"id\" ASC"):
121 dt=row[2].strftime("%Y%m%d")
122 if ct[row[1]]["dt"] != dt:
123 ct[row[1]]["dt"] = dt
125 ct[row[1]]["c"] = ct[row[1]]["c"] + 1
126 name=prefix[app.config.get("DEFAULT_HOST")][row[1]]+"."+dt+"."+("%03d" % ct[row[1]]["c"])
128 db.prepare("ALTER TABLE \"motion\" ALTER COLUMN \"identifier\" SET NOT NULL")()
129 db.prepare("UPDATE \"schema_version\" SET \"version\"=2")()
130 db.prepare("CREATE UNIQUE INDEX motion_ident ON motion (identifier)")()
133 with app.open_resource('sql/from_2.sql', mode='r') as f:
135 db.prepare("UPDATE \"motion\" SET \"host\"=$1")(app.config.get("DEFAULT_HOST"))
136 db.prepare("ALTER TABLE \"motion\" ALTER COLUMN \"host\" SET NOT NULL")()
137 db.prepare("UPDATE \"schema_version\" SET \"version\"=3")()
144 start=int(request.args.get("start", "-1"));
145 q = "SELECT motion.*, votes.*, poser.email AS poser, canceler.email AS canceler, (motion.deadline > CURRENT_TIMESTAMP AND canceled is NULL) AS running FROM motion LEFT JOIN (SELECT motion_id, "\
146 + "COUNT(CASE WHEN result='yes' THEN 'yes' ELSE NULL END) as yes, "\
147 + "COUNT(CASE WHEN result='no' THEN 'no' ELSE NULL END) as no, "\
148 + "COUNT(CASE WHEN result='abstain' THEN 'abstain' ELSE NULL END) as abstain "\
149 + "FROM vote GROUP BY motion_id) as votes ON votes.motion_id=motion.id "\
150 + "LEFT JOIN voter poser ON poser.id = motion.posed_by "\
151 + "LEFT JOIN voter canceler ON canceler.id = motion.canceled_by "
154 p = get_db().prepare(q + "WHERE motion.host = $1 ORDER BY motion.id DESC LIMIT 11")
157 p = get_db().prepare(q + "WHERE motion.host = $1 AND motion.id <= $2 ORDER BY motion.id DESC LIMIT 11")
158 rv = p(request.host, start)
159 rs = get_db().prepare("SELECT id FROM motion WHERE motion.host = $1 AND motion.id > $2 ORDER BY id ASC LIMIT 10")(request.host, start)
164 return render_template('index.html', motions=rv[:10], more=rv[10]["id"] if len(rv) == 11 else None, times=times[request.host], prev=prev,
165 categories=get_allowed_cats("create"))
167 def rel_redirect(loc):
169 r.autocorrect_location_header = False
172 @app.route("/motion", methods=['POST'])
174 cat=request.form.get("category", "")
175 if cat not in get_allowed_cats("create"):
176 return "Forbidden", 403
177 time = int(request.form.get("days", "3"));
178 if time not in times[request.host]:
179 return "Error, invalid length", 500
182 t = db.prepare("SELECT CURRENT_TIMESTAMP")()[0][0];
183 s = db.prepare("SELECT MAX(\"identifier\") FROM \"motion\" WHERE \"type\"=$1 AND \"host\"=$2 AND DATE(\"posed\")=DATE(CURRENT_TIMESTAMP)")
184 sr = s(cat, request.host)
186 if len(sr) == 0 or sr[0][0] is None:
187 ident=prefix[request.host][cat]+"."+t.strftime("%Y%m%d")+".001"
189 ident=prefix[request.host][cat]+"."+t.strftime("%Y%m%d")+"."+("%03d" % (int(sr[0][0].split(".")[2])+1))
190 p = db.prepare("INSERT INTO motion(\"name\", \"content\", \"deadline\", \"posed_by\", \"type\", \"identifier\", \"host\") VALUES($1, $2, CURRENT_TIMESTAMP + $3 * interval '1 days', $4, $5, $6, $7)")
191 p(request.form.get("title", ""), request.form.get("content",""), time, g.voter, cat, ident, request.host)
192 return rel_redirect("/")
194 def motion_edited(motion):
195 return rel_redirect("/?start=" + str(motion) + "#motion-" + str(motion))
197 @app.route("/motion/<string:motion>/cancel", methods=['POST'])
198 def cancel_motion(motion):
199 rv = get_db().prepare("SELECT id, type FROM motion WHERE identifier=$1 AND host=$2")(motion, request.host);
201 return "Error, Not found", 404
203 if not may("cancel", rv[0].get("type")):
204 return "Forbidden", 403
205 if request.form.get("reason", "none") == "none":
206 return "Error, form requires reason", 500
207 rv = get_db().prepare("UPDATE motion SET canceled=CURRENT_TIMESTAMP, cancelation_reason=$1, canceled_by=$2 WHERE identifier=$3 AND host=$4 AND canceled is NULL")(request.form.get("reason", ""), g.voter, motion, request.host)
208 return motion_edited(id)
210 @app.route("/motion/<string:motion>")
211 def show_motion(motion):
212 p = get_db().prepare("SELECT motion.*, poser.email AS poser, canceler.email AS canceler, (motion.deadline > CURRENT_TIMESTAMP AND canceled is NULL) AS running, vote.result FROM motion "\
213 + "LEFT JOIN vote on vote.motion_id=motion.id AND vote.voter_id=$2 "\
214 + "LEFT JOIN voter poser ON poser.id = motion.posed_by "\
215 + "LEFT JOIN voter canceler ON canceler.id = motion.canceled_by "
216 + "WHERE motion.identifier=$1 AND motion.host=$3")
217 rv = p(motion, g.voter, request.host)
219 if may("audit", rv[0].get("type")) and not rv[0].get("running") and not rv[0].get("canceled"):
220 votes = get_db().prepare("SELECT vote.result, voter.email FROM vote INNER JOIN voter ON voter.id = vote.voter_id WHERE vote.motion_id=$1")(rv[0].get("id"));
221 return render_template('single_motion.html', motion=rv[0], may_vote=may("vote", rv[0].get("type")), may_cancel=may("cancel", rv[0].get("type")), votes=votes)
223 @app.route("/motion/<string:motion>/vote", methods=['POST'])
225 v = request.form.get("vote", "abstain")
228 rv = db.prepare("SELECT id, type FROM motion WHERE identifier=$1 AND host=$2")(motion, request.host);
230 return "Error, Not found", 404
231 if not may("vote", rv[0].get("type")):
232 return "Forbidden", 403
233 p = db.prepare("SELECT deadline > CURRENT_TIMESTAMP FROM motion WHERE identifier = $1 AND host=$2")
235 if not p(motion, request.host)[0][0]:
236 return "Error, motion deadline has passed", 500
237 p = db.prepare("SELECT * FROM vote WHERE motion_id = $1 AND voter_id = $2")
240 db.prepare("INSERT INTO vote(motion_id, voter_id, result) VALUES($1,$2,$3)")(id, g.voter, v)
242 db.prepare("UPDATE vote SET result=$3, entered=CURRENT_TIMESTAMP WHERE motion_id=$1 AND voter_id = $2")(id, g.voter, v)
243 return motion_edited(id)