mock db and interface

Author: vrunjeti

app.py

@@ -1,4 +1,5 @@
 from flask import Flask, jsonify, request
+import mockdb.mockdb_interface as db
 
 app = Flask(__name__)
 
@@ -27,18 +28,17 @@ def create_response(data={}, status=200, message=''):
 """
 
 @app.route('/')
-def my_first_route():
+def hello_world():
     return create_response('hello world!')
 
 @app.route('/mirror/<name>')
-def my_second_route(name):
-    return create_response(name)
-
-@app.route('/users', methods=['GET'])
-def get_all_users():
-    if request.method == 'GET':
-        data = { 'users': ['aria', 'tim', 'varun', 'alex'] }
-        return create_response(data)
+def mirror(name):
+    data = {
+        'name': name
+    }
+    return create_response(data)
+
+# TODO: Implement the rest of the API here!
 
 """
 ~~~~~~~~~~~~ END API ~~~~~~~~~~~~

mockdb/dummy_data.py

@@ -0,0 +1,28 @@
+initial_db_state = {
+    'users': [
+        {
+            "id": 1,
+            "name": "Aria",
+            "age": 19,
+            "team": "LWB"
+        },
+        {
+            "id": 2,
+            "name": "Tim",
+            "age": 20,
+            "team": "LWB"
+        },
+        {
+            "id": 3,
+            "name": "Varun",
+            "age": 23,
+            "team": "NNB"
+        },
+        {
+            "id": 4,
+            "name": "Alex",
+            "age": 24,
+            "team": "C2TC"
+        }
+    ]
+}

mockdb/mockdb_interface.py

@@ -0,0 +1,31 @@
+from mockdb.dummy_data import initial_db_state
+import json
+
+db_state = initial_db_state
+
+def get(type):
+    return db_state[type]
+
+def getById(type, id):
+    queried = [i for i in get(type) if i['id'] == id]
+    if (len(queried)):
+        return queried[0]
+    return None
+
+def create(type, payload):
+    last_id = max([i['id'] for i in get(type)])
+    new_id = last_id + 1
+    payload['id'] = new_id
+    db_state[type].append(payload)
+    return payload
+
+def updateById(type, id, update_values):
+    item = getById(type, id)
+    if not item:
+        return None
+    for k, v in update_values.items():
+        item[k] = v
+    return item
+
+def deleteById(type, id):
+    db_state[type] = [i for i in get(type) if i['id'] != id]