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 config
   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(config.DATABASE, user=config.USER, password=config.PASSWORD)
  14     #db.row_factory = sqlite3.Row
  15     return db
  16
  17 app = Flask(__name__)
  18
  19 @app.teardown_appcontext
  20 def close_connection(exception):
  21     db = getattr(g, '_database', None)
  22     if db is not None:
  23         db.close()
  24
  25 def init_db():
  26     with app.app_context():
  27         db = get_db()
  28         with app.open_resource('schema.sql', mode='r') as f:
  29             db.execute(f.read())
  30
  31 @app.route("/")
  32 def main():
  33     start=int(request.args.get("start", "-1"));
  34     q = "SELECT *, motion.deadline > CURRENT_TIMESTAMP AS running FROM motion LEFT JOIN (SELECT motion_id, voter_id, "\
  35                              + "COUNT(CASE WHEN result='yes' THEN 'yes' ELSE NULL END) as yes, "\
  36                              + "COUNT(CASE WHEN result='no' THEN 'no' ELSE NULL END) as no, "\
  37                              + "COUNT(CASE WHEN result='abstain' THEN 'abstain' ELSE NULL END) as abstain "\
  38                              + "FROM vote GROUP BY motion_id, voter_id) as votes ON votes.motion_id=motion.id "
  39     prev=None
  40     if start == -1:
  41         p = get_db().prepare(q + "ORDER BY id DESC LIMIT 11")
  42         rv = p()
  43     else:
  44         p = get_db().prepare(q + "WHERE id <= $1 ORDER BY id DESC LIMIT 11")
  45         rv = p(start)
  46         rs = get_db().prepare("SELECT id FROM motion WHERE id > $1 ORDER BY id ASC LIMIT 10")(start)
  47         if len(rs) == 10:
  48             prev = rs[9][0]
  49         else:
  50             prev = -1
  51     return render_template('index.html', motions=rv[:10], more=rv[10]["id"] if len(rv) == 11 else None, times=times, prev=prev)
  52
  53 @app.route("/motion", methods=['POST'])
  54 def put_motion():
  55     time = int(request.form.get("days", "3"));
  56     if time not in times:
  57         return "Error, invalid length"
  58     p = get_db().prepare("INSERT INTO motion(\"name\", \"content\", \"deadline\") VALUES($1, $2, CURRENT_TIMESTAMP + $3 * interval '1 days')")
  59     p(request.form.get("title", ""), request.form.get("content",""), time)
  60     return redirect("/")
  61
  62 voter=1
  63
  64 @app.route("/motion/<int:id>")
  65 def show_motion(id):
  66     p = get_db().prepare("SELECT motion.*, motion.deadline > CURRENT_TIMESTAMP AS running, vote.result FROM motion LEFT JOIN vote on vote.motion_id=motion.id AND vote.voter_id=$2 WHERE id=$1")
  67     rv = p(id,voter)
  68     return render_template('single_motion.html', motion=rv[0])
  69
  70 @app.route("/motion/<int:motion>/vote", methods=['POST'])
  71 def vote(motion):
  72     v = request.form.get("vote", "abstain")
  73     db = get_db()
  74     with db.xact():
  75         p = db.prepare("SELECT deadline > CURRENT_TIMESTAMP FROM motion WHERE id = $1")
  76         if not p(motion)[0][0]:
  77             return "Error, motion deadline has passed"
  78         p = db.prepare("SELECT * FROM vote WHERE motion_id = $1 AND voter_id = $2")
  79         rv = p(motion, voter)
  80         if len(rv) == 0:
  81             db.prepare("INSERT INTO vote (motion_id, voter_id, result) VALUES($1,$2,$3)")(motion,voter,v)
  82         else:
  83             db.prepare("UPDATE vote SET result=$3, entered=CURRENT_TIMESTAMP WHERE motion_id=$1 AND voter_id = $2")(motion,voter,v)
  84     return redirect("/?start=" + str(motion) + "#motion-" + str(motion))
  85
  86 # TODO cancel running motion (with comment)
  87 # TODO authentication/user management
  88 # TODO crop time at second precision