times = ConfigProxy("DURATION")
debuguser = ConfigProxy("DEBUGUSER")
+max_proxy=app.config.get("MAX_PROXY")
+
@app.before_request
def lookup_user():
global prefix
db.prepare("INSERT INTO voter(\"email\") VALUES($1)")(user)
rv = db.prepare("SELECT id FROM voter WHERE email=$1")(user)
g.voter = rv[0].get("id");
+ 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)
+ if len(rv) != 0:
+ g.proxies_given = rv[0].get("email")
+ 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)
+ if len(rv) != 0:
+ sep = ""
+ g.proxies_received = ""
+ for x in range(0, len(rv)):
+ g.proxies_received += sep + rv[x].get("email")
+ sep =", "
+
g.user = user
g.roles = {}
g.roles[a[0]] = [group for group in prefix.per_host]
else:
g.roles[a[0]].append(val)
+
return None
@app.context_processor
def may(action, motion):
return motion in get_allowed_cats(action)
+def may_admin(action):
+ return action in g.roles
+
+def get_voters():
+ rv = get_db().prepare("SELECT email FROM voter")
+ return rv
+
+def get_all_proxies():
+ 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")
+ return rv
+
@app.teardown_appcontext
def close_connection(exception):
db = getattr(g, '_database', None)
db.prepare("ALTER TABLE \"motion\" ALTER COLUMN \"host\" SET NOT NULL")()
db.prepare("UPDATE \"schema_version\" SET \"version\"=3")()
+ if ver < 4:
+ with app.open_resource('sql/from_3.sql', mode='r') as f:
+ db.execute(f.read())
+ db.prepare("UPDATE \"schema_version\" SET \"version\"=4")()
+
init_db()
+
@app.route("/")
def main():
start=int(request.args.get("start", "-1"));
else:
prev = -1
return render_template('index.html', motions=rv[:10], more=rv[10]["id"] if len(rv) == 11 else None, times=times.per_host, prev=prev,
- categories=get_allowed_cats("create"), singlemotion=False)
+ categories=get_allowed_cats("create"), singlemotion=False, may_proxyadmin=may_admin("proxyadmin"))
def rel_redirect(loc):
r = redirect(loc)
votes = None
if may("audit", rv[0].get("type")) and not rv[0].get("running") and not rv[0].get("canceled"):
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"));
- 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)
+ 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"))
@app.route("/motion/<string:motion>/vote", methods=['POST'])
@validate_motion_access('vote')
else:
db.prepare("UPDATE vote SET result=$3, entered=CURRENT_TIMESTAMP WHERE motion_id=$1 AND voter_id = $2")(id, g.voter, v)
return motion_edited(id)
+
+@app.route("/proxy")
+def proxy():
+ if not may_admin("proxyadmin"):
+ return "Forbidden", 403
+ return render_template('proxy.html', voters=get_voters(), proxies=get_all_proxies(), may_proxyadmin=may_admin("proxyadmin"))
+
+@app.route("/proxy/add", methods=['POST'])
+def add_proxy():
+ if not may_admin("proxyadmin"):
+ return "Forbidden", 403
+ voter=request.form.get("voter", "")
+ proxy=request.form.get("proxy", "")
+ if voter == proxy :
+ return "Error, voter equals proxy.", 400
+ rv = get_db().prepare("SELECT id FROM voter WHERE email=$1")(voter);
+ if len(rv) == 0:
+ return "Error, voter not found.", 400
+ voterid = rv[0].get("id")
+ rv = get_db().prepare("SELECT id FROM voter WHERE email=$1")(proxy);
+ if len(rv) == 0:
+ return "Error, proxy not found.", 400
+ proxyid = rv[0].get("id")
+ rv = get_db().prepare("SELECT id FROM proxy WHERE voter_id=$1 AND revoked is NULL")(voterid);
+ if len(rv) != 0:
+ return "Error, proxy allready given.", 400
+ rv = get_db().prepare("SELECT COUNT(id) as c FROM proxy WHERE proxy_id=$1 AND revoked is NULL GROUP BY proxy_id")(proxyid);
+ if len(rv) != 0:
+ if rv[0].get("c") >= max_proxy:
+ return "Error, Max proxy for '" + proxy + "' reached.", 400
+ rv = get_db().prepare("INSERT INTO proxy(voter_id, proxy_id, granted_by) VALUES ($1,$2,$3)")(voterid, proxyid, g.voter)
+ return rel_redirect("/proxy")
+
+@app.route("/proxy/revoke", methods=['POST'])
+def revoke_proxy():
+ if not may_admin("proxyadmin"):
+ return "Forbidden", 403
+ id=request.form.get("id", "")
+ rv = get_db().prepare("UPDATE proxy SET revoked=CURRENT_TIMESTAMP, revoked_by=$1 WHERE id=$2")(g.voter, int(id))
+ return rel_redirect("/proxy")
+
+@app.route("/proxy/revokeall", methods=['POST'])
+def revoke_proxy_all():
+ if not may_admin("proxyadmin"):
+ return "Forbidden", 403
+ rv = get_db().prepare("UPDATE proxy SET revoked=CURRENT_TIMESTAMP, revoked_by=$1 WHERE revoked IS NULL")(g.voter)
+ return rel_redirect("/proxy")
+
DEBUGUSER = {},
GROUP_PREFIX = {'127.0.0.1:5000': {'group1': 'g1', 'group2': 'g2'}},
DURATION = {'127.0.0.1:5000':[3, 7, 14]},
- SERVER_NAME = '127.0.0.1:5000'
+ SERVER_NAME = '127.0.0.1:5000',
+ MAX_PROXY=2
)
app.config['TESTING'] = True
environ_base={'USER_ROLES': user}
)
+ def addProxy(self, user, voter, proxy):
+ return self.app.post(
+ '/proxy/add',
+ environ_base={'USER_ROLES': user},
+ data=dict(voter=voter, proxy=proxy)
+ )
+
+ def revokeProxy(self, user, id):
+ return self.app.post(
+ '/proxy/revoke',
+ environ_base={'USER_ROLES': user},
+ data=dict(id=id)
+ )
+
def buildResultText(self, motiontext, yes, no, abstain):
return '<p>'+motiontext+'</p></p>\n <p>\nYes <span class=\"badge badge-pill badge-secondary\">'+str(yes)+'</span><br>'\
+ '\nNo <span class=\"badge badge-pill badge-secondary\">'+str(no)+'</span><br>'\
+ '\nNo <span class=\"badge badge-pill badge-secondary\">1</span><br>'\
+ '\nAbstain <span class=\"badge badge-pill badge-secondary\">0</span><br>\n </p>\n </div>\n</div>\n</div>'
self.assertIn(str.encode(testtext), result.data)
+ testtext= 'Proxy management'
+ self.assertNotIn(str.encode(testtext), result.data)
# start with second motion
result = self.app.get('/', environ_base={'USER_ROLES': user}, query_string=dict(start=2))
self.assertEqual(result.status_code, 404)
self.assertIn(str.encode('Error, Not found'), result.data)
+ def test_no_proxy(self):
+ result = self.app.get('proxy', environ_base={'USER_ROLES': user}, follow_redirects=True)
+ self.assertEqual(result.status_code, 403)
+ self.assertIn(str.encode('Forbidden'), result.data)
+ def test_no_proxy_add(self):
+ result = self.app.post('proxy/add', environ_base={'USER_ROLES': user}, follow_redirects=True)
+ self.assertEqual(result.status_code, 403)
+ self.assertIn(str.encode('Forbidden'), result.data)
+
+ def test_no_proxy_revoke(self):
+ result = self.app.post('proxy/revoke', environ_base={'USER_ROLES': user}, follow_redirects=True)
+ self.assertEqual(result.status_code, 403)
+ self.assertIn(str.encode('Forbidden'), result.data)
+
+ def test_no_proxy_revokeAll(self):
+ result = self.app.post('proxy/revokeall', environ_base={'USER_ROLES': user}, follow_redirects=True)
+ self.assertEqual(result.status_code, 403)
+ self.assertIn(str.encode('Forbidden'), result.data)
+
class VoterTests(BasicTest):
def setUp(self):
+ '\n <div>User C: no</div>\n </div>\n</div>\n<a href="/?start=2#motion-2" class="btn btn-primary">Back</a>'
self.assertIn(str.encode(testtext), result.data)
+class ProxyManagementTests(BasicTest):
+
+ def setUp(self):
+ self.init_test()
+ global user
+ user='testuser/proxyadmin:*'
+ self.db_sampledata()
+
+ def tearDown(self):
+ pass
+
+ def test_see_proxy(self):
+ result = self.app.get('proxy', environ_base={'USER_ROLES': user}, follow_redirects=True)
+ testtext= 'div class="container">\n<form action="/proxy/add" method="POST">'
+ self.assertIn(str.encode(testtext), result.data)
+ testtext= 'proxy granted to:'
+ self.assertNotIn(str.encode(testtext), result.data)
+ testtext= 'holds proxy of:'
+ self.assertNotIn(str.encode(testtext), result.data)
+ testtext= '<select class="float form-control" name="voter">\n '\
+ + '<option>User A</option>\n <option>User B</option>\n '\
+ + '<option>User C</option>\n '\
+ + '<option>testuser</option>\n '\
+ + '</select>\n'
+ self.assertIn(str.encode(testtext), result.data)
+ testtext= '<select class="float form-control" name="proxy">\n '\
+ + '<option>User A</option>\n '\
+ + '<option>User B</option>\n '\
+ + '<option>User C</option>\n '\
+ + '<option>testuser</option>\n '\
+ + '</select>\n'
+ self.assertIn(str.encode(testtext), result.data)
+ testtext= '<table>\n '\
+ + '<thead>\n '\
+ + '<th>Voter</th>\n <th>Proxy</th>\n <th></th>\n </thead>\n '\
+ + '</table>\n'
+ self.assertIn(str.encode(testtext), result.data)
+ testtext= '<a class="nav-link" href="/proxy">Proxy management</a>'
+ self.assertIn(str.encode(testtext), result.data)
+
+ def test_add_proxy(self):
+ voter=''
+ proxy=''
+ response = self.addProxy(user, voter, proxy)
+ self.assertEqual(response.status_code, 400)
+ self.assertIn(str.encode('Error, voter equals proxy.'), response.data)
+
+ voter='User A'
+ response = self.addProxy(user, voter, proxy)
+ self.assertEqual(response.status_code, 400)
+ self.assertIn(str.encode('Error, proxy not found.'), response.data)
+
+ voter='User Z'
+ response = self.addProxy(user, voter, proxy)
+ self.assertEqual(response.status_code, 400)
+ self.assertIn(str.encode('Error, voter not found.'), response.data)
+
+ voter=''
+ proxy='User B'
+ response = self.addProxy(user, voter, proxy)
+ self.assertEqual(response.status_code, 400)
+ self.assertIn(str.encode('Error, voter not found.'), response.data)
+
+ voter='User B'
+ proxy='User B'
+ response = self.addProxy(user, voter, proxy)
+ self.assertEqual(response.status_code, 400)
+ self.assertIn(str.encode('Error, voter equals proxy.'), response.data)
+
+ voter='User A'
+ proxy='User B'
+ response = self.addProxy(user, voter, proxy)
+ self.assertEqual(response.status_code, 302)
+ result = self.app.get('proxy', environ_base={'USER_ROLES': user}, follow_redirects=True)
+ testtext= '<form action="/proxy/revoke" method="POST">'
+ self.assertIn(str.encode(testtext), result.data)
+ testtext= '<table>\n '\
+ + '<thead>\n '\
+ + '<th>Voter</th>\n '\
+ + '<th>Proxy</th>\n <th></th>\n </thead>\n '\
+ + '<tr>\n <td>User A</td>\n <td>User B</td>\n '\
+ + '<td><button type="submit" class="btn btn-danger" name="id" value="1">Revoke</button></td>\n '\
+ + '</tr>\n </table>\n'
+ self.assertIn(str.encode(testtext), result.data)
+
+ response = self.addProxy(user, voter, proxy)
+ self.assertEqual(response.status_code, 400)
+ self.assertIn(str.encode('Error, proxy allready given.'), response.data)
+
+ voter='User A'
+ proxy='User C'
+ response = self.addProxy(user, voter, proxy)
+ self.assertEqual(response.status_code, 400)
+ self.assertIn(str.encode('Error, proxy allready given.'), response.data)
+
+ voter='User C'
+ proxy='User B'
+ response = self.addProxy(user, voter, proxy)
+ self.assertEqual(response.status_code, 302)
+ result = self.app.get('proxy', environ_base={'USER_ROLES': user}, follow_redirects=True)
+ testtext= '<table>\n '\
+ + '<thead>\n '\
+ + '<th>Voter</th>\n '\
+ + '<th>Proxy</th>\n <th></th>\n </thead>\n '\
+ + '<tr>\n <td>User A</td>\n <td>User B</td>\n '\
+ + '<td><button type="submit" class="btn btn-danger" name="id" value="1">Revoke</button></td>\n </tr>\n '\
+ + '<tr>\n <td>User C</td>\n <td>User B</td>\n '\
+ + '<td><button type="submit" class="btn btn-danger" name="id" value="2">Revoke</button></td>\n '\
+ + '</tr>\n </table>\n'
+ self.assertIn(str.encode(testtext), result.data)
+ testtext= 'proxy granted to:'
+ self.assertNotIn(str.encode(testtext), result.data)
+ testtext= 'holds proxy of:'
+ self.assertNotIn(str.encode(testtext), result.data)
+
+ voter='testuser'
+ proxy='User B'
+ response = self.addProxy(user, voter, proxy)
+ self.assertEqual(response.status_code, 400)
+ self.assertIn(str.encode('Error, Max proxy for \'User B\' reached.'), response.data)
+
+ voter='testuser'
+ proxy='User A'
+ response = self.addProxy(user, voter, proxy)
+ self.assertEqual(response.status_code, 302)
+ result = self.app.get('proxy', environ_base={'USER_ROLES': user}, follow_redirects=True)
+ testtext= '<table>\n '\
+ + '<thead>\n '\
+ + '<th>Voter</th>\n <th>Proxy</th>\n <th></th>\n </thead>\n '\
+ + '<tr>\n <td>testuser</td>\n <td>User A</td>\n '\
+ + '<td><button type="submit" class="btn btn-danger" name="id" value="3">Revoke</button></td>\n </tr>\n '\
+ + '<tr>\n <td>User A</td>\n <td>User B</td>\n '\
+ + '<td><button type="submit" class="btn btn-danger" name="id" value="1">Revoke</button></td>\n </tr>\n '\
+ + '<tr>\n <td>User C</td>\n <td>User B</td>\n '\
+ + '<td><button type="submit" class="btn btn-danger" name="id" value="2">Revoke</button></td>\n '\
+ + '</tr>\n </table>\n'
+ self.assertIn(str.encode(testtext), result.data)
+ testtext= 'proxy granted to: User A\n'
+ self.assertIn(str.encode(testtext), result.data)
+ testtext= 'holds proxy of:'
+ self.assertNotIn(str.encode(testtext), result.data)
+
+ voter='User B'
+ proxy='testuser'
+ response = self.addProxy(user, voter, proxy)
+ self.assertEqual(response.status_code, 302)
+ result = self.app.get('proxy', environ_base={'USER_ROLES': user}, follow_redirects=True)
+ testtext= '<table>\n '\
+ + '<thead>\n '\
+ + '<th>Voter</th>\n <th>Proxy</th>\n <th></th>\n </thead>\n '\
+ + '<tr>\n <td>testuser</td>\n <td>User A</td>\n '\
+ + '<td><button type="submit" class="btn btn-danger" name="id" value="3">Revoke</button></td>\n </tr>\n '\
+ + '<tr>\n <td>User A</td>\n <td>User B</td>\n '\
+ + '<td><button type="submit" class="btn btn-danger" name="id" value="1">Revoke</button></td>\n </tr>\n '\
+ + '<tr>\n <td>User B</td>\n <td>testuser</td>\n '\
+ + '<td><button type="submit" class="btn btn-danger" name="id" value="4">Revoke</button></td>\n </tr>\n '\
+ + '<tr>\n <td>User C</td>\n <td>User B</td>\n '\
+ + '<td><button type="submit" class="btn btn-danger" name="id" value="2">Revoke</button></td>\n '\
+ + '</tr>\n </table>\n'
+ self.assertIn(str.encode(testtext), result.data)
+ testtext= 'proxy granted to: User A\n'
+ self.assertIn(str.encode(testtext), result.data)
+ testtext= 'holds proxy of: User B\n'
+ self.assertIn(str.encode(testtext), result.data)
+
+ response = self.revokeProxy(user, 4)
+ self.assertEqual(response.status_code, 302)
+ result = self.app.get('proxy', environ_base={'USER_ROLES': user}, follow_redirects=True)
+ testtext= '<table>\n '\
+ + '<thead>\n '\
+ + '<th>Voter</th>\n <th>Proxy</th>\n <th></th>\n </thead>\n '\
+ + '<tr>\n <td>testuser</td>\n <td>User A</td>\n '\
+ + '<td><button type="submit" class="btn btn-danger" name="id" value="3">Revoke</button></td>\n </tr>\n '\
+ + '<tr>\n <td>User A</td>\n <td>User B</td>\n '\
+ + '<td><button type="submit" class="btn btn-danger" name="id" value="1">Revoke</button></td>\n </tr>\n '\
+ + '<tr>\n <td>User C</td>\n <td>User B</td>\n '\
+ + '<td><button type="submit" class="btn btn-danger" name="id" value="2">Revoke</button></td>\n '\
+ + '</tr>\n </table>\n'
+ self.assertIn(str.encode(testtext), result.data)
+ testtext= 'proxy granted to: User A\n'
+ self.assertIn(str.encode(testtext), result.data)
+ testtext= 'holds proxy of:'
+ self.assertNotIn(str.encode(testtext), result.data)
+
+ response = self.revokeProxy(user, 3)
+ self.assertEqual(response.status_code, 302)
+ result = self.app.get('proxy', environ_base={'USER_ROLES': user}, follow_redirects=True)
+ testtext= '<table>\n '\
+ + '<thead>\n '\
+ + '<th>Voter</th>\n <th>Proxy</th>\n <th></th>\n </thead>\n '\
+ + '<tr>\n <td>User A</td>\n <td>User B</td>\n '\
+ + '<td><button type="submit" class="btn btn-danger" name="id" value="1">Revoke</button></td>\n </tr>\n '\
+ + '<tr>\n <td>User C</td>\n <td>User B</td>\n '\
+ + '<td><button type="submit" class="btn btn-danger" name="id" value="2">Revoke</button></td>\n '\
+ + '</tr>\n </table>\n'
+ self.assertIn(str.encode(testtext), result.data)
+ testtext= 'proxy granted to:'
+ self.assertNotIn(str.encode(testtext), result.data)
+ testtext= 'holds proxy of:'
+ self.assertNotIn(str.encode(testtext), result.data)
+
+ result = self.app.post('proxy/revokeall', environ_base={'USER_ROLES': user}, follow_redirects=True)
+ self.assertEqual(response.status_code, 302)
+ result = self.app.get('proxy', environ_base={'USER_ROLES': user}, follow_redirects=True)
+ testtext= '<table>\n '\
+ + '<thead>\n '\
+ + '<th>Voter</th>\n <th>Proxy</th>\n <th></th>\n </thead>\n '\
+ + '</table>\n'
+ self.assertIn(str.encode(testtext), result.data)
+
if __name__ == "__main__":
unittest.main()