From 869beb6f11bcf151c9f90be36dc7a798d8e2bb2c Mon Sep 17 00:00:00 2001
From: neodarz <neodarz@neodarz.net>
Date: Wed, 10 Jul 2019 11:34:58 +0200
Subject: Add reset function

---
 umosapi/api.py            | 8 ++++++++
 umosapi/app_db/uobject.py | 6 ++++++
 2 files changed, 14 insertions(+)

diff --git a/umosapi/api.py b/umosapi/api.py
index 8a0945f..45bab9c 100644
--- a/umosapi/api.py
+++ b/umosapi/api.py
@@ -98,3 +98,11 @@ class ObjectsEq(Resource):
             "msg": status['msg'],
             "datas": sanitize(status['datas'])
         }, status['code']
+
+
+@api.route("/objects/reset")
+class ObjectsClean(Resource):
+    def get(self):
+        """ Reset collection """
+        uobject.reset()
+        return "done"
diff --git a/umosapi/app_db/uobject.py b/umosapi/app_db/uobject.py
index 60094c7..00aa9b7 100644
--- a/umosapi/app_db/uobject.py
+++ b/umosapi/app_db/uobject.py
@@ -85,3 +85,9 @@ class UObject(object):
                 "datas": datas,
                 "code": 404
             }
+
+    def reset(self):
+        db = MongoDB(self.app)
+        mongo = db.connection()
+        mongo.db.uobjects.remove()
+        return "done"
-- 
cgit v1.2.1