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 import postgresql
   7 import filters
   8 from flaskext.markdown import Markdown
   9 from markdown.extensions import Extension
  10 from datetime import date, time, datetime
  11
  12 def get_db():
  13     db = getattr(g, '_database', None)
  14     if db is None:
  15         db = g._database = postgresql.open(app.config.get("DATABASE"), user=app.config.get("USER"), password=app.config.get("PASSWORD"))
  16     #db.row_factory = sqlite3.Row
  17     return db
  18
  19 app = Flask(__name__)
  20 app.register_blueprint(filters.blueprint)
  21
  22 class EscapeHtml(Extension):
  23     def extendMarkdown(self, md, md_globals):
  24         del md.preprocessors['html_block']
  25         del md.inlinePatterns['html']
  26
  27 md = Markdown(app, extensions=[EscapeHtml()])
  28
  29 class default_settings(object):
  30     COPYRIGHTSTART="2017"
  31     COPYRIGHTNAME="WPIA"
  32     COPYRIGHTLINK="https://wpia.club"
  33     IMPRINTLINK="https://documents.wpia.club/imprint.html"
  34     DATAPROTECTIONLINK="https://documents.wpia.club/data_privacy_policy_html_pages_en.html"
  35
  36
  37 # Load config
  38 app.config.from_object('motion.default_settings')
  39 app.config.from_pyfile('config.py')
  40
  41
  42 class ConfigProxy:
  43     def __init__(self, name):
  44         self.name = name
  45     @property
  46     def per_host(self):
  47         dict = app.config.get(self.name)
  48         if dict is None:
  49             return None
  50         return dict.get(request.host)
  51
  52 prefix = ConfigProxy("GROUP_PREFIX")
  53 times = ConfigProxy("DURATION")
  54 debuguser = ConfigProxy("DEBUGUSER")
  55
  56 max_proxy=app.config.get("MAX_PROXY")
  57
  58 @app.before_request
  59 def lookup_user():
  60     global prefix
  61
  62     env = request.environ
  63     user = None
  64     my_debuguser = debuguser.per_host
  65     if my_debuguser is not None:
  66         parts = my_debuguser.split("/", 1)
  67         user = parts[0]
  68         roles = parts[1]
  69
  70     if "USER_ROLES" in env:
  71         parts = env.get("USER_ROLES").split("/", 1)
  72         user = parts[0]
  73         roles = parts[1]
  74
  75     if "USER" in env and "ROLES" in env:
  76         user = env.get("USER")
  77         roles = env.get("ROLES")
  78
  79     if user is None:
  80         return "Server misconfigured", 500
  81     roles = roles.split(" ")
  82
  83     if user == "<invalid>":
  84         return "Access denied", 403;
  85
  86     db = get_db()
  87     with db.xact():
  88         rv = db.prepare("SELECT id FROM voter WHERE email=$1")(user)
  89         if len(rv) == 0:
  90             db.prepare("INSERT INTO voter(\"email\") VALUES($1)")(user)
  91             rv = db.prepare("SELECT id FROM voter WHERE email=$1")(user)
  92         g.voter = rv[0].get("id");
  93         rv = db.prepare("SELECT email FROM voter, proxy WHERE proxy.proxy_id = voter.id AND proxy.revoked IS NULL AND proxy.voter_id = $1 ")(g.voter)
  94         if len(rv) != 0:
  95             g.proxies_given = rv[0].get("email")
  96         rv = db.prepare("SELECT email FROM voter, proxy WHERE proxy.voter_id = voter.id AND proxy.revoked IS NULL AND proxy.proxy_id = $1 ")(g.voter)
  97         if len(rv) != 0:
  98             sep = ""
  99             g.proxies_received = ""
 100             for x in range(0, len(rv)):
 101                 g.proxies_received += sep + rv[x].get("email")
 102                 sep =", "
 103
 104     g.user = user
 105     g.roles = {}
 106
 107     for r in roles:
 108         a = r.split(":", 1)
 109         if len(r)!=0:
 110             val = a[1]
 111             if a[0] not in g.roles:
 112                 g.roles[a[0]] = []
 113             if val == "*":
 114                 g.roles[a[0]] = [group for group in prefix.per_host]
 115             else:
 116                 g.roles[a[0]].append(val)
 117
 118     return None
 119
 120 @app.context_processor
 121 def init_footer_variables():
 122     if int(app.config.get("COPYRIGHTSTART"))<datetime.now().year:
 123         version_year = "%s - %s" % (app.config.get("COPYRIGHTSTART"), datetime.now().year)
 124     else:
 125         version_year = datetime.now().year
 126
 127     return dict(
 128         footer = dict( version_year=version_year,
 129             copyright_link=app.config.get("COPYRIGHTLINK"),
 130             copyright_name=app.config.get("COPYRIGHTNAME"),
 131             imprint_link=app.config.get("DATAPROTECTIONLINK"),
 132             dataprotection_link=app.config.get("DATAPROTECTIONLINK")
 133         )
 134     )
 135
 136
 137 def get_allowed_cats(action):
 138     return g.roles.get(action, []);
 139
 140 def may(action, motion):
 141     return motion in get_allowed_cats(action)
 142
 143 def may_admin(action):
 144     return action in g.roles
 145
 146 def get_voters():
 147     rv = get_db().prepare("SELECT email FROM voter")
 148     return rv
 149
 150 def get_all_proxies():
 151     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")
 152     return rv
 153
 154 @app.teardown_appcontext
 155 def close_connection(exception):
 156     db = getattr(g, '_database', None)
 157     if db is not None:
 158         db.close()
 159
 160 def init_db():
 161     with app.app_context():
 162         db = get_db()
 163         try:
 164             ver = db.prepare("SELECT version FROM schema_version")()[0][0];
 165             print("Database Schema version: ", ver)
 166         except postgresql.exceptions.UndefinedTableError:
 167             g._database = None
 168             db = get_db()
 169             ver = 0
 170
 171         if ver < 1:
 172             with app.open_resource('sql/schema.sql', mode='r') as f:
 173                 db.execute(f.read())
 174             return
 175
 176         if ver < 2:
 177             with app.open_resource('sql/from_1.sql', mode='r') as f:
 178                 db.execute(f.read())
 179                 ct={}
 180                 for group in [group for group in prefix[app.config.get("DEFAULT_HOST")]]:
 181                     ct[group] = {"dt": "", "c": 0}
 182
 183                 p = db.prepare("UPDATE \"motion\" SET \"identifier\"=$1 WHERE \"id\"=$2")
 184                 for row in db.prepare("SELECT id, \"type\", \"posed\" FROM \"motion\" ORDER BY \"id\" ASC"):
 185                     dt=row[2].strftime("%Y%m%d")
 186                     if ct[row[1]]["dt"] != dt:
 187                         ct[row[1]]["dt"] = dt
 188                         ct[row[1]]["c"] = 0
 189                     ct[row[1]]["c"] = ct[row[1]]["c"] + 1
 190                     name=prefix[app.config.get("DEFAULT_HOST")][row[1]]+"."+dt+"."+("%03d" % ct[row[1]]["c"])
 191                     p(name, row[0])
 192                 db.prepare("ALTER TABLE \"motion\" ALTER COLUMN \"identifier\" SET NOT NULL")()
 193                 db.prepare("UPDATE \"schema_version\" SET \"version\"=2")()
 194                 db.prepare("CREATE UNIQUE INDEX motion_ident ON motion (identifier)")()
 195
 196         if ver < 3:
 197             with app.open_resource('sql/from_2.sql', mode='r') as f:
 198                 db.execute(f.read())
 199                 db.prepare("UPDATE \"motion\" SET \"host\"=$1")(app.config.get("DEFAULT_HOST"))
 200                 db.prepare("ALTER TABLE \"motion\" ALTER COLUMN \"host\" SET NOT NULL")()
 201                 db.prepare("UPDATE \"schema_version\" SET \"version\"=3")()
 202
 203         if ver < 4:
 204             with app.open_resource('sql/from_3.sql', mode='r') as f:
 205                 db.execute(f.read())
 206                 db.prepare("UPDATE \"schema_version\" SET \"version\"=4")()
 207
 208 init_db()
 209
 210
 211 @app.route("/")
 212 def main():
 213     start=int(request.args.get("start", "-1"));
 214     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, "\
 215                              + "COUNT(CASE WHEN result='yes' THEN 'yes' ELSE NULL END) as yes, "\
 216                              + "COUNT(CASE WHEN result='no' THEN 'no' ELSE NULL END) as no, "\
 217                              + "COUNT(CASE WHEN result='abstain' THEN 'abstain' ELSE NULL END) as abstain "\
 218                              + "FROM vote GROUP BY motion_id) as votes ON votes.motion_id=motion.id "\
 219                              + "LEFT JOIN voter poser ON poser.id = motion.posed_by "\
 220                              + "LEFT JOIN voter canceler ON canceler.id = motion.canceled_by "
 221     prev=None
 222     if start == -1:
 223         p = get_db().prepare(q + "WHERE motion.host = $1 ORDER BY motion.id DESC LIMIT 11")
 224         rv = p(request.host)
 225     else:
 226         p = get_db().prepare(q + "WHERE motion.host = $1 AND motion.id <= $2 ORDER BY motion.id DESC LIMIT 11")
 227         rv = p(request.host, start)
 228         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)
 229         if len(rs) == 10:
 230             prev = rs[9][0]
 231         else:
 232             prev = -1
 233     return render_template('index.html', motions=rv[:10], more=rv[10]["id"] if len(rv) == 11 else None, times=times.per_host, prev=prev,
 234                            categories=get_allowed_cats("create"), singlemotion=False, may_proxyadmin=may_admin("proxyadmin"))
 235
 236 def rel_redirect(loc):
 237     r = redirect(loc)
 238     r.autocorrect_location_header = False
 239     return r
 240
 241 @app.route("/motion", methods=['POST'])
 242 def put_motion():
 243     cat=request.form.get("category", "")
 244     if cat not in get_allowed_cats("create"):
 245         return "Forbidden", 403
 246     time = int(request.form.get("days", "3"));
 247     if time not in times.per_host:
 248         return "Error, invalid length", 400
 249     title=request.form.get("title", "")
 250     title=title.strip()
 251     if title =='':
 252         return "Error, missing title", 400
 253     content=request.form.get("content", "")
 254     content=content.strip()
 255     if content =='':
 256         return "Error, missing content", 400
 257
 258     db = get_db()
 259     with db.xact():
 260         t = db.prepare("SELECT CURRENT_TIMESTAMP")()[0][0];
 261         s = db.prepare("SELECT MAX(\"identifier\") FROM \"motion\" WHERE \"type\"=$1 AND \"host\"=$2 AND DATE(\"posed\")=DATE(CURRENT_TIMESTAMP)")
 262         sr = s(cat, request.host)
 263         ident=""
 264         if len(sr) == 0 or sr[0][0] is None:
 265             ident=prefix.per_host[cat]+"."+t.strftime("%Y%m%d")+".001"
 266         else:
 267             ident=prefix.per_host[cat]+"."+t.strftime("%Y%m%d")+"."+("%03d" % (int(sr[0][0].split(".")[2])+1))
 268         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)")
 269         p(title, content, time, g.voter, cat, ident, request.host)
 270     return rel_redirect("/")
 271
 272 def motion_edited(motion):
 273     return rel_redirect("/?start=" + str(motion) + "#motion-" + str(motion))
 274
 275 def validate_motion_access(privilege):
 276     def decorator(f):
 277         def decorated_function(motion):
 278             db = get_db()
 279             with db.xact():
 280                 rv = db.prepare("SELECT id, type, deadline < CURRENT_TIMESTAMP AS expired, canceled FROM motion WHERE identifier=$1 AND host=$2")(motion, request.host);
 281                 if len(rv) == 0:
 282                     return "Error, Not found", 404
 283                 id = rv[0].get("id")
 284                 if not may(privilege, rv[0].get("type")):
 285                     return "Forbidden", 403
 286                 if rv[0].get("canceled") is not None:
 287                     return "Error, motion was canceled", 403
 288                 if rv[0].get("expired"):
 289                     return "Error, out of time", 403
 290             return f(motion, id)
 291         decorated_function.__name__ = f.__name__
 292         return decorated_function
 293     return decorator
 294
 295 @app.route("/motion/<string:motion>/cancel", methods=['POST'])
 296 @validate_motion_access('cancel')
 297 def cancel_motion(motion, id):
 298     if request.form.get("reason", "none") == "none":
 299         return "Error, form requires reason", 500
 300     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)
 301     return motion_edited(id)
 302
 303 @app.route("/motion/<string:motion>/finish", methods=['POST'])
 304 @validate_motion_access('finish')
 305 def finish_motion(motion, id):
 306     rv = get_db().prepare("UPDATE motion SET deadline=CURRENT_TIMESTAMP WHERE identifier=$1 AND host=$2 AND canceled is NULL")(motion, request.host)
 307     return motion_edited(id)
 308
 309 @app.route("/motion/<string:motion>")
 310 def show_motion(motion):
 311     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 "\
 312                          + "LEFT JOIN vote on vote.motion_id=motion.id AND vote.voter_id=$2 "\
 313                          + "LEFT JOIN voter poser ON poser.id = motion.posed_by "\
 314                          + "LEFT JOIN voter canceler ON canceler.id = motion.canceled_by "
 315                          + "WHERE motion.identifier=$1 AND motion.host=$3")
 316     rv = p(motion, g.voter, request.host)
 317     if len(rv) == 0:
 318         return "Error, Not found", 404
 319     votes = None
 320     if may("audit", rv[0].get("type")) and not rv[0].get("running") and not rv[0].get("canceled"):
 321         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"));
 322     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")), may_finish=may("finish", rv[0].get("type")), votes=votes, singlemotion=True, may_proxyadmin=may_admin("proxyadmin"))
 323
 324 @app.route("/motion/<string:motion>/vote", methods=['POST'])
 325 @validate_motion_access('vote')
 326 def vote(motion, id):
 327     v = request.form.get("vote", "abstain")
 328     db = get_db()
 329     p = db.prepare("SELECT * FROM vote WHERE motion_id = $1 AND voter_id = $2")
 330     rv = p(id, g.voter)
 331     if len(rv) == 0:
 332         db.prepare("INSERT INTO vote(motion_id, voter_id, result) VALUES($1,$2,$3)")(id, g.voter, v)
 333     else:
 334         db.prepare("UPDATE vote SET result=$3, entered=CURRENT_TIMESTAMP WHERE motion_id=$1 AND voter_id = $2")(id, g.voter, v)
 335     return motion_edited(id)
 336
 337 @app.route("/proxy")
 338 def proxy():
 339     if not may_admin("proxyadmin"):
 340         return "Forbidden", 403
 341     return render_template('proxy.html', voters=get_voters(), proxies=get_all_proxies(), may_proxyadmin=may_admin("proxyadmin"))
 342
 343 @app.route("/proxy/add", methods=['POST'])
 344 def add_proxy():
 345     if not may_admin("proxyadmin"):
 346         return "Forbidden", 403
 347     voter=request.form.get("voter", "")
 348     proxy=request.form.get("proxy", "")
 349     if voter == proxy :
 350         return "Error, voter equals proxy.", 400
 351     rv = get_db().prepare("SELECT id FROM voter WHERE email=$1")(voter);
 352     if len(rv) == 0:
 353         return "Error, voter not found.", 400
 354     voterid = rv[0].get("id")
 355     rv = get_db().prepare("SELECT id FROM voter WHERE email=$1")(proxy);
 356     if len(rv) == 0:
 357         return "Error, proxy not found.", 400
 358     proxyid = rv[0].get("id")
 359     rv = get_db().prepare("SELECT id FROM proxy WHERE voter_id=$1 AND revoked is NULL")(voterid);
 360     if len(rv) != 0:
 361         return "Error, proxy allready given.", 400
 362     rv = get_db().prepare("SELECT COUNT(id) as c FROM proxy WHERE proxy_id=$1 AND revoked is NULL GROUP BY proxy_id")(proxyid);
 363     if len(rv) != 0:
 364         if rv[0].get("c") >= max_proxy:
 365             return "Error, Max proxy for '" + proxy + "' reached.", 400
 366     rv = get_db().prepare("INSERT INTO proxy(voter_id, proxy_id, granted_by) VALUES ($1,$2,$3)")(voterid, proxyid, g.voter)
 367     return rel_redirect("/proxy")
 368
 369 @app.route("/proxy/revoke", methods=['POST'])
 370 def revoke_proxy():
 371     if not may_admin("proxyadmin"):
 372         return "Forbidden", 403
 373     id=request.form.get("id", "")
 374     rv = get_db().prepare("UPDATE proxy SET revoked=CURRENT_TIMESTAMP, revoked_by=$1 WHERE id=$2")(g.voter, int(id))
 375     return rel_redirect("/proxy")
 376
 377 @app.route("/proxy/revokeall", methods=['POST'])
 378 def revoke_proxy_all():
 379     if not may_admin("proxyadmin"):
 380         return "Forbidden", 403
 381     rv = get_db().prepare("UPDATE proxy SET revoked=CURRENT_TIMESTAMP, revoked_by=$1 WHERE revoked IS NULL")(g.voter)
 382     return rel_redirect("/proxy")
 383