]> WPIA git - motion.git/blob - motion.py
Merge branch 'user_host' into 'master'
[motion.git] / motion.py
1 from flask import g
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
7 import postgresql
8 import filters
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
13 import gettext
14 import click
15
16 def get_db():
17     db = getattr(g, '_database', None)
18     if db is 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
21     return db
22
23 app = Flask(__name__)
24 app.register_blueprint(filters.blueprint)
25 babel = Babel(app)
26 lang = Language(app)
27 gettext.install('motion')
28
29 class EscapeHtml(Extension):
30     def extendMarkdown(self, md, md_globals):
31         del md.preprocessors['html_block']
32         del md.postprocessors['raw_html']
33         del md.inlinePatterns['html']
34
35 md = Markdown(app, extensions=[EscapeHtml()])
36
37 class default_settings(object):
38     COPYRIGHTSTART="2017"
39     COPYRIGHTNAME="WPIA"
40     COPYRIGHTLINK="https://wpia.club"
41     IMPRINTLINK="https://documents.wpia.club/imprint.html"
42     DATAPROTECTIONLINK="https://documents.wpia.club/data_privacy_policy_html_pages_en.html"
43     MAX_PROXY=2
44
45
46 # Load config
47 app.config.from_object('motion.default_settings')
48 app.config.from_pyfile('config.py')
49
50
51 class ConfigProxy:
52     def __init__(self, name):
53         self.name = name
54     @property
55     def per_host(self):
56         dict = app.config.get(self.name)
57         if dict is None:
58             return None
59         return dict.get(request.host)
60
61 prefix = ConfigProxy("GROUP_PREFIX")
62 times = ConfigProxy("DURATION")
63 debuguser = ConfigProxy("DEBUGUSER")
64
65 max_proxy=app.config.get("MAX_PROXY")
66
67 @babel.localeselector
68 def get_locale():
69     return str(current_language)
70
71 @lang.allowed_languages
72 def get_allowed_languages():
73     return app.config['LANGUAGES'].keys()
74
75 @lang.default_language
76 def get_default_language():
77     return 'en'
78
79 def get_languages():
80     return app.config['LANGUAGES']
81
82 # Manually add vote options to the translation strings. They are used as keys in loops.
83 TRANSLATION_STRINGS={_('yes'), _('no'), _('abstain')}
84
85 @app.before_request
86 def lookup_user():
87     global prefix
88
89     env = request.environ
90     user = None
91     my_debuguser = debuguser.per_host
92     if my_debuguser is not None:
93         parts = my_debuguser.split("/", 1)
94         user = parts[0]
95         roles = parts[1]
96
97     if "USER_ROLES" in env:
98         parts = env.get("USER_ROLES").split("/", 1)
99         user = parts[0]
100         roles = parts[1]
101
102     if "USER" in env and "ROLES" in env:
103         user = env.get("USER")
104         roles = env.get("ROLES")
105
106     if user is None:
107         return _('Server misconfigured'), 500
108     roles = roles.split(" ")
109
110     if user == "<invalid>":
111         return _('Access denied'), 403;
112
113     db = get_db()
114     with db.xact():
115         rv = db.prepare("SELECT id FROM voter WHERE email=$1 AND host=$2")(user, request.host)
116         if len(rv) == 0:
117             db.prepare("INSERT INTO voter(\"email\", \"host\") VALUES($1, $2)")(user, request.host)
118             rv = db.prepare("SELECT id FROM voter WHERE email=$1 AND host=$2")(user, request.host)
119         g.voter = rv[0].get("id");
120         g.proxies_given = ""
121         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)
122         if len(rv) != 0:
123             g.proxies_given = rv[0].get("email")
124         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         if len(rv) != 0:
126             sep = ""
127             g.proxies_received = ""
128             for x in range(0, len(rv)):
129                 g.proxies_received += sep + rv[x].get("email")
130                 sep =", "
131
132     g.user = user
133     g.roles = {}
134
135     for r in roles:
136         a = r.split(":", 1)
137         if len(r)!=0:
138             val = a[1]
139             if a[0] not in g.roles:
140                 g.roles[a[0]] = []
141             if val == "*":
142                 g.roles[a[0]] = [group for group in prefix.per_host]
143             else:
144                 g.roles[a[0]].append(val)
145
146     return None
147
148 @app.context_processor
149 def init_footer_variables():
150     if int(app.config.get("COPYRIGHTSTART"))<datetime.now().year:
151         version_year = "%s - %s" % (app.config.get("COPYRIGHTSTART"), datetime.now().year)
152     else:
153         version_year = datetime.now().year
154
155     return dict(
156         footer = dict( version_year=version_year, 
157             copyright_link=app.config.get("COPYRIGHTLINK"),
158             copyright_name=app.config.get("COPYRIGHTNAME"),
159             imprint_link=app.config.get("DATAPROTECTIONLINK"),
160             dataprotection_link=app.config.get("DATAPROTECTIONLINK")
161         )
162     )
163
164
165 def get_allowed_cats(action):
166     return g.roles.get(action, []);
167
168 def may(action, motion):
169     return motion in get_allowed_cats(action)
170
171 def may_admin(action):
172     return action in g.roles
173
174 def get_voters():
175     rv = get_db().prepare("SELECT email FROM voter WHERE host=$1")(request.host)
176     return rv
177
178 def get_all_proxies():
179     rv = get_db().prepare("SELECT p.id as id, v1.email as voter_email, v1.id as voterid, "\
180                              + "v2.email as proxy_email, v2.id as proxyid "\
181                              + "FROM voter AS v1, voter AS v2, proxy AS p "\
182                              + "WHERE v2.id = p.proxy_id AND v1.id = p.voter_id AND p.revoked is NULL "\
183                              + "AND v1.host=$1 AND v2.host=$1 ORDER BY voter_email, proxy_email")(request.host)
184     return rv
185
186 @app.teardown_appcontext
187 def close_connection(exception):
188     db = getattr(g, '_database', None)
189     if db is not None:
190         db.close()
191
192 def init_db():
193     with app.app_context():
194         db = get_db()
195         try:
196             ver = db.prepare("SELECT version FROM schema_version")()[0][0];
197             print("Database Schema version: ", ver)
198         except postgresql.exceptions.UndefinedTableError:
199             g._database = None
200             db = get_db()
201             ver = 0
202
203         if ver < 1:
204             with app.open_resource('sql/schema.sql', mode='r') as f:
205                 db.execute(f.read())
206             return
207
208         if ver < 2:
209             with app.open_resource('sql/from_1.sql', mode='r') as f:
210                 db.execute(f.read())
211                 ct={}
212                 for group in [group for group in prefix[app.config.get("DEFAULT_HOST")]]:
213                     ct[group] = {"dt": "", "c": 0}
214
215                 p = db.prepare("UPDATE \"motion\" SET \"identifier\"=$1 WHERE \"id\"=$2")
216                 for row in db.prepare("SELECT id, \"type\", \"posed\" FROM \"motion\" ORDER BY \"id\" ASC"):
217                     dt=row[2].strftime("%Y%m%d")
218                     if ct[row[1]]["dt"] != dt:
219                         ct[row[1]]["dt"] = dt
220                         ct[row[1]]["c"] = 0
221                     ct[row[1]]["c"] = ct[row[1]]["c"] + 1
222                     name=prefix[app.config.get("DEFAULT_HOST")][row[1]]+"."+dt+"."+("%03d" % ct[row[1]]["c"])
223                     p(name, row[0])
224                 db.prepare("ALTER TABLE \"motion\" ALTER COLUMN \"identifier\" SET NOT NULL")()
225                 db.prepare("UPDATE \"schema_version\" SET \"version\"=2")()
226                 db.prepare("CREATE UNIQUE INDEX motion_ident ON motion (identifier)")()
227
228         if ver < 3:
229             with app.open_resource('sql/from_2.sql', mode='r') as f:
230                 db.execute(f.read())
231                 db.prepare("UPDATE \"motion\" SET \"host\"=$1")(app.config.get("DEFAULT_HOST"))
232                 db.prepare("ALTER TABLE \"motion\" ALTER COLUMN \"host\" SET NOT NULL")()
233                 db.prepare("UPDATE \"schema_version\" SET \"version\"=3")()
234
235         if ver < 4:
236             with app.open_resource('sql/from_3.sql', mode='r') as f:
237                 db.execute(f.read())
238                 db.prepare("UPDATE \"schema_version\" SET \"version\"=4")()
239
240         if ver < 5:
241             with app.open_resource('sql/from_4.sql', mode='r') as f:
242                 db.execute(f.read())
243                 db.prepare("UPDATE \"schema_version\" SET \"version\"=5")()
244
245         if ver < 6:
246             with app.open_resource('sql/from_5.sql', mode='r') as f:
247                 db.execute(f.read())
248                 rv=db.prepare("INSERT INTO voter (email, host) (SELECT vt.email, m.host FROM motion AS m, voter AS vt, vote as v "\
249                              + "WHERE (m.id=v.motion_id AND v.voter_id = vt.id) OR (m.id=v.motion_id AND v.proxy_id = vt.id) "\
250                              + "GROUP BY m.host, vt.email ORDER BY m.host, vt.email)")()
251                 rv=db.prepare("UPDATE vote SET voter_id = "\
252                              + "(SELECT v_new.id FROM motion AS m, voter AS v_new, voter as v_old "\
253                              + "WHERE v_new.email = v_old.email AND v_old.id = vote.voter_id AND "\
254                              + "vote.motion_id = m.id AND m.host = v_new.host AND v_old.host is NULL)")()
255                 rv=db.prepare("UPDATE vote SET proxy_id = "\
256                              + "(SELECT v_new.id FROM motion AS m, voter AS v_new, voter as v_old "\
257                              + "WHERE v_new.email = v_old.email AND v_old.id = vote.proxy_id AND "\
258                              + "vote.motion_id = m.id AND m.host = v_new.host AND v_old.host is NULL)")()
259                 db.prepare("DELETE FROM voter WHERE host IS Null")()
260                 db.prepare("ALTER TABLE \"voter\" ALTER COLUMN \"host\" SET NOT NULL")()
261                 db.prepare("UPDATE \"schema_version\" SET \"version\"=6")()
262
263 init_db()
264
265
266 @app.route("/")
267 def main():
268     start=int(request.args.get("start", "-1"));
269     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, "\
270                              + "COUNT(CASE WHEN result='yes' THEN 'yes' ELSE NULL END) as yes, "\
271                              + "COUNT(CASE WHEN result='no' THEN 'no' ELSE NULL END) as no, "\
272                              + "COUNT(CASE WHEN result='abstain' THEN 'abstain' ELSE NULL END) as abstain "\
273                              + "FROM vote GROUP BY motion_id) as votes ON votes.motion_id=motion.id "\
274                              + "LEFT JOIN voter poser ON poser.id = motion.posed_by "\
275                              + "LEFT JOIN voter canceler ON canceler.id = motion.canceled_by "
276     prev=None
277     if start == -1:
278         p = get_db().prepare(q + "WHERE motion.host = $1 ORDER BY motion.id DESC LIMIT 11")
279         rv = p(request.host)
280     else:
281         p = get_db().prepare(q + "WHERE motion.host = $1 AND motion.id <= $2 ORDER BY motion.id DESC LIMIT 11")
282         rv = p(request.host, start)
283         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)
284         if len(rs) == 10:
285             prev = rs[9][0]
286         else:
287             prev = -1
288     return render_template('index.html', motions=rv[:10], more=rv[10]["id"] if len(rv) == 11 else None, times=times.per_host, prev=prev,
289                            categories=get_allowed_cats("create"), singlemotion=False, may_proxyadmin=may_admin("proxyadmin"), languages=get_languages())
290
291 def rel_redirect(loc):
292     r = redirect(loc)
293     r.autocorrect_location_header = False
294     return r
295
296 @app.route("/motion", methods=['POST'])
297 def put_motion():
298     cat=request.form.get("category", "")
299     if cat not in get_allowed_cats("create"):
300         return _('Forbidden'), 403
301     time = int(request.form.get("days", "3"));
302     if time not in times.per_host:
303         return _('Error, invalid length'), 400
304     title=request.form.get("title", "")
305     title=title.strip()
306     if title =='':
307         return _('Error, missing title'), 400
308     content=request.form.get("content", "")
309     content=content.strip()
310     if content =='':
311         return _('Error, missing content'), 400
312
313     db = get_db()
314     with db.xact():
315         t = db.prepare("SELECT CURRENT_TIMESTAMP")()[0][0];
316         s = db.prepare("SELECT MAX(\"identifier\") FROM \"motion\" WHERE \"type\"=$1 AND \"host\"=$2 AND DATE(\"posed\")=DATE(CURRENT_TIMESTAMP)")
317         sr = s(cat, request.host)
318         ident=""
319         if len(sr) == 0 or sr[0][0] is None:
320             ident=prefix.per_host[cat]+"."+t.strftime("%Y%m%d")+".001"
321         else:
322             nextId = int(sr[0][0].split(".")[2])+1
323             if nextId >= 1000:
324                 return _('Too many motions for this day'), 500
325             ident=prefix.per_host[cat]+"."+t.strftime("%Y%m%d")+"."+("%03d" % nextId)
326         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)")
327         p(title, content, time, g.voter, cat, ident, request.host)
328     return rel_redirect("/")
329
330 def motion_edited(motion):
331     return rel_redirect("/motion/" + motion)
332
333 def validate_motion_access(privilege):
334     def decorator(f):
335         def decorated_function(motion):
336             db = get_db()
337             with db.xact():
338                 rv = db.prepare("SELECT id, type, deadline < CURRENT_TIMESTAMP AS expired, canceled FROM motion WHERE identifier=$1 AND host=$2")(motion, request.host);
339                 if len(rv) == 0:
340                     return _('Error, Not found'), 404
341                 id = rv[0].get("id")
342                 if not may(privilege, rv[0].get("type")):
343                     return _('Forbidden'), 403
344                 if rv[0].get("canceled") is not None:
345                     return _('Error, motion was canceled'), 403
346                 if rv[0].get("expired"):
347                     return _('Error, out of time'), 403
348             return f(motion, id)
349         decorated_function.__name__ = f.__name__
350         return decorated_function
351     return decorator
352     
353 def validate_motion_access_vote(privilege):
354     simple_decorator = validate_motion_access(privilege)
355     def decorator(f):
356         def decorated_function(motion, voter):
357             return simple_decorator(lambda motion, id : f(motion, voter, id))(motion)
358         decorated_function.__name__ = f.__name__
359         return decorated_function
360     return decorator
361
362 @app.route("/motion/<string:motion>/cancel", methods=['POST'])
363 @validate_motion_access('cancel')
364 def cancel_motion(motion, id):
365     if request.form.get("reason", "none") == "none":
366         return _('Error, form requires reason'), 500
367     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)
368     return motion_edited(motion)
369
370 @app.route("/motion/<string:motion>/finish", methods=['POST'])
371 @validate_motion_access('finish')
372 def finish_motion(motion, id):
373     rv = get_db().prepare("UPDATE motion SET deadline=CURRENT_TIMESTAMP WHERE identifier=$1 AND host=$2 AND canceled is NULL")(motion, request.host)
374     return motion_edited(motion)
375
376 @app.route("/motion/<string:motion>")
377 def show_motion(motion):
378     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 "\
379                          + "LEFT JOIN vote on vote.motion_id=motion.id AND vote.voter_id=$2 "\
380                          + "LEFT JOIN voter poser ON poser.id = motion.posed_by "\
381                          + "LEFT JOIN voter canceler ON canceler.id = motion.canceled_by "
382                          + "WHERE motion.identifier=$1 AND motion.host=$3")
383     resultmotion = p(motion, g.voter, request.host)
384     if len(resultmotion) == 0:
385         return _('Error, Not found'), 404
386
387     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")
388     resultproxyname = p(resultmotion[0][0], g.voter)
389
390     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 "\
391                           + "(SELECT vote.voter_id, vote.result, vote.proxy_id FROM vote "\
392                           + "WHERE vote.motion_id=$1) AS v ON proxy.voter_id = v.voter_id "\
393                           + "LEFT JOIN voter ON proxy.voter_id = voter.id "\
394                           + "WHERE proxy.proxy_id=$2 AND proxy.revoked IS NULL")
395     resultproxyvote = p(resultmotion[0][0], g.voter)
396
397     votes = None
398     if may("audit", resultmotion[0].get("type")) and not resultmotion[0].get("running") and not resultmotion[0].get("canceled"):
399         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"));
400         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"));
401     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")), may_finish=may("finish", resultmotion[0].get("type")), votes=votes, proxyvote=resultproxyvote, proxyname=resultproxyname, languages=get_languages())
402
403 @app.route("/motion/<string:motion>/vote/<string:voter>", methods=['POST'])
404 @validate_motion_access_vote('vote')
405 def vote(motion, voter, id):
406     v = request.form.get("vote", "abstain")
407     voterid=int(voter)
408     db = get_db()
409
410     # test if voter is proxy
411     if (voterid != g.voter):
412         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);
413         if len(rv) == 0:
414             return _('Error, proxy not found.'), 400
415
416     p = db.prepare("SELECT * FROM vote WHERE motion_id = $1 AND voter_id = $2")
417     rv = p(id, voterid)
418     if len(rv) == 0:
419         db.prepare("INSERT INTO vote(motion_id, voter_id, result, proxy_id) VALUES($1,$2,$3,$4)")(id, voterid, v, g.voter)
420     else:
421         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)
422     return motion_edited(motion)
423
424 @app.route("/proxy")
425 def proxy():
426     if not may_admin("proxyadmin"):
427         return _('Forbidden'), 403
428     return render_template('proxy.html', voters=get_voters(), proxies=get_all_proxies(), may_proxyadmin=may_admin("proxyadmin"), languages=get_languages())
429
430 @app.route("/proxy/add", methods=['POST'])
431 def add_proxy():
432     if not may_admin("proxyadmin"):
433         return _('Forbidden'), 403
434     voter=request.form.get("voter", "")
435     proxy=request.form.get("proxy", "")
436     if voter == proxy :
437         return _('Error, voter equals proxy.'), 400
438     rv = get_db().prepare("SELECT id FROM voter WHERE email=$1 AND host=$2")(voter, request.host);
439     if len(rv) == 0:
440         return _('Error, voter not found.'), 400
441     voterid = rv[0].get("id")
442     rv = get_db().prepare("SELECT id, host FROM voter WHERE email=$1 AND host=$2")(proxy, request.host);
443     if len(rv) == 0:
444         return _('Error, proxy not found.'), 400
445     proxyid = rv[0].get("id")
446     rv = get_db().prepare("SELECT id FROM proxy WHERE voter_id=$1 AND revoked is NULL")(voterid);
447     if len(rv) != 0:
448         return _('Error, proxy allready given.'), 400
449     rv = get_db().prepare("SELECT COUNT(id) as c FROM proxy WHERE proxy_id=$1 AND revoked is NULL GROUP BY proxy_id")(proxyid);
450     if len(rv) != 0:
451         if rv[0].get("c") is None or rv[0].get("c") >= max_proxy:
452             return _("Error, Max proxy for '%s' reached.") % (proxy), 400
453     rv = get_db().prepare("INSERT INTO proxy(voter_id, proxy_id, granted_by) VALUES ($1,$2,$3)")(voterid, proxyid, g.voter)
454     return rel_redirect("/proxy")
455
456 @app.route("/proxy/revoke", methods=['POST'])
457 def revoke_proxy():
458     if not may_admin("proxyadmin"):
459         return _('Forbidden'), 403
460     id=request.form.get("id", "")
461     rv = get_db().prepare("UPDATE proxy SET revoked=CURRENT_TIMESTAMP, revoked_by=$1 WHERE id=$2")(g.voter, int(id))
462     return rel_redirect("/proxy")
463
464 @app.route("/proxy/revokeall", methods=['POST'])
465 def revoke_proxy_all():
466     if not may_admin("proxyadmin"):
467         return _('Forbidden'), 403
468     rv = get_db().prepare("UPDATE proxy SET revoked=CURRENT_TIMESTAMP, revoked_by=$1 WHERE revoked IS NULL")(g.voter)
469     return rel_redirect("/proxy")
470
471 @app.route("/language/<string:language>")
472 def set_language(language):
473     lang.change_language(language)
474     return rel_redirect("/")
475
476 @app.cli.command("create-user")
477 @click.argument("email")
478 @click.argument("host")
479 def create_user(email, host):
480     db = get_db()
481     with db.xact():
482         rv = db.prepare("SELECT id FROM voter WHERE lower(email)=lower($1) AND host=$2")(email, host)
483         messagetext=_("User '%s' already exists on %s.") % (email, host)
484         if len(rv) == 0:
485             db.prepare("INSERT INTO voter(\"email\", \"host\") VALUES($1, $2)")(email, host)
486             messagetext=_("User '%s' inserted to %s.") % (email, host)
487     click.echo(messagetext)