Send email to new user
[cavote.git] / main.py
diff --git a/main.py b/main.py
index 22d8c87..51af2a9 100755 (executable)
--- a/main.py
+++ b/main.py
@@ -22,6 +22,7 @@ EMAIL = '"' + TITLE + '"' + ' <' + u"cavote@ffdn.org" + '>'
 BASEURL = "http://localhost:5000"
 VERSION = "cavote 0.0.1"
 SMTP_SERVER = "10.33.33.30"
+PATTERNS = {u'Oui/Non': [u'Oui', u'Non'], u'Oui/Non/Peut-être': [u'Oui', u'Non', u'Peut-être']}
 
 app = Flask(__name__)
 app.config.from_object(__name__)
@@ -32,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):
@@ -87,17 +89,17 @@ def login():
     if request.method == 'POST':
         user = valid_login(request.form['username'], request.form['password'])
         if user is None:
-            flash('Invalid username/password', 'error')
+            flash('Email ou mot de passe invalide.', 'error')
         else:
             connect_user(user)
-            flash('You were logged in', 'success')
+            flash(u'Vous êtes connecté. Bienvenue, %s !' % user['name'], 'success')
             return redirect(url_for('home'))
     return render_template('login.html')
 
 @app.route('/logout')
 def logout():
     disconnect_user()
-    flash('You were logged out', 'info')
+    flash('Vous avez été déconnecté.', 'info')
     return redirect(url_for('home'))
 
 #-----------------
@@ -223,26 +225,56 @@ def admin_user_add():
         abort(401)
     if request.method == 'POST':
         if request.form['email']:
-            # :TODO:maethor:120528: Check fields
-            password = "toto" # :TODO:maethor:120528: Generate password
-            admin = 0
-            if 'admin' in request.form.keys():
-                admin = 1
-            g.db.execute('insert into users (email, name, organization, password, is_admin, key) values (?, ?, ?, ?, ?, "invalid")',
-                    [request.form['email'], request.form['username'], request.form['organization'], password, admin])
-            g.db.commit()
-            user = query_db('select * from users where email = ?', [request.form["email"]], one=True)
-            if user:
-              for group in request.form.getlist('groups'):
-                  if query_db('select id from groups where id = ?', group, one=True) is None:
-                      abort(401)
-                  g.db.execute('insert into user_group values (?, ?)', [user['id'], group])
-                  g.db.commit()
-              # :TODO:maethor:120528: Send mail
-              flash(u'Le nouvel utilisateur a été créé avec succès', 'success')
-              return redirect(url_for('admin_users'))
+            if query_db('select * from users where email=?', [request.form['email']], one=True) is None:
+                if request.form['username']:
+                    if query_db('select * from users where name=?', [request.form['username']], one=True) is None:
+                        password = keygen()
+                        admin = 0
+                        if 'admin' in request.form.keys():
+                            admin = 1
+                        key = keygen()
+                        g.db.execute('insert into users (email, name, organization, password, is_admin, key) values (?, ?, ?, ?, ?, ?)',
+                                [request.form['email'], request.form['username'], request.form['organization'], password, admin, key])
+                        g.db.commit()
+                        user = query_db('select * from users where email = ?', [request.form["email"]], one=True)
+                        if user:
+                            groups = request.form.getlist('groups')
+                            groups.append('1')
+                            for group in groups:
+                                if query_db('select id from groups where id = ?', group, one=True) is None:
+                                    flash(u'Le groupe portant l\'id %s n\'existe pas.' % group, 'warning')
+                                else:
+                                    g.db.execute('insert into user_group values (?, ?)', [user['id'], group])
+                                    g.db.commit()
+                            link = BASEURL + url_for('login_key', userid=user['id'], key=user['key'])
+                            BODY = string.join((
+                            "From: %s" % EMAIL,
+                            "To: %s" % user['email'],
+                            "Subject: [Cavote] Welcome",
+                            "Date: %s" % time.strftime("%a, %d %b %Y %H:%M:%S +0000", time.gmtime()),
+                            "X-Mailer: %s" % VERSION,
+                            "",
+                            "Hi %s!" % user['name'],
+                            "Welcome on %s." % TITLE,
+                            "Your account's adresse is : %s." % user['email'],
+                            "",
+                            "To log in for the first time and set your password, please follow this link :",
+                            link,
+                            ""
+                            ), "\r\n")
+                            server = smtplib.SMTP(SMTP_SERVER)
+                            server.sendmail(EMAIL, [user['email']], BODY)
+                            server.quit()
+                            flash(u'Le nouvel utilisateur a été créé avec succès', 'success')
+                            return redirect(url_for('admin_users'))
+                        else:
+                            flash(u'Une erreur s\'est produite.', 'error')
+                    else:
+                        flash(u'Le nom ' + request.form['username'] + u' est déjà pris ! Veuillez en choisir un autre.', 'error')
+                else:
+                    flash(u"Vous devez spécifier un nom d'utilisateur.", 'error')
             else:
-                flash(u'Une erreur s\'est produite.', 'error')
+                flash(u'Il existe déjà un compte pour cette adresse e-mail : ' + request.form['email'], 'error')
         else:
             flash(u"Vous devez spécifier une adresse email.", 'error')
     groups = query_db('select * from groups where system=0') 
@@ -330,11 +362,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
 
@@ -436,6 +468,11 @@ def admin_vote_add():
                     flash(u'Une erreur est survenue !', 'error')
                     return redirect(url_for('home'))
                 else:
+                    if request.form['pattern'] in PATTERNS.keys():
+                        pattern = PATTERNS[request.form['pattern']]
+                        for choice in pattern:
+                            g.db.execute('insert into choices (name, id_vote) values (?, ?)', [choice, vote['id']])
+                            g.db.commit()
                     flash(u"Le vote a été créé", 'info')
                     return redirect(url_for('admin_vote_edit', voteid=vote['id']))
             else:
@@ -443,7 +480,7 @@ def admin_vote_add():
         else:
             flash(u'Vous devez spécifier un titre.', 'error')
     groups = query_db('select * from groups') 
-    return render_template('admin_vote_new.html', groups=groups)
+    return render_template('admin_vote_new.html', groups=groups, patterns=PATTERNS)
 
 @app.route('/admin/votes/edit/<voteid>', methods=['GET', 'POST'])
 def admin_vote_edit(voteid):