From 3c22dcf5e095a8b82d7707cae909749e994cee90 Mon Sep 17 00:00:00 2001 From: jakobsn Date: Tue, 26 Nov 2019 12:59:10 +0100 Subject: [PATCH] Fix bug getting wrong project id upon creation --- src/app/models/project.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/src/app/models/project.py b/src/app/models/project.py index 46e5305..ecdddae 100644 --- a/src/app/models/project.py +++ b/src/app/models/project.py @@ -36,7 +36,9 @@ def set_project(categoryid, userid, project_title, project_description, project_ cursor.execute(query) db.commit() cursor.close() - projectid = get_projects_by_status_and_owner(userid, "open")[-1][0] + users_projects = get_projects_by_owner(userid) + print("Project", users_projects) + projectid = users_projects[-1][0] return projectid def get_project_by_id(projectid): @@ -70,6 +72,13 @@ def get_projects_by_status_and_category(categoryid, project_status): projects = cursor.fetchall() return projects +def get_projects_by_owner(userid): + cursor = db.cursor() + query = ("SELECT * FROM projects WHERE userid = \"" + userid + "\"") + cursor.execute(query) + projects = cursor.fetchall() + return projects + def get_projects_by_status_and_owner(userid, project_status): """ Retrieve all projects owned by a user with a specific status