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