flask-exercise

app.py

@@ -1,5 +1,5 @@
 from typing import Tuple
-
+import json
 from flask import Flask, jsonify, request, Response
 import mockdb.mockdb_interface as db
 
@@ -18,7 +18,6 @@ def create_response(
     IMPORTANT: data must be a dictionary where:
     - the key is the name of the type of data
     - the value is the data itself
-
     :param data <str> optional data
     :param status <int> optional status code, defaults to 200
     :param message <str> optional message
@@ -39,8 +38,11 @@ def create_response(
 """
 ~~~~~~~~~~~~ API ~~~~~~~~~~~~
 """
-
-
+def write_data():
+     new_data = json.dumps({"users" :db.get("users") })
+     with open("mockdb/dummy_data.py" ,'w') as f:
+          f.write( "initial_db_state = " + new_data )
+
 @app.route("/")
 def hello_world():
     return create_response({"content": "hello world!"})
@@ -51,9 +53,56 @@ def mirror(name):
     data = {"name": name}
     return create_response(data)
 
-
 # TODO: Implement the rest of the API here!
+@app.route("/users")
+def get_users():
+     data = db.get('users')
+     team = request.args.get('team')
+     if team == None:
+        return create_response({"users":data}) 
+     data_team = (list(filter(lambda x:x["team"]==team,data)))
+     return create_response({"users":data_team}) 
+
+@app.route("/users/<id>")
+def get_users_by_id(id):
+     data = db.getById("users",int(id))
+     if data == None:
+         return create_response({} , 404 , "User does not exist" )
+     return create_response({"user":data})    
+
+@app.route("/users", methods=['POST'])
+def create_user():
+        data = request.get_json()
+        required_fildes = ["name","age","team"]
+        if not all(filed in data for filed in required_fildes):
+            return create_response({"None":None} , 422 ,"Missing data The object should contain: ID, name and team" )
+        new_user={
+           "name":data["name"], "age": data["age"], "team": data["team"]
+         }
+        res = db.create("users",new_user)
+        write_data()
+        return create_response({"users":res} , 201 ) 
+
+
+@app.route("/users/<id>" , methods=['PUT'] )
+def update_users(id):
+     data = request.get_json()
+     update_user = db.updateById("users", int(id) , data)
+     if update_user == None:
+        return create_response({"users":data} ,404 ,"" ) 
+     write_data()
+     return create_response({"users":update_user})
+
 
+@app.route("/users/<id>" , methods=['DELETE'] )
+def delete_user(id):
+     data = db.getById("users",int(id))
+     if data == None:
+         return create_response({} , 404 , "User does not exist" )
+     db.deleteById("users", int(id) )
+     write_data()
+     return create_response({} , 200 ,'')
+
 """
 ~~~~~~~~~~~~ END API ~~~~~~~~~~~~
 """

conftest.py

@@ -1,7 +1,7 @@
 import pytest
 
 
-@pytest.fixture("session")
+@pytest.fixture(scope = "session")
 def client():
     from app import app
 

mockdb/dummy_data.py

@@ -1,8 +1 @@
-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"},
-    ]
-}
+initial_db_state = {"users": [{"id": 2, "name": "Tim", "age": 20, "team": "LWB"}, {"id": 3, "name": "Varun", "age": 23, "team": "NNB"}, {"id": 4, "name": "Alex", "age": 24, "team": "C2TC"}, {"name": "Ben", "age": 25, "team": "MNB", "id": 5}]}

mockdb/mockdb_interface.py

@@ -29,6 +29,5 @@ def updateById(type, id, update_values):
             item[k] = v
     return item
 
-
 def deleteById(type, id):
     db_state[type] = [i for i in get(type) if i["id"] != id]

test_app.py

@@ -39,3 +39,37 @@ def test_get_user_id(client):
     res_user = res.json["result"]["user"]
     assert res_user["name"] == "Aria"
     assert res_user["age"] == 19
+
+
+def test_create_user(client):
+    new_user ={
+            "name":"Ben",
+            "age": 25,
+            "team":"MNB"
+    }
+    res = client.post("/users", json=new_user)
+    assert res.status_code == 201
+
+    res_user = res.json["result"]["users"]
+    assert res_user["name"] == "Ben"
+    assert res_user["age"] == 25
+
+def test_update_users(client):
+    new_user ={
+            "name":"Beni",
+            "age": 2,
+            "team":"MNB"
+    }
+    res = client.put("/users/1", json=new_user)
+    assert res.status_code == 200
+
+    res_user = res.json["result"]["users"]
+    assert res_user["name"] == "Beni"
+    assert res_user["age"] == 2
+
+def test_delete_user(client):
+    res = client.delete("/users/1")
+    assert res.status_code == 200
+
+
+