From 0eeb5b788aed5e2b422ad73886754991a4ef9243 Mon Sep 17 00:00:00 2001
From: neodarz <neodarz@neodarz.net>
Date: Tue, 9 Jul 2019 13:22:07 +0200
Subject: Add patch method
---
umosapi/api.py | 11 +++++++++++
umosapi/app_db/uobject.py | 19 +++++++++++++++++++
2 files changed, 30 insertions(+)
diff --git a/umosapi/api.py b/umosapi/api.py
index f9e90f8..57ae00a 100644
--- a/umosapi/api.py
+++ b/umosapi/api.py
@@ -67,6 +67,17 @@ class Register(Resource):
return {"msg": status['msg']}, status['code']
+@api.route("/objects/<_id>")
+class Update(Resource):
+ @api.expect(user_model)
+ def patch(self, _id):
+ """ Edit an uobject. """
+ args = request.get_json(force=True)
+ status = uobject.update(_id, args.get('datas'))
+
+ return {"msg": status['msg']}, status['code']
+
+
@api.route('/objects/<_id>')
@api.doc(params={'_id': '5d244cc13f3d46cb739912ae'})
class Remove(Resource):
diff --git a/umosapi/app_db/uobject.py b/umosapi/app_db/uobject.py
index df07450..ebb7a9e 100644
--- a/umosapi/app_db/uobject.py
+++ b/umosapi/app_db/uobject.py
@@ -46,3 +46,22 @@ class UObject(object):
mongo.db.uobjects.remove({"_id": ObjectId(_id)})
return {"msg": "UObject {} deleted".format(_id), "code": 200}
return error
+
+ def update(self, _id, datas):
+ db = MongoDB(self.app)
+ mongo = db.connection()
+ error = {}
+
+ if not _id:
+ error = {"msg": "UObject _id is required", "code": 400}
+ elif len(list(mongo.db.uobjects.find({"_id": ObjectId(_id)}))) == 0:
+ error = {
+ "msg": "UObject {} does not exist.".format(_id),
+ "code": 404
+ }
+
+ datas = dict(("datas.{}".format(key), value) for (key, value) in datas.items())
+ if not error:
+ mongo.db.uobjects.update_one({"_id": ObjectId(_id)}, {"$set": datas})
+ return {"msg": "UObject updated.", "code": 200}
+ return error
--
cgit v1.2.1