SWCodeG
diff --git a/‎section3_starter_source_code.zip
-6.23 KB b/‎section3_starter_source_code.zip
-6.23 KB
diff --git a/‎section5/starter_code/resources/__init__.py b/‎section5/starter_code/resources/__init__.py
diff --git a/‎section5/starter_code/resources/item.py
Lines changed: 59 additions & 0 deletions b/‎section5/starter_code/resources/item.py
Lines changed: 59 additions & 0 deletions
@@ -0,0 +1,59 @@
+from flask_restful import Resource, reqparse
+from flask_jwt import jwt_required
+from models.item import ItemModel
+
+
+class Item(Resource):
+    parser = reqparse.RequestParser()
+    parser.add_argument('price',
+                        type=float,
+                        required=True,
+                        help="This field cannot be left blank!")
+
+    @jwt_required()
+    def get(self, name):
+        item = ItemModel.find_by_name(name)
+        if item:
+            return item.json()
+        return {'message': 'Item not found'}, 404
+
+    def post(self, name):
+        if ItemModel.find_by_name(name):
+            return {'message': "An item with name '{}' already exists.".format(name)}, 400
+
+        data = Item.parser.parse_args()
+
+        item = ItemModel(name, **data)
+
+        try:
+            item.save_to_db()
+        except:
+            return {"message": "An error occurred inserting the item."}, 500
+
+        return item.json(), 201
+
+    def delete(self, name):
+        item = ItemModel.find_by_name(name)
+        if item:
+            item.delete_from_db()
+
+        return {'message': 'Item deleted'}
+
+    def put(self, name):
+        data = Item.parser.parse_args()
+
+        item = ItemModel.find_by_name(name)
+
+        if item is None:
+            item = ItemModel(name, **data)
+        else:
+            item.price = data['price']
+
+        item.save_to_db()
+
+        return item.json()
+
+
+class ItemList(Resource):
+    def get(self):
+        return {'items': [x.json() for x in ItemModel.query.all()]}