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")(user)
 116         if len(rv) == 0:
 117             db.prepare("INSERT INTO voter(\"email\") VALUES($1)")(user)
 118             rv = db.prepare("SELECT id FROM voter WHERE email=$1")(user)
 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")
 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, 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     return rv
 181
 182 @app.teardown_appcontext
 183 def close_connection(exception):
 184     db = getattr(g, '_database', None)
 185     if db is not None:
 186         db.close()
 187
 188 def init_db():
 189     with app.app_context():
 190         db = get_db()
 191         try:
 192             ver = db.prepare("SELECT version FROM schema_version")()[0][0];
 193             print("Database Schema version: ", ver)
 194         except postgresql.exceptions.UndefinedTableError:
 195             g._database = None
 196             db = get_db()
 197             ver = 0
 198
 199         if ver < 1:
 200             with app.open_resource('sql/schema.sql', mode='r') as f:
 201                 db.execute(f.read())
 202             return
 203
 204         if ver < 2:
 205             with app.open_resource('sql/from_1.sql', mode='r') as f:
 206                 db.execute(f.read())
 207                 ct={}
 208                 for group in [group for group in prefix[app.config.get("DEFAULT_HOST")]]:
 209                     ct[group] = {"dt": "", "c": 0}
 210
 211                 p = db.prepare("UPDATE \"motion\" SET \"identifier\"=$1 WHERE \"id\"=$2")
 212                 for row in db.prepare("SELECT id, \"type\", \"posed\" FROM \"motion\" ORDER BY \"id\" ASC"):
 213                     dt=row[2].strftime("%Y%m%d")
 214                     if ct[row[1]]["dt"] != dt:
 215                         ct[row[1]]["dt"] = dt
 216                         ct[row[1]]["c"] = 0
 217                     ct[row[1]]["c"] = ct[row[1]]["c"] + 1
 218                     name=prefix[app.config.get("DEFAULT_HOST")][row[1]]+"."+dt+"."+("%03d" % ct[row[1]]["c"])
 219                     p(name, row[0])
 220                 db.prepare("ALTER TABLE \"motion\" ALTER COLUMN \"identifier\" SET NOT NULL")()
 221                 db.prepare("UPDATE \"schema_version\" SET \"version\"=2")()
 222                 db.prepare("CREATE UNIQUE INDEX motion_ident ON motion (identifier)")()
 223
 224         if ver < 3:
 225             with app.open_resource('sql/from_2.sql', mode='r') as f:
 226                 db.execute(f.read())
 227                 db.prepare("UPDATE \"motion\" SET \"host\"=$1")(app.config.get("DEFAULT_HOST"))
 228                 db.prepare("ALTER TABLE \"motion\" ALTER COLUMN \"host\" SET NOT NULL")()
 229                 db.prepare("UPDATE \"schema_version\" SET \"version\"=3")()
 230
 231         if ver < 4:
 232             with app.open_resource('sql/from_3.sql', mode='r') as f:
 233                 db.execute(f.read())
 234                 db.prepare("UPDATE \"schema_version\" SET \"version\"=4")()
 235
 236         if ver < 5:
 237             with app.open_resource('sql/from_4.sql', mode='r') as f:
 238                 db.execute(f.read())
 239                 db.prepare("UPDATE \"schema_version\" SET \"version\"=5")()
 240
 241 init_db()
 242
 243
 244 @app.route("/")
 245 def main():
 246     start=int(request.args.get("start", "-1"));
 247     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, "\
 248                              + "COUNT(CASE WHEN result='yes' THEN 'yes' ELSE NULL END) as yes, "\
 249                              + "COUNT(CASE WHEN result='no' THEN 'no' ELSE NULL END) as no, "\
 250                              + "COUNT(CASE WHEN result='abstain' THEN 'abstain' ELSE NULL END) as abstain "\
 251                              + "FROM vote GROUP BY motion_id) as votes ON votes.motion_id=motion.id "\
 252                              + "LEFT JOIN voter poser ON poser.id = motion.posed_by "\
 253                              + "LEFT JOIN voter canceler ON canceler.id = motion.canceled_by "
 254     prev=None
 255     if start == -1:
 256         p = get_db().prepare(q + "WHERE motion.host = $1 ORDER BY motion.id DESC LIMIT 11")
 257         rv = p(request.host)
 258     else:
 259         p = get_db().prepare(q + "WHERE motion.host = $1 AND motion.id <= $2 ORDER BY motion.id DESC LIMIT 11")
 260         rv = p(request.host, start)
 261         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)
 262         if len(rs) == 10:
 263             prev = rs[9][0]
 264         else:
 265             prev = -1
 266     return render_template('index.html', motions=rv[:10], more=rv[10]["id"] if len(rv) == 11 else None, times=times.per_host, prev=prev,
 267                            categories=get_allowed_cats("create"), singlemotion=False, may_proxyadmin=may_admin("proxyadmin"), languages=get_languages())
 268
 269 def rel_redirect(loc):
 270     r = redirect(loc)
 271     r.autocorrect_location_header = False
 272     return r
 273
 274 @app.route("/motion", methods=['POST'])
 275 def put_motion():
 276     cat=request.form.get("category", "")
 277     if cat not in get_allowed_cats("create"):
 278         return _('Forbidden'), 403
 279     time = int(request.form.get("days", "3"));
 280     if time not in times.per_host:
 281         return _('Error, invalid length'), 400
 282     title=request.form.get("title", "")
 283     title=title.strip()
 284     if title =='':
 285         return _('Error, missing title'), 400
 286     content=request.form.get("content", "")
 287     content=content.strip()
 288     if content =='':
 289         return _('Error, missing content'), 400
 290
 291     db = get_db()
 292     with db.xact():
 293         t = db.prepare("SELECT CURRENT_TIMESTAMP")()[0][0];
 294         s = db.prepare("SELECT MAX(\"identifier\") FROM \"motion\" WHERE \"type\"=$1 AND \"host\"=$2 AND DATE(\"posed\")=DATE(CURRENT_TIMESTAMP)")
 295         sr = s(cat, request.host)
 296         ident=""
 297         if len(sr) == 0 or sr[0][0] is None:
 298             ident=prefix.per_host[cat]+"."+t.strftime("%Y%m%d")+".001"
 299         else:
 300             nextId = int(sr[0][0].split(".")[2])+1
 301             if nextId >= 1000:
 302                 return _('Too many motions for this day'), 500
 303             ident=prefix.per_host[cat]+"."+t.strftime("%Y%m%d")+"."+("%03d" % nextId)
 304         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)")
 305         p(title, content, time, g.voter, cat, ident, request.host)
 306     return rel_redirect("/")
 307
 308 def motion_edited(motion):
 309     return rel_redirect("/motion/" + motion)
 310
 311 def validate_motion_access(privilege):
 312     def decorator(f):
 313         def decorated_function(motion):
 314             db = get_db()
 315             with db.xact():
 316                 rv = db.prepare("SELECT id, type, deadline < CURRENT_TIMESTAMP AS expired, canceled FROM motion WHERE identifier=$1 AND host=$2")(motion, request.host);
 317                 if len(rv) == 0:
 318                     return _('Error, Not found'), 404
 319                 id = rv[0].get("id")
 320                 if not may(privilege, rv[0].get("type")):
 321                     return _('Forbidden'), 403
 322                 if rv[0].get("canceled") is not None:
 323                     return _('Error, motion was canceled'), 403
 324                 if rv[0].get("expired"):
 325                     return _('Error, out of time'), 403
 326             return f(motion, id)
 327         decorated_function.__name__ = f.__name__
 328         return decorated_function
 329     return decorator
 330
 331 def validate_motion_access_vote(privilege):
 332     simple_decorator = validate_motion_access(privilege)
 333     def decorator(f):
 334         def decorated_function(motion, voter):
 335             return simple_decorator(lambda motion, id : f(motion, voter, id))(motion)
 336         decorated_function.__name__ = f.__name__
 337         return decorated_function
 338     return decorator
 339
 340 @app.route("/motion/<string:motion>/cancel", methods=['POST'])
 341 @validate_motion_access('cancel')
 342 def cancel_motion(motion, id):
 343     if request.form.get("reason", "none") == "none":
 344         return _('Error, form requires reason'), 500
 345     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)
 346     return motion_edited(motion)
 347
 348 @app.route("/motion/<string:motion>/finish", methods=['POST'])
 349 @validate_motion_access('finish')
 350 def finish_motion(motion, id):
 351     rv = get_db().prepare("UPDATE motion SET deadline=CURRENT_TIMESTAMP WHERE identifier=$1 AND host=$2 AND canceled is NULL")(motion, request.host)
 352     return motion_edited(motion)
 353
 354 @app.route("/motion/<string:motion>")
 355 def show_motion(motion):
 356     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 "\
 357                          + "LEFT JOIN vote on vote.motion_id=motion.id AND vote.voter_id=$2 "\
 358                          + "LEFT JOIN voter poser ON poser.id = motion.posed_by "\
 359                          + "LEFT JOIN voter canceler ON canceler.id = motion.canceled_by "
 360                          + "WHERE motion.identifier=$1 AND motion.host=$3")
 361     resultmotion = p(motion, g.voter, request.host)
 362     if len(resultmotion) == 0:
 363         return _('Error, Not found'), 404
 364
 365     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")
 366     resultproxyname = p(resultmotion[0][0], g.voter)
 367
 368     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 "\
 369                           + "(SELECT vote.voter_id, vote.result, vote.proxy_id FROM vote "\
 370                           + "WHERE vote.motion_id=$1) AS v ON proxy.voter_id = v.voter_id "\
 371                           + "LEFT JOIN voter ON proxy.voter_id = voter.id "\
 372                           + "WHERE proxy.proxy_id=$2 AND proxy.revoked IS NULL")
 373     resultproxyvote = p(resultmotion[0][0], g.voter)
 374
 375     votes = None
 376     if may("audit", resultmotion[0].get("type")) and not resultmotion[0].get("running") and not resultmotion[0].get("canceled"):
 377         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"));
 378         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"));
 379     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())
 380
 381 @app.route("/motion/<string:motion>/vote/<string:voter>", methods=['POST'])
 382 @validate_motion_access_vote('vote')
 383 def vote(motion, voter, id):
 384     v = request.form.get("vote", "abstain")
 385     voterid=int(voter)
 386     db = get_db()
 387
 388     # test if voter is proxy
 389     if (voterid != g.voter):
 390         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);
 391         if len(rv) == 0:
 392             return _('Error, proxy not found.'), 400
 393
 394     p = db.prepare("SELECT * FROM vote WHERE motion_id = $1 AND voter_id = $2")
 395     rv = p(id, voterid)
 396     if len(rv) == 0:
 397         db.prepare("INSERT INTO vote(motion_id, voter_id, result, proxy_id) VALUES($1,$2,$3,$4)")(id, voterid, v, g.voter)
 398     else:
 399         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)
 400     return motion_edited(motion)
 401
 402 @app.route("/proxy")
 403 def proxy():
 404     if not may_admin("proxyadmin"):
 405         return _('Forbidden'), 403
 406     return render_template('proxy.html', voters=get_voters(), proxies=get_all_proxies(), may_proxyadmin=may_admin("proxyadmin"), languages=get_languages())
 407
 408 @app.route("/proxy/add", methods=['POST'])
 409 def add_proxy():
 410     if not may_admin("proxyadmin"):
 411         return _('Forbidden'), 403
 412     voter=request.form.get("voter", "")
 413     proxy=request.form.get("proxy", "")
 414     if voter == proxy :
 415         return _('Error, voter equals proxy.'), 400
 416     rv = get_db().prepare("SELECT id FROM voter WHERE email=$1")(voter);
 417     if len(rv) == 0:
 418         return _('Error, voter not found.'), 400
 419     voterid = rv[0].get("id")
 420     rv = get_db().prepare("SELECT id FROM voter WHERE email=$1")(proxy);
 421     if len(rv) == 0:
 422         return _('Error, proxy not found.'), 400
 423     proxyid = rv[0].get("id")
 424     rv = get_db().prepare("SELECT id FROM proxy WHERE voter_id=$1 AND revoked is NULL")(voterid);
 425     if len(rv) != 0:
 426         return _('Error, proxy allready given.'), 400
 427     rv = get_db().prepare("SELECT COUNT(id) as c FROM proxy WHERE proxy_id=$1 AND revoked is NULL GROUP BY proxy_id")(proxyid);
 428     if len(rv) != 0:
 429         if rv[0].get("c") is None or rv[0].get("c") >= max_proxy:
 430             return _("Error, Max proxy for '%s' reached.") % (proxy), 400
 431     rv = get_db().prepare("INSERT INTO proxy(voter_id, proxy_id, granted_by) VALUES ($1,$2,$3)")(voterid, proxyid, g.voter)
 432     return rel_redirect("/proxy")
 433
 434 @app.route("/proxy/revoke", methods=['POST'])
 435 def revoke_proxy():
 436     if not may_admin("proxyadmin"):
 437         return _('Forbidden'), 403
 438     id=request.form.get("id", "")
 439     rv = get_db().prepare("UPDATE proxy SET revoked=CURRENT_TIMESTAMP, revoked_by=$1 WHERE id=$2")(g.voter, int(id))
 440     return rel_redirect("/proxy")
 441
 442 @app.route("/proxy/revokeall", methods=['POST'])
 443 def revoke_proxy_all():
 444     if not may_admin("proxyadmin"):
 445         return _('Forbidden'), 403
 446     rv = get_db().prepare("UPDATE proxy SET revoked=CURRENT_TIMESTAMP, revoked_by=$1 WHERE revoked IS NULL")(g.voter)
 447     return rel_redirect("/proxy")
 448
 449 @app.route("/language/<string:language>")
 450 def set_language(language):
 451     lang.change_language(language)
 452     return rel_redirect("/")
 453
 454 @app.cli.command("create-user")
 455 @click.argument("email")
 456 def create_user(email):
 457     db = get_db()
 458     with db.xact():
 459         rv = db.prepare("SELECT id FROM voter WHERE lower(email)=lower($1)")(email)
 460         messagetext=_("User '%s' already exists.") % (email)
 461         if len(rv) == 0:
 462             db.prepare("INSERT INTO voter(\"email\") VALUES($1)")(email)
 463             messagetext=_("User '%s' inserted.") % (email)
 464     click.echo(messagetext)