diff --git a/apiwrapper/views.py b/apiwrapper/views.py index a3cee4a..9454378 100644 --- a/apiwrapper/views.py +++ b/apiwrapper/views.py @@ -7,6 +7,11 @@ CORS(api_wrapper) +@api_wrapper.errorhandler(404) +def page_not_found(e): + return jsonify({'error': 'invalid request'}) + + @api_wrapper.route('/') def index(): di = {'status': 'Up! Good work team'} @@ -26,36 +31,36 @@ def get_courses(term=None, code=None): @api_wrapper.route('/class_description//', strict_slashes=False) -def get_class_description(class_number='-1', term='-1'): - if term == '-1' or class_number == '-1': +def get_class_description(class_number=None, term=None): + if term is None or class_number is None: raise ValueError() return jsonify(course.get_class_description(class_number, term)) @api_wrapper.route('/lab_status/', strict_slashes=False) -def get_lab_status(lab_name='-1'): - if lab_name == '-1': +def get_lab_status(lab_name=None): + if lab_name is None: raise ValueError() return jsonify(lab.get_status(lab_name)) @api_wrapper.route('/laundry_simple/', strict_slashes=False) -def get_laundry_status_simple(loc='-1'): - if loc == '-1': +def get_laundry_status_simple(loc=None): + if loc is None: raise ValueError() return jsonify(laundry.get_status_simple(loc)) @api_wrapper.route('/laundry_detailed/', strict_slashes=False) -def get_laundry_status_detailed(loc='-1'): - if loc == '-1': +def get_laundry_status_detailed(loc=None): + if loc is None: raise ValueError() return jsonify(laundry.get_status_detailed(loc)) -if __name__ == '__main__': +if __name__ is '__main__': api_wrapper.run(debug=True, port=8000)