add project page

taskmanager/routes.py

@@ -8,20 +8,48 @@ def index():
     tasks = Task.query.all()
     return render_template('index.html', tasks = tasks)
 
-@app.route('/submit', methods=['POST', 'GET'])
-def submit():
+@app.route('/register', methods=['POST', 'GET'])
+def register():
     if request.method == 'GET':
-        return render_template('submit.html')
+        return render_template('register.html')
     elif request.method == 'POST':
-        userinput1 = request.form['forminput1']
-        userinput2 = request.form['forminput2']
-        sqlrow = Table(variable1 = int(userinput1), variable2 = int(userinput2))
+        username = request.form['username']
+        contact = request.form['contact']
+        sqladduser = User(username = username, contact = contact)
         try:
-            db.session.add(sqlrow)
+            db.session.add(sqladduser)
             db.session.commit()
-            return 'Row added'
+            return 'User added'
         except:
-            return 'Adding row to table failed'
+            return 'Adding user failed'
 
     else:
         return 'HTTP request method not recogniezed'
+
+
+@app.route('/projects/<int:task_id>', methods=['GET','POST']) 
+def project(task_id:int):
+    if request.method == 'GET':
+        try:
+            task = Task.query.get(task_id)
+        except:
+            return 'Task not found, bad URL'
+        try:
+            userid = TaskUser.query.filter_by(taskid = task_id).first().userid
+            users = User.query.get(userid).username
+        except:
+            users = "No users added to this task"
+        return render_template("project.html", task = task, users = users)
+    elif request.method == 'POST':
+        username = request.form['username']
+        try:
+            userid = User.query.filter_by(username = username).first().id
+        except:
+            return 'User not found, please <a href="/register">register</a>.'
+        sqladduser = TaskUser(userid = userid, taskid = task_id)
+        try:
+            db.session.add(sqladduser)
+            db.session.commit()
+            return 'User added'
+        except:
+            return 'Adding user failed'