Add equal operator and fix bug in remove page from admin

database/database.py

@@ -43,6 +43,12 @@ class VizData(mongo.Document):
     percentage = mongo.FloatField()
     categories = mongo.ListField(mongo.StringField(), default=[])
     detail = mongo.StringField()
+    
+    def __eq__(self, other):
+        return self.id == other.id
+
+    def __str__(self):
+        return "name : " + str(self.name) + " level : " + str(self.level) + " value : " + str(self.value) + " year : " + str(self.year) + " subdata : " + str(self.subdata)
 
 class Page(mongo.Document):
     level = mongo.IntField(required=True)

modules/admin.py

@@ -30,8 +30,16 @@ def remove_page():
     data = request.get_json()
 
     admin = db.User.objects.with_id(data.get("admin_id"))
-    admin.update(pull__page_list = db.VizData.objects.with_id(data.get("page_id")))
+    # admin.update(pull__page_list = db.VizData.objects.with_id(data.get("page_id")))    
+    page = db.VizData.objects.with_id(data.get("page_id"))
     
+    idx = -1
+    for i in range(len(admin.page_list)):
+        if (page == admin.page_list[i]):
+            idx = i
+
+    admin.page_list.pop(idx)
+    admin.save()
     return jsonify({
         "status":200,
         "message":"Page removed"