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