From d2362c5e88a31f10b750c561c761b5cff0148121 Mon Sep 17 00:00:00 2001
From: caoqianming <caoqianming@foxmial.com>
Date: Mon, 22 Mar 2021 22:35:04 +0800
Subject: [PATCH] apicountsafe part1

---
 safesite/viewsn.py | 20 +++++++++++++++++---
 1 file changed, 17 insertions(+), 3 deletions(-)

diff --git a/safesite/viewsn.py b/safesite/viewsn.py
index 24eae68e..3d2febfd 100644
--- a/safesite/viewsn.py
+++ b/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
\ No newline at end of file
+    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)
\ No newline at end of file