From cb8492f3ecdaabdc751cfe513f31cdf70462e0a3 Mon Sep 17 00:00:00 2001
From: Kostanjevec <kostanjevecj@fdv.uni-lj.si>
Date: Wed, 15 Jun 2022 11:10:05 +0200
Subject: [PATCH] added email form

---
 auth.py                      | 9 ++++++++-
 templates/auth/register.html | 2 ++
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/auth.py b/auth.py
index 4660785..4fc493e 100644
--- a/auth.py
+++ b/auth.py
@@ -1,3 +1,4 @@
+import email
 import functools
 import imp
 
@@ -18,18 +19,24 @@ def register():
     if request.method == 'POST':
         username = request.form['username']
         password = request.form['password']
+        mail     = request.form['email'] 
         error = None
 
+        #@TODO check if this really is an email.
         if not username:
             error = 'Username is required.'
         elif not password:
             error = 'Password is required.'
+        elif not mail:
+            error = 'Please enter your email adress'
         elif dbsession.query(User).filter(User.username == username).first() != None:
             error = "Username already exists, please choose another one"
+        elif dbsession.query(User).filter(User.email == mail).first() != None:
+            error = "This email adress is already in use, please choose another one"
 
         if error is None:
             try:
-                user = User(username=username, password=md5(password.encode("utf-8")).hexdigest())
+                user = User(username=username, password=md5(password.encode("utf-8")).hexdigest(), email=mail)
                 dbsession.add(user)
                 dbsession.commit()
                 dbsession.close()
diff --git a/templates/auth/register.html b/templates/auth/register.html
index a3c73cc..25ccbe5 100644
--- a/templates/auth/register.html
+++ b/templates/auth/register.html
@@ -10,6 +10,8 @@
     <input name="username" id="username" required>
     <label for="password">Password</label>
     <input type="password" name="password" id="password" required>
+    <label for="email">Email</label>
+    <input type="email" name="email" id="email" required>
     <input type="submit" value="Register">
   </form>
 {% endblock %}
\ No newline at end of file