Added delete endpoint and tests.

app/dao/services_dao.py

@@ -30,6 +30,11 @@ def save_model_service(service, update_dict=None):
     db.session.commit()
 
 
+def delete_model_service(service):
+    db.session.delete(service)
+    db.session.commit()
+
+
 def get_model_services(service_id=None, user_id=None):
     # TODO need better mapping from function params to sql query.
     if user_id and service_id:

app/dao/users_dao.py

@@ -15,6 +15,11 @@ def save_model_user(usr, update_dict={}):
     db.session.commit()
 
 
+def delete_model_user(user):
+    db.session.delete(user)
+    db.session.commit()
+
+
 def get_model_users(user_id=None):
     if user_id:
         return User.query.filter_by(id=user_id).one()

app/service/views/rest.py

@@ -1,7 +1,8 @@
 from flask import (jsonify, request)
 from sqlalchemy.exc import DataError
 from sqlalchemy.orm.exc import NoResultFound
-from app.dao.services_dao import (save_model_service, get_model_services)
+from app.dao.services_dao import (
+    save_model_service, get_model_services, delete_model_service)
 from app.dao.users_dao import get_model_users
 from app.dao import DAOException
 from .. import service
@@ -26,7 +27,7 @@ def create_service():
 
 
 # TODO auth to be added
-@service.route('/<int:service_id>', methods=['PUT'])
+@service.route('/<int:service_id>', methods=['PUT', 'DELETE'])
 def update_service(service_id):
     try:
         service = get_model_services(service_id=service_id)
@@ -34,19 +35,24 @@ def update_service(service_id):
         return jsonify(result="error", message="Invalid service id"), 400
     except NoResultFound:
         return jsonify(result="error", message="Service not found"), 404
-    # TODO there has got to be a better way to do the next three lines
-    upd_serv, errors = service_schema.load(request.get_json())
-    if errors:
-        return jsonify(result="error", message=errors), 400
-    update_dict, errors = service_schema.dump(upd_serv)
-    # TODO FIX ME
-    # Remove update_service model which is added to db.session
-    db.session.rollback()
-    try:
-        save_model_service(service, update_dict=update_dict)
-    except DAOException as e:
-        return jsonify(result="error", message=str(e)), 400
-    return jsonify(data=service_schema.dump(service).data)
+    if request.method == 'DELETE':
+        status_code = 202
+        delete_model_service(service)
+    else:
+        status_code = 200
+        # TODO there has got to be a better way to do the next three lines
+        upd_serv, errors = service_schema.load(request.get_json())
+        if errors:
+            return jsonify(result="error", message=errors), 400
+        update_dict, errors = service_schema.dump(upd_serv)
+        # TODO FIX ME
+        # Remove update_service model which is added to db.session
+        db.session.rollback()
+        try:
+            save_model_service(service, update_dict=update_dict)
+        except DAOException as e:
+            return jsonify(result="error", message=str(e)), 400
+    return jsonify(data=service_schema.dump(service).data), status_code
 
 
 # TODO auth to be added.

app/user/views/rest.py

@@ -2,7 +2,8 @@ from flask import (jsonify, request)
 from sqlalchemy.exc import DataError
 from sqlalchemy.orm.exc import NoResultFound
 from app.dao.services_dao import get_model_services
-from app.dao.users_dao import (get_model_users, save_model_user)
+from app.dao.users_dao import (
+    get_model_users, save_model_user, delete_model_user)
 from app.schemas import (
     user_schema, users_schema, service_schema, services_schema)
 from .. import user
@@ -20,7 +21,7 @@ def create_user():
 
 
 # TODO auth to be added
-@user.route('/<int:user_id>', methods=['PUT'])
+@user.route('/<int:user_id>', methods=['PUT', 'DELETE'])
 def update_user(user_id):
     try:
         user = get_model_users(user_id=user_id)
@@ -28,16 +29,21 @@ def update_user(user_id):
         return jsonify(result="error", message="Invalid user id"), 400
     except NoResultFound:
         return jsonify(result="error", message="User not found"), 404
-    # TODO there has got to be a better way to do the next three lines
-    update_user, errors = user_schema.load(request.get_json())
-    if errors:
-        return jsonify(result="error", message=errors), 400
-    update_dict, errors = user_schema.dump(update_user)
-    # TODO FIX ME
-    # Remove update_service model which is added to db.session
-    db.session.rollback()
-    save_model_user(user, update_dict=update_dict)
-    return jsonify(data=user_schema.dump(user).data)
+    if request.method == 'DELETE':
+        status_code = 202
+        delete_model_user(user)
+    else:
+        status_code = 200
+        # TODO there has got to be a better way to do the next three lines
+        update_user, errors = user_schema.load(request.get_json())
+        if errors:
+            return jsonify(result="error", message=errors), 400
+        update_dict, errors = user_schema.dump(update_user)
+        # TODO FIX ME
+        # Remove update_service model which is added to db.session
+        db.session.rollback()
+        save_model_user(user, update_dict=update_dict)
+    return jsonify(data=user_schema.dump(user).data), status_code
 
 
 # TODO auth to be added.