Raise 404 when bookmark(s) doesn't exist

This commit is contained in:
David Todd 2020-01-18 21:55:43 -06:00
parent b5a67b1216
commit 74161608df
1 changed files with 7 additions and 1 deletions

View File

@ -10,7 +10,7 @@ import json
import datetime
from pprint import pprint
from uuid import UUID
from bottle import response
from bottle import response, abort
from .db import Database
class API:
@ -56,6 +56,9 @@ class API:
response.content_type = self.response_type
bookmark = self.database.get_bookmark(UUID(bookmark_id))
if bookmark == None:
return abort(404, "Provided bookmark doesn't exist or has been deleted")
# Maybe I need to use `sqlite3.Cursor.description` or `sqlite3.Row.keys()`
bookmark_object = {
'uuid': bookmark[0].hex,
@ -77,6 +80,9 @@ class API:
response.content_type = self.response_type
bookmarks = self.database.get_all_bookmarks()
if len(bookmarks) == 0:
return abort(404, "There are no bookmarks saved")
bookmark_object = [
{
'uuid': bookmark[0].hex,