MAD 1 End Term 30 Apr 2023 — Question 24

1from flask import Flask, make_response
2from flask_restful import Resource, Api, reqparse
3from werkzeug.exceptions import HTTPException
4
5app = Flask(__name__)
6api = Api(app)
7
8objects = {
9            "bot101": {"obj_code": "BOT01", "obj_name": "bottles"},
10            "sop109": {"obj_code": "SOP09", "obj_name": "soaps"},
11            "can103": {"obj_code": "CAN03", "obj_name": "candles"}
12          }
13
14to_parse = reqparse.RequestParser()
15to_parse.add_argument("obj_code")
16to_parse.add_argument("obj_name")
17
18class NoObjectError(HTTPException):
19    def __init__(self, status, error):
20        self.response = make_response({"Error": error}, status)
21
22class BadRequest(HTTPException):
23     def __init__(self, status, error):
24        self.response = make_response({"Error": error}, status)
25
26class Objects(Resource):
27    def get(self, id):
28        args = to_parse.parse_args()
29        if id in objects:
30            my_obj = objects[id]
31            if args["obj_code"] == None:
32                raise BadRequest(404, "Object code missing.")
33            if args["obj_name"] == None:
34                raise BadRequest(404, "Object name missing.")
35            else:
36                my_obj["obj_code"] = args["obj_code"]
37                my_obj["obj_name"] = args["obj_name"]
38            return my_obj
39        else:
40            raise NoObjectError(404, "Object doesn't exist in the database.")
41api.add_resource(Objects, "/get_object/<id>")
42
43app.run(debug = True)