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 import postgresql
   6 import filters
   7
   8 times=[3,5,14]
   9
  10 def get_db():
  11     db = getattr(g, '_database', None)
  12     if db is 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
  15     return db
  16
  17 app = Flask(__name__)
  18 app.register_blueprint(filters.blueprint)
  19
  20 # Load config
  21 app.config.from_pyfile('config.py')
  22
  23 groups=["fellowship", "board"]
  24
  25 @app.before_request
  26 def lookup_user():
  27     env = request.environ
  28     user = None
  29     if "USER_ROLES" in env:
  30         parts = env.get("USER_ROLES").split("/", 1)
  31         user = parts[0]
  32         roles = parts[1]
  33
  34     if "USER" in env and "ROLES" in env:
  35         user = env.get("USER")
  36         roles = env.get("ROLES")
  37
  38     if user is None:
  39         return "Server misconfigured", 500
  40     roles = roles.split(" ")
  41
  42     if user == "<invalid>":
  43         return "Access denied", 403;
  44
  45     db = get_db()
  46     with db.xact():
  47         rv = db.prepare("SELECT id FROM voter WHERE email=$1")(user)
  48         if len(rv) == 0:
  49             db.prepare("INSERT INTO voter(\"email\") VALUES($1)")(user)
  50             rv = db.prepare("SELECT id FROM voter WHERE email=$1")(user)
  51         g.voter = rv[0].get("id");
  52     g.user = user
  53     g.roles = {}
  54
  55     for r in roles:
  56         a = r.split(":", 1)
  57         val = a[1]
  58         if a[0] not in g.roles:
  59             g.roles[a[0]] = []
  60         if val == "*":
  61             g.roles[a[0]] = groups
  62         else:
  63             g.roles[a[0]].append(val)
  64     return None
  65
  66 def get_allowed_cats(action):
  67     return g.roles.get(action, []);
  68
  69 def may(action, motion):
  70     return motion in get_allowed_cats(action)
  71
  72 @app.teardown_appcontext
  73 def close_connection(exception):
  74     db = getattr(g, '_database', None)
  75     if db is not None:
  76         db.close()
  77
  78 def init_db():
  79     with app.app_context():
  80         db = get_db()
  81         try:
  82             ver = db.prepare("SELECT version FROM schema_version")()[0][0];
  83             print("Database Schema version: ", ver)
  84         except postgresql.exceptions.UndefinedTableError:
  85             ver = 0
  86         if ver < 1:
  87             with app.open_resource('sql/schema.sql', mode='r') as f:
  88                 db.execute(f.read())
  89
  90 init_db()
  91
  92 @app.route("/")
  93 def main():
  94     start=int(request.args.get("start", "-1"));
  95     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, "\
  96                              + "COUNT(CASE WHEN result='yes' THEN 'yes' ELSE NULL END) as yes, "\
  97                              + "COUNT(CASE WHEN result='no' THEN 'no' ELSE NULL END) as no, "\
  98                              + "COUNT(CASE WHEN result='abstain' THEN 'abstain' ELSE NULL END) as abstain "\
  99                              + "FROM vote GROUP BY motion_id) as votes ON votes.motion_id=motion.id "\
 100                              + "LEFT JOIN voter poser ON poser.id = motion.posed_by "\
 101                              + "LEFT JOIN voter canceler ON canceler.id = motion.canceled_by "
 102     prev=None
 103     if start == -1:
 104         p = get_db().prepare(q + "ORDER BY motion.id DESC LIMIT 11")
 105         rv = p()
 106     else:
 107         p = get_db().prepare(q + "WHERE motion.id <= $1 ORDER BY motion.id DESC LIMIT 11")
 108         rv = p(start)
 109         rs = get_db().prepare("SELECT id FROM motion WHERE motion.id > $1 ORDER BY id ASC LIMIT 10")(start)
 110         if len(rs) == 10:
 111             prev = rs[9][0]
 112         else:
 113             prev = -1
 114     return render_template('index.html', motions=rv[:10], more=rv[10]["id"] if len(rv) == 11 else None, times=times, prev=prev,
 115                            categories=get_allowed_cats("create"))
 116
 117 def rel_redirect(loc):
 118     r = redirect(loc)
 119     r.autocorrect_location_header = False
 120     return r
 121
 122 @app.route("/motion", methods=['POST'])
 123 def put_motion():
 124     cat=request.form.get("category", "")
 125     if cat not in get_allowed_cats("create"):
 126         return "Forbidden", 403
 127     time = int(request.form.get("days", "3"));
 128     if time not in times:
 129         return "Error, invalid length", 500
 130     p = get_db().prepare("INSERT INTO motion(\"name\", \"content\", \"deadline\", \"posed_by\", \"type\") VALUES($1, $2, CURRENT_TIMESTAMP + $3 * interval '1 days', $4, $5)")
 131     p(request.form.get("title", ""), request.form.get("content",""), time, g.voter, cat)
 132     return rel_redirect("/")
 133
 134 def motion_edited(motion):
 135     return rel_redirect("/?start=" + str(motion) + "#motion-" + str(motion))
 136
 137 @app.route("/motion/<int:motion>/cancel", methods=['POST'])
 138 def cancel_motion(motion):
 139     rv = get_db().prepare("SELECT type FROM motion WHERE id=$1")(motion);
 140     if len(rv) == 0:
 141         return "Error, Not found", 404
 142     if not may("cancel", rv[0].get("type")):
 143         return "Forbidden", 403
 144     if request.form.get("reason", "none") == "none":
 145         return "Error, form requires reason", 500
 146     rv = get_db().prepare("UPDATE motion SET canceled=CURRENT_TIMESTAMP, cancelation_reason=$1, canceled_by=$2 WHERE id=$3 AND canceled is NULL")(request.form.get("reason", ""), g.voter, motion)
 147     return motion_edited(motion)
 148
 149 @app.route("/motion/<int:motion>")
 150 def show_motion(motion):
 151     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 "\
 152                          + "LEFT JOIN vote on vote.motion_id=motion.id AND vote.voter_id=$2 "\
 153                          + "LEFT JOIN voter poser ON poser.id = motion.posed_by "\
 154                          + "LEFT JOIN voter canceler ON canceler.id = motion.canceled_by "
 155                          + "WHERE motion.id=$1")
 156     rv = p(motion, g.voter)
 157     votes = None
 158     if may("audit", rv[0].get("type")) and not rv[0].get("running") and not rv[0].get("canceled"):
 159         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")(motion);
 160     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)
 161
 162 @app.route("/motion/<int:motion>/vote", methods=['POST'])
 163 def vote(motion):
 164     v = request.form.get("vote", "abstain")
 165     db = get_db()
 166     with db.xact():
 167         rv = db.prepare("SELECT type FROM motion WHERE id=$1")(motion);
 168         if len(rv) == 0:
 169             return "Error, Not found", 404
 170         if not may("vote", rv[0].get("type")):
 171             return "Forbidden", 403
 172         p = db.prepare("SELECT deadline > CURRENT_TIMESTAMP FROM motion WHERE id = $1")
 173         if not p(motion)[0][0]:
 174             return "Error, motion deadline has passed", 500
 175         p = db.prepare("SELECT * FROM vote WHERE motion_id = $1 AND voter_id = $2")
 176         rv = p(motion, g.voter)
 177         if len(rv) == 0:
 178             db.prepare("INSERT INTO vote(motion_id, voter_id, result) VALUES($1,$2,$3)")(motion, g.voter, v)
 179         else:
 180             db.prepare("UPDATE vote SET result=$3, entered=CURRENT_TIMESTAMP WHERE motion_id=$1 AND voter_id = $2")(motion, g.voter, v)
 181     return motion_edited(motion)