Minor fix

3 years ago · 058eb0d702
parent 1322f5f1ba
commit 058eb0d702
2 changed files with 8 additions and 4 deletions
--- a/wesweb01/todo2021/app.rb
+++ b/wesweb01/todo2021/app.rb
@ -46,10 +46,14 @@ get "/todos/:tid/edit" do
 	db.results_as_hash = true
 	todo = db.execute("SELECT * FROM Todos WHERE id = ?", tid).first  

-	if( todo["user_id"] == session[:id] ) then
-		slim :"todos/edit", locals: {todo: todo}
-	else
-		"401, access denied!"
+	begin
+		if( todo["user_id"] == session[:id] ) then
+			slim :"todos/edit", locals: {todo: todo}
+		else
+			"401, access denied!"
+		end
+	rescue => err
+		"500, TODO not found."
 	end
 end

--- a/wesweb01/todo2021/db/todo.db
+++ b/wesweb01/todo2021/db/todo.db