Updated code to match ebook's code

Author: LUS24

project/01-first-rest-api/app.py

@@ -1,57 +1,54 @@
-import uuid
 from flask import Flask, request
 
 app = Flask(__name__)
 
-stores = {}
-items = {}
-
-
-@app.get("/item/<string:id>")
-def get_item(id):
-    try:
-        return items[id]
-    except KeyError:
-        return {"message": "Item not found"}, 404
-
-
-@app.post("/item")
-def create_item():
-    request_data = request.get_json()
-    new_item_id = uuid.uuid4().hex
-    new_item = {
-        "name": request_data["name"],
-        "price": request_data["price"],
-        "store_id": request_data["store_id"],
+stores = [
+    {
+        "name": "My Store",
+        "items": [
+            {
+                "name": "Chair",
+                "price": 15.99
+            }
+        ]
     }
-    items[new_item_id] = new_item
-    return new_item
-
-
-@app.get("/item")
-def get_all_items():
-    return {"items": list(items.values())}
+]
 
-
-@app.get("/store/<string:id>")
-def get_store(id):
-    try:
-        # Here you might also want to add the items in this store
-        # We'll do that later on in the course
-        return stores[id]
-    except KeyError:
-        return {"message": "Store not found"}, 404
+@app.get("/store")
+def get_stores():
+    return {"stores": stores}
 
 
 @app.post("/store")
 def create_store():
     request_data = request.get_json()
-    new_store_id = uuid.uuid4().hex
-    new_store = {"id": new_store_id, "name": request_data["name"]}
-    stores[new_store_id] = new_store
+    new_store = {"name": request_data["name"], "items": []}
+    stores.append(new_store)
     return new_store, 201
 
 
-@app.get("/store")
-def get_stores():
-    return {"stores": list(stores.values())}
+@app.post("/store/<string:name>/item")
+def create_item(name):
+    request_data = request.get_json()
+    for store in stores:
+        if store["name"] == name:
+            new_item = {"name": request_data["name"], "price": request_data["price"]}
+            store["items"].append(new_item)
+            return new_item, 201
+    return {"message": "Store not found"}, 404
+
+
+@app.get("/store/<string:name>")
+def get_store(name):
+    for store in stores:
+        if store["name"] == name:
+            return store
+    return {"message": "Store not found"}, 404
+
+
+@app.get("/store/<string:name>/item")
+def get_item_in_store(name):
+    for store in stores:
+        if store["name"] == name:
+            return {"items": store["items"]}
+    return {"message": "Store not found"}, 404

project/02-first-rest-api-docker/app.py

@@ -1,57 +1,54 @@
-import uuid
 from flask import Flask, request
 
 app = Flask(__name__)
 
-stores = {}
-items = {}
-
-
-@app.get("/item/<string:id>")
-def get_item(id):
-    try:
-        return items[id]
-    except KeyError:
-        return {"message": "Item not found"}, 404
-
-
-@app.post("/item")
-def create_item():
-    request_data = request.get_json()
-    new_item_id = uuid.uuid4().hex
-    new_item = {
-        "name": request_data["name"],
-        "price": request_data["price"],
-        "store_id": request_data["store_id"],
+stores = [
+    {
+        "name": "My Store",
+        "items": [
+            {
+                "name": "Chair",
+                "price": 15.99
+            }
+        ]
     }
-    items[new_item_id] = new_item
-    return new_item
-
-
-@app.get("/item")
-def get_all_items():
-    return {"items": list(items.values())}
+]
 
-
-@app.get("/store/<string:id>")
-def get_store(id):
-    try:
-        # Here you might also want to add the items in this store
-        # We'll do that later on in the course
-        return stores[id]
-    except KeyError:
-        return {"message": "Store not found"}, 404
+@app.get("/store")
+def get_stores():
+    return {"stores": stores}
 
 
 @app.post("/store")
 def create_store():
     request_data = request.get_json()
-    new_store_id = uuid.uuid4().hex
-    new_store = {"id": new_store_id, "name": request_data["name"]}
-    stores[new_store_id] = new_store
+    new_store = {"name": request_data["name"], "items": []}
+    stores.append(new_store)
     return new_store, 201
 
 
-@app.get("/store")
-def get_stores():
-    return {"stores": list(stores.values())}
+@app.post("/store/<string:name>/item")
+def create_item(name):
+    request_data = request.get_json()
+    for store in stores:
+        if store["name"] == name:
+            new_item = {"name": request_data["name"], "price": request_data["price"]}
+            store["items"].append(new_item)
+            return new_item, 201
+    return {"message": "Store not found"}, 404
+
+
+@app.get("/store/<string:name>")
+def get_store(name):
+    for store in stores:
+        if store["name"] == name:
+            return store
+    return {"message": "Store not found"}, 404
+
+
+@app.get("/store/<string:name>/item")
+def get_item_in_store(name):
+    for store in stores:
+        if store["name"] == name:
+            return {"items": store["items"]}
+    return {"message": "Store not found"}, 404