Skip to content

Instantly share code, notes, and snippets.

@ivancorrales
Last active Apr 26, 2016
Embed
What would you like to do?
# books.py
__author__ = "Iván Corrales Solera <developer@wesovi.com>"
__written_date = "24/04/2016"
import falcon
import json
from database.book import BookDB
class BookResource(object):
def __init__(self):
self.props = ''
self.bookDB = BookDB()
def check_secret_token(req, resp, resource, params):
if req.get_header('X-MyHeader') is None:
raise falcon.HTTPBadRequest('Bad request', 'No valid')
@falcon.before(check_secret_token)
def on_get(self,req,resp):
if req.get_param('id'):
book = self.bookDB.one(int(req.get_param('id')))
if book is None:
raise falcon.HTTPError(falcon.HTTP_404,'Invalid ID','There is not a book with this id')
else:
resp.status = falcon.HTTP_200
resp.body = json.dumps(book.__dict__)
else:
books = self.bookDB.all()
resp.status = falcon.HTTP_200
resp.body = json.dumps([book.__dict__ for book in books])
@falcon.before(check_secret_token)
def on_delete(self,req,resp):
resp.status = falcon.HTTP_200
def on_post(self,req,resp):
try:
raw_json = req.stream.read()
except Exception as ex:
raise falcon.HTTPError(falcon.HTTP_400, 'Error', ex.message)
try:
book = json.loads(raw_json, encoding='utf-8')
except ValueError:
raise falcon.HTTPError(falcon.HTTP_400, 'Invalid JSON','Could not decode the request body. The ''JSON was incorrect.')
print(book)
resp.status = falcon.HTTP_201
resp.body = book
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment