2 from flask import Flask
3 from flask import render_template, redirect
4 from flask import request
5 from functools import wraps
6 from flask_babel import Babel, gettext
9 from flaskext.markdown import Markdown
10 from markdown.extensions import Extension
11 from datetime import date, time, datetime
12 from flask_language import Language, current_language
17 db = getattr(g, '_database', None)
19 db = g._database = postgresql.open(app.config.get("DATABASE"), user=app.config.get("USER"), password=app.config.get("PASSWORD"))
20 #db.row_factory = sqlite3.Row
24 app.register_blueprint(filters.blueprint)
27 gettext.install('motion')
29 class EscapeHtml(Extension):
30 def extendMarkdown(self, md, md_globals):
31 del md.preprocessors['html_block']
32 del md.inlinePatterns['html']
34 md = Markdown(app, extensions=[EscapeHtml()])
36 class default_settings(object):
39 COPYRIGHTLINK="https://wpia.club"
40 IMPRINTLINK="https://documents.wpia.club/imprint.html"
41 DATAPROTECTIONLINK="https://documents.wpia.club/data_privacy_policy_html_pages_en.html"
45 app.config.from_object('motion.default_settings')
46 app.config.from_pyfile('config.py')
50 def __init__(self, name):
54 dict = app.config.get(self.name)
57 return dict.get(request.host)
59 prefix = ConfigProxy("GROUP_PREFIX")
60 times = ConfigProxy("DURATION")
61 debuguser = ConfigProxy("DEBUGUSER")
63 max_proxy=app.config.get("MAX_PROXY")
67 return str(current_language)
69 @lang.allowed_languages
70 def get_allowed_languages():
71 return app.config['LANGUAGES'].keys()
73 @lang.default_language
74 def get_default_language():
78 return app.config['LANGUAGES']
80 # Manually add vote options to the translation strings. They are used as keys in loops.
81 TRANSLATION_STRINGS={_('yes'), _('no'), _('abstain')}
89 my_debuguser = debuguser.per_host
90 if my_debuguser is not None:
91 parts = my_debuguser.split("/", 1)
95 if "USER_ROLES" in env:
96 parts = env.get("USER_ROLES").split("/", 1)
100 if "USER" in env and "ROLES" in env:
101 user = env.get("USER")
102 roles = env.get("ROLES")
105 return _('Server misconfigured'), 500
106 roles = roles.split(" ")
108 if user == "<invalid>":
109 return _('Access denied'), 403;
113 rv = db.prepare("SELECT id FROM voter WHERE email=$1")(user)
115 db.prepare("INSERT INTO voter(\"email\") VALUES($1)")(user)
116 rv = db.prepare("SELECT id FROM voter WHERE email=$1")(user)
117 g.voter = rv[0].get("id");
119 rv = db.prepare("SELECT email, voter_id FROM voter, proxy WHERE proxy.proxy_id = voter.id AND proxy.revoked IS NULL AND proxy.voter_id = $1 ")(g.voter)
121 g.proxies_given = rv[0].get("email")
122 rv = db.prepare("SELECT email, voter_id FROM voter, proxy WHERE proxy.voter_id = voter.id AND proxy.revoked IS NULL AND proxy.proxy_id = $1 ")(g.voter)
125 g.proxies_received = ""
126 for x in range(0, len(rv)):
127 g.proxies_received += sep + rv[x].get("email")
137 if a[0] not in g.roles:
140 g.roles[a[0]] = [group for group in prefix.per_host]
142 g.roles[a[0]].append(val)
146 @app.context_processor
147 def init_footer_variables():
148 if int(app.config.get("COPYRIGHTSTART"))<datetime.now().year:
149 version_year = "%s - %s" % (app.config.get("COPYRIGHTSTART"), datetime.now().year)
151 version_year = datetime.now().year
154 footer = dict( version_year=version_year,
155 copyright_link=app.config.get("COPYRIGHTLINK"),
156 copyright_name=app.config.get("COPYRIGHTNAME"),
157 imprint_link=app.config.get("DATAPROTECTIONLINK"),
158 dataprotection_link=app.config.get("DATAPROTECTIONLINK")
163 def get_allowed_cats(action):
164 return g.roles.get(action, []);
166 def may(action, motion):
167 return motion in get_allowed_cats(action)
169 def may_admin(action):
170 return action in g.roles
173 rv = get_db().prepare("SELECT email FROM voter")
176 def get_all_proxies():
177 rv = get_db().prepare("SELECT p.id as id, v1.email as voter_email, v1.id as voterid, v2.email as proxy_email, v2.id as proxyid FROM voter AS v1, voter AS v2, proxy AS p WHERE v2.id = p.proxy_id AND v1.id = p.voter_id AND p.revoked is NULL ORDER BY voter_email, proxy_email")
180 @app.teardown_appcontext
181 def close_connection(exception):
182 db = getattr(g, '_database', None)
187 with app.app_context():
190 ver = db.prepare("SELECT version FROM schema_version")()[0][0];
191 print("Database Schema version: ", ver)
192 except postgresql.exceptions.UndefinedTableError:
198 with app.open_resource('sql/schema.sql', mode='r') as f:
203 with app.open_resource('sql/from_1.sql', mode='r') as f:
206 for group in [group for group in prefix[app.config.get("DEFAULT_HOST")]]:
207 ct[group] = {"dt": "", "c": 0}
209 p = db.prepare("UPDATE \"motion\" SET \"identifier\"=$1 WHERE \"id\"=$2")
210 for row in db.prepare("SELECT id, \"type\", \"posed\" FROM \"motion\" ORDER BY \"id\" ASC"):
211 dt=row[2].strftime("%Y%m%d")
212 if ct[row[1]]["dt"] != dt:
213 ct[row[1]]["dt"] = dt
215 ct[row[1]]["c"] = ct[row[1]]["c"] + 1
216 name=prefix[app.config.get("DEFAULT_HOST")][row[1]]+"."+dt+"."+("%03d" % ct[row[1]]["c"])
218 db.prepare("ALTER TABLE \"motion\" ALTER COLUMN \"identifier\" SET NOT NULL")()
219 db.prepare("UPDATE \"schema_version\" SET \"version\"=2")()
220 db.prepare("CREATE UNIQUE INDEX motion_ident ON motion (identifier)")()
223 with app.open_resource('sql/from_2.sql', mode='r') as f:
225 db.prepare("UPDATE \"motion\" SET \"host\"=$1")(app.config.get("DEFAULT_HOST"))
226 db.prepare("ALTER TABLE \"motion\" ALTER COLUMN \"host\" SET NOT NULL")()
227 db.prepare("UPDATE \"schema_version\" SET \"version\"=3")()
230 with app.open_resource('sql/from_3.sql', mode='r') as f:
232 db.prepare("UPDATE \"schema_version\" SET \"version\"=4")()
235 with app.open_resource('sql/from_4.sql', mode='r') as f:
237 db.prepare("UPDATE \"schema_version\" SET \"version\"=5")()
244 start=int(request.args.get("start", "-1"));
245 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, "\
246 + "COUNT(CASE WHEN result='yes' THEN 'yes' ELSE NULL END) as yes, "\
247 + "COUNT(CASE WHEN result='no' THEN 'no' ELSE NULL END) as no, "\
248 + "COUNT(CASE WHEN result='abstain' THEN 'abstain' ELSE NULL END) as abstain "\
249 + "FROM vote GROUP BY motion_id) as votes ON votes.motion_id=motion.id "\
250 + "LEFT JOIN voter poser ON poser.id = motion.posed_by "\
251 + "LEFT JOIN voter canceler ON canceler.id = motion.canceled_by "
254 p = get_db().prepare(q + "WHERE motion.host = $1 ORDER BY motion.id DESC LIMIT 11")
257 p = get_db().prepare(q + "WHERE motion.host = $1 AND motion.id <= $2 ORDER BY motion.id DESC LIMIT 11")
258 rv = p(request.host, start)
259 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)
264 return render_template('index.html', motions=rv[:10], more=rv[10]["id"] if len(rv) == 11 else None, times=times.per_host, prev=prev,
265 categories=get_allowed_cats("create"), singlemotion=False, may_proxyadmin=may_admin("proxyadmin"), languages=get_languages())
267 def rel_redirect(loc):
269 r.autocorrect_location_header = False
272 @app.route("/motion", methods=['POST'])
274 cat=request.form.get("category", "")
275 if cat not in get_allowed_cats("create"):
276 return _('Forbidden'), 403
277 time = int(request.form.get("days", "3"));
278 if time not in times.per_host:
279 return _('Error, invalid length'), 400
280 title=request.form.get("title", "")
283 return _('Error, missing title'), 400
284 content=request.form.get("content", "")
285 content=content.strip()
287 return _('Error, missing content'), 400
291 t = db.prepare("SELECT CURRENT_TIMESTAMP")()[0][0];
292 s = db.prepare("SELECT MAX(\"identifier\") FROM \"motion\" WHERE \"type\"=$1 AND \"host\"=$2 AND DATE(\"posed\")=DATE(CURRENT_TIMESTAMP)")
293 sr = s(cat, request.host)
295 if len(sr) == 0 or sr[0][0] is None:
296 ident=prefix.per_host[cat]+"."+t.strftime("%Y%m%d")+".001"
298 ident=prefix.per_host[cat]+"."+t.strftime("%Y%m%d")+"."+("%03d" % (int(sr[0][0].split(".")[2])+1))
299 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)")
300 p(title, content, time, g.voter, cat, ident, request.host)
301 return rel_redirect("/")
303 def motion_edited(motion):
304 return rel_redirect("/motion/" + motion)
306 def validate_motion_access(privilege):
308 def decorated_function(motion):
311 rv = db.prepare("SELECT id, type, deadline < CURRENT_TIMESTAMP AS expired, canceled FROM motion WHERE identifier=$1 AND host=$2")(motion, request.host);
313 return _('Error, Not found'), 404
315 if not may(privilege, rv[0].get("type")):
316 return _('Forbidden'), 403
317 if rv[0].get("canceled") is not None:
318 return _('Error, motion was canceled'), 403
319 if rv[0].get("expired"):
320 return _('Error, out of time'), 403
322 decorated_function.__name__ = f.__name__
323 return decorated_function
326 def validate_motion_access_vote(privilege):
327 simple_decorator = validate_motion_access(privilege)
329 def decorated_function(motion, voter):
330 return simple_decorator(lambda motion, id : f(motion, voter, id))(motion)
331 decorated_function.__name__ = f.__name__
332 return decorated_function
335 @app.route("/motion/<string:motion>/cancel", methods=['POST'])
336 @validate_motion_access('cancel')
337 def cancel_motion(motion, id):
338 if request.form.get("reason", "none") == "none":
339 return _('Error, form requires reason'), 500
340 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)
341 return motion_edited(motion)
343 @app.route("/motion/<string:motion>/finish", methods=['POST'])
344 @validate_motion_access('finish')
345 def finish_motion(motion, id):
346 rv = get_db().prepare("UPDATE motion SET deadline=CURRENT_TIMESTAMP WHERE identifier=$1 AND host=$2 AND canceled is NULL")(motion, request.host)
347 return motion_edited(motion)
349 @app.route("/motion/<string:motion>")
350 def show_motion(motion):
351 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 "\
352 + "LEFT JOIN vote on vote.motion_id=motion.id AND vote.voter_id=$2 "\
353 + "LEFT JOIN voter poser ON poser.id = motion.posed_by "\
354 + "LEFT JOIN voter canceler ON canceler.id = motion.canceled_by "
355 + "WHERE motion.identifier=$1 AND motion.host=$3")
356 resultmotion = p(motion, g.voter, request.host)
357 if len(resultmotion) == 0:
358 return _('Error, Not found'), 404
360 p = get_db().prepare("SELECT voter.email FROM vote INNER JOIN voter ON vote.proxy_id = voter.id WHERE vote.motion_id=$1 AND vote.voter_id=$2 AND vote.proxy_id <> vote.voter_id")
361 resultproxyname = p(resultmotion[0][0], g.voter)
363 p = get_db().prepare("SELECT v.result, proxy.voter_id, voter.email, CASE WHEN proxy.proxy_id = v.proxy_id THEN NULL ELSE voter.email END AS owneremail FROM proxy LEFT JOIN "\
364 + "(SELECT vote.voter_id, vote.result, vote.proxy_id FROM vote "\
365 + "WHERE vote.motion_id=$1) AS v ON proxy.voter_id = v.voter_id "\
366 + "LEFT JOIN voter ON proxy.voter_id = voter.id "\
367 + "WHERE proxy.proxy_id=$2 AND proxy.revoked IS NULL")
368 resultproxyvote = p(resultmotion[0][0], g.voter)
371 if may("audit", resultmotion[0].get("type")) and not resultmotion[0].get("running") and not resultmotion[0].get("canceled"):
372 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")(resultmotion[0].get("id"));
373 votes = get_db().prepare("SELECT vote.result, voter.email, CASE voter.email WHEN proxy.email THEN NULL ELSE proxy.email END as proxyemail FROM vote INNER JOIN voter ON voter.id = vote.voter_id INNER JOIN voter as proxy ON proxy.id = vote.proxy_id WHERE vote.motion_id=$1")(resultmotion[0].get("id"));
374 return render_template('single_motion.html', motion=resultmotion[0], may_vote=may("vote", resultmotion[0].get("type")), may_cancel=may("cancel", resultmotion[0].get("type")), votes=votes, proxyvote=resultproxyvote, proxyname=resultproxyname, languages=get_languages())
376 @app.route("/motion/<string:motion>/vote/<string:voter>", methods=['POST'])
377 @validate_motion_access_vote('vote')
378 def vote(motion, voter, id):
379 v = request.form.get("vote", "abstain")
383 # test if voter is proxy
384 if (voterid != g.voter):
385 rv = db.prepare("SELECT voter_id FROM proxy WHERE proxy.revoked IS NULL AND proxy.proxy_id = $1 AND proxy.voter_id = $2")(g.voter, voterid);
387 return _('Error, proxy not found.'), 400
389 p = db.prepare("SELECT * FROM vote WHERE motion_id = $1 AND voter_id = $2")
392 db.prepare("INSERT INTO vote(motion_id, voter_id, result, proxy_id) VALUES($1,$2,$3,$4)")(id, voterid, v, g.voter)
394 db.prepare("UPDATE vote SET result=$3, entered=CURRENT_TIMESTAMP, proxy_id=$4 WHERE motion_id=$1 AND voter_id = $2")(id, voterid, v, g.voter)
395 return motion_edited(motion)
399 if not may_admin("proxyadmin"):
400 return _('Forbidden'), 403
401 return render_template('proxy.html', voters=get_voters(), proxies=get_all_proxies(), may_proxyadmin=may_admin("proxyadmin"), languages=get_languages())
403 @app.route("/proxy/add", methods=['POST'])
405 if not may_admin("proxyadmin"):
406 return _('Forbidden'), 403
407 voter=request.form.get("voter", "")
408 proxy=request.form.get("proxy", "")
410 return _('Error, voter equals proxy.'), 400
411 rv = get_db().prepare("SELECT id FROM voter WHERE email=$1")(voter);
413 return _('Error, voter not found.'), 400
414 voterid = rv[0].get("id")
415 rv = get_db().prepare("SELECT id FROM voter WHERE email=$1")(proxy);
417 return _('Error, proxy not found.'), 400
418 proxyid = rv[0].get("id")
419 rv = get_db().prepare("SELECT id FROM proxy WHERE voter_id=$1 AND revoked is NULL")(voterid);
421 return _('Error, proxy allready given.'), 400
422 rv = get_db().prepare("SELECT COUNT(id) as c FROM proxy WHERE proxy_id=$1 AND revoked is NULL GROUP BY proxy_id")(proxyid);
424 if rv[0].get("c") >= max_proxy:
425 return _("Error, Max proxy for '%s' reached.") % (proxy), 400
426 rv = get_db().prepare("INSERT INTO proxy(voter_id, proxy_id, granted_by) VALUES ($1,$2,$3)")(voterid, proxyid, g.voter)
427 return rel_redirect("/proxy")
429 @app.route("/proxy/revoke", methods=['POST'])
431 if not may_admin("proxyadmin"):
432 return _('Forbidden'), 403
433 id=request.form.get("id", "")
434 rv = get_db().prepare("UPDATE proxy SET revoked=CURRENT_TIMESTAMP, revoked_by=$1 WHERE id=$2")(g.voter, int(id))
435 return rel_redirect("/proxy")
437 @app.route("/proxy/revokeall", methods=['POST'])
438 def revoke_proxy_all():
439 if not may_admin("proxyadmin"):
440 return _('Forbidden'), 403
441 rv = get_db().prepare("UPDATE proxy SET revoked=CURRENT_TIMESTAMP, revoked_by=$1 WHERE revoked IS NULL")(g.voter)
442 return rel_redirect("/proxy")
444 @app.route("/language/<string:language>")
445 def set_language(language):
446 lang.change_language(language)
447 return rel_redirect("/")
449 @app.cli.command("create-user")
450 @click.argument("email")
451 def create_user(email):
454 rv = db.prepare("SELECT id FROM voter WHERE lower(email)=lower($1)")(email)
455 messagetext="User '%s' already exists." % (email)
457 db.prepare("INSERT INTO voter(\"email\") VALUES($1)")(email)
458 messagetext="User '%s' inserted." % (email)
459 click.echo(messagetext)