def put(self):
data = BookList.parser.parse_args()
fetched_book = BookModel.find_by_name(data['name'])
book = BookModel(**data)
if not fetched_book:
book.save_to_db()
else:
fetched_book.price = data['price']
fetched_book.save_to_db()
return book.json(), 200
def post(self):
book = request.get_json()
if not book:
return {'message': 'Please insert book name and price'}, 400
if BookModel.find_by_name(book.get('name')):
return {'message': 'A book with the same name already exists'}, 400
try:
BookModel(**book).save_to_db()
except:
return {'message': 'Something went wrong'}
return book, 201
def __init__(self, application, request, **kwargs):
super(BookHandler, self).__init__(application, request, **kwargs)
self.book_model = BookModel()
