From 29847557ce89078d0099a51fd0849c35226b0b14 Mon Sep 17 00:00:00 2001
From: jakobsn <jakob_sn_@hotmail.com>
Date: Wed, 13 Nov 2019 15:06:42 +0100
Subject: [PATCH] remove user field

---
 src/app/models/login.py      |  9 +++++++++
 src/app/models/project.py    |  2 +-
 src/app/views/new_project.py | 21 +++++++++++++++++++--
 3 files changed, 29 insertions(+), 3 deletions(-)

diff --git a/src/app/models/login.py b/src/app/models/login.py
index 7c12600..fc038de 100644
--- a/src/app/models/login.py
+++ b/src/app/models/login.py
@@ -12,6 +12,15 @@ def get_users():
     cursor.close()
     return users
 
+def get_user_id_by_name(username):
+    cursor = db.cursor()
+    query = ("SELECT userid from users WHERE username =\"" + username + "\"")
+    cursor.execute(query)
+    userid = cursor.fetchall()
+    cursor.close()
+    return userid
+
+
 def match_user(username, password):
     """
     Check if user credentials are correct, return if exists
diff --git a/src/app/models/project.py b/src/app/models/project.py
index 45d81bb..57d2164 100644
--- a/src/app/models/project.py
+++ b/src/app/models/project.py
@@ -160,7 +160,7 @@ def get_task_files(taskid):
     cursor.close
     return filenames
 
-def set_project_users(projectid, userid, read_permission="TRUE", 
+def set_project_user(projectid, userid, read_permission="TRUE", 
         write_permission="NULL", modify_permission="NULL"):
     cursor = db.cursor()
     query = ("INSERT INTO project_users VALUES (\"" + project_id + "\", \"" + 
diff --git a/src/app/views/new_project.py b/src/app/views/new_project.py
index 910984a..e69d3bc 100644
--- a/src/app/views/new_project.py
+++ b/src/app/views/new_project.py
@@ -1,6 +1,7 @@
 import web
 from views.forms import get_task_form_elements, get_new_project_form, get_project_form_elements, get_user_form_elements
 import models.project
+import models.login
 from views.utils import get_nav_bar
 
 # Get html templates
@@ -46,19 +47,35 @@ class New_project:
                     project_form = self.compose_form(data, "remove_task")
                     return render.new_project(nav, project_form)     
             except Exception as e:
+
                 try:
                     # Post the form data and save the project in the database
                     if data["Create Project"]:
-                        print("hei")
                         projectid = models.project.set_project(data.category_name, str(session.userid), 
                         data.project_title, data.project_description, "open")
                         task_count = self.get_task_count(data)
+                        user_count = self.get_user_count(data)
                         # Save the tasks in the database
                         for i in range(0, task_count):
                             models.project.set_task(str(projectid), (data["task_title_" + str(i)]), 
                             (data["task_description_" + str(i)]), (data["budget_" + str(i)]))
-                        print("HIHI")
+                        for i in range(0, user_count):
+                            userid = models.login.get_user_id_by_name(data["user_name_"+str(i)])
+                            read, write, modify = "FALSE"
+                            try:
+                                if data["read_permission_"str(i)]:
+                                    read = "TRUE"
+                                except Exception as e:
+                                    try:
+                                        if data["write_permission_"str(i)]:
+                                            write = "TRUE"
+                                    except Exception as e:
+                                        try:
+                                            if data["modify_permission_"str(i)]:
+                                                modify = "TRUE"        
+                            models.project.set_project_user(str(projectid), str(userid), )
                         raise web.seeother('/')
+                        
                 except Exception as e:
                     try:
                         if data["Add User"]: