apicountsafe part1

safesite/viewsn.py

@@ -1,5 +1,19 @@
+from .models import *
+from .views import apicheck_login
+from django.http import JsonResponse
 
-
-
+@apicheck_login
 def apicountsafe(req):
-    pass
+    type = req.GET.get('type')
+    userid = req.session['userid']
+    if req.GET.get('userid', None):
+        userid = req.GET.get('userid')
+    if type == 'trouble':
+        objs = Trouble.objects.filter(deletemark=1)
+        ret = {}
+        if req.GET.get('dept', None):
+            pass
+        else:
+            ret['fx_count'] = objs.filter(fxr__userid=userid).count()
+            ret['zg_count'] = objs.filter(zgr__userid=userid).count()
+        return JsonResponse(ret)