Coverage for demo_server/api/restplus.py: 58%

1# Copyright (c) Microsoft Corporation. 

2# Licensed under the MIT License. 

3 

4import logging 

5import traceback 

6 

7from flask_restplus import Api 

8from demo_server import settings 

9from sqlalchemy.orm.exc import NoResultFound 

10 

11log = logging.getLogger(__name__) 

12 

13api = Api(version='1.0', title='My Blog API', 

14 description='A simple demonstration of a Flask RestPlus powered API') 

15 

16 

17@api.errorhandler 

18def default_error_handler(e): 

19 message = 'An unhandled exception occurred.' 

20 log.exception(message) 

21 

22 if not settings.FLASK_DEBUG: 

23 return {'message': message}, 500 

24 

25 

26@api.errorhandler(NoResultFound) 

27def database_not_found_error_handler(e): 

28 log.warning(traceback.format_exc()) 

29 return {'message': 'A database result was required but none was found.'}, 404