apijson: fix _filter_owner problem

zhangchunlin · zhangchunlin · commit 7b7ff6afac5d · 2018-12-08T15:34:08.000+08:00
diff --git a/uliweb_apijson/apijson/views.py b/uliweb_apijson/apijson/views.py
@@ -9,17 +9,18 @@
 @expose('/apijson')
 class ApiJson(object):
     def __begin__(self):
+        self.rdict = {
+            "code":200,
+            "msg":"success"
+        }
+
         try:
             self.request_data = loads(request.data)
         except Exception as e:
             log.error("try to load json but get exception: '%s', request data: %s"%(e,request.data))
             return json({"code":400,"msg":"not json data in the request"})
     
     def get(self):
-        self.rdict = {
-            "code":200,
-            "msg":"success"
-        }
         for key in self.request_data:
             if key[-2:]=="[]":
                 rsp = self._query_array(key)
@@ -44,7 +45,7 @@ def _query_one(self,key):
         if not public:
             if not request.user:
                 return json({"code":401,"msg":"'%s' not accessable for unauthorized request"%(modelname)})
-            owner_filtered,q = self._filter_owner(model,q)
+            owner_filtered,q = self._filter_owner(model,model_setting,q)
             if owner_filtered:
                 filtered = True
             else:
@@ -157,7 +158,7 @@ def _query_array(self,key):
         if not public:
             if not request.user:
                 return json({"code":401,"msg":"'%s' not accessable for unauthorized request"%(modelname)})
-            owner_filtered,q = self._filter_owner(model,q)
+            owner_filtered,q = self._filter_owner(model,model_setting,q)
             if not owner_filtered:
                 return json({"code":401,"msg":"'%s' not accessable because not public"%(modelname)})
 
@@ -175,7 +176,7 @@ def _get_info(i):
         l = [_get_info(i) for i in q]
         self.rdict[key] = l
 
-    def _filter_owner(model,q):
+    def _filter_owner(self,model,model_setting,q):
         owner_filtered = False
         if hasattr(model,"owner_condition"):
             q = q.filter(model.owner_condition())
@@ -186,3 +187,6 @@ def _filter_owner(model,q):
                 q = q.filter(getattr(model.c,user_id_field)==request.user.id)
                 owner_filtered = True
         return owner_filtered,q
+
+    def post(self):
+        return json(self.rdict)
