Added constraints in DB

[cavote.git] / main.py

diff --git a/main.py b/main.py
index 78998b0..7cd53f3 100755 (executable)
--- a/main.py
+++ b/main.py
@@ -33,6 +33,7 @@ def connect_db():
 @app.before_request
 def before_request():
     g.db = connect_db()
+    g.db.execute("PRAGMA foreign_keys = ON")
 
 @app.teardown_request
 def teardown_request(exception):
@@ -342,11 +343,11 @@ def can_vote(idvote, iduser=-1):
     vote = query_db('select * from votes where id=?', [idvote], one=True)
     if vote is None:
         return False
-    if not vote['is_finished']:
+    if vote['is_finished'] == 0:
         if iduser > 0: 
             if can_see_vote(idvote, iduser): 
                 if not has_voted(idvote, iduser):
-                    if query_db('select * from user_group where id_user = ? and id_group = ?', [iduser, vote['id']], one=True):
+                    if query_db('select * from user_group where id_user = ? and id_group = ?', [iduser, vote['id_group']], one=True):
                         return True
     return False