U api update

spug_api/apps/home/urls.py

@@ -0,0 +1,10 @@
+from django.urls import path
+
+from .views import *
+
+urlpatterns = [
+    path('statistic/', get_statistic),
+    path('alarm/', get_alarm),
+    path('deploy/', get_deploy),
+    path('request/', get_request),
+]

spug_api/apps/home/views.py

@@ -0,0 +1,49 @@
+from django.db.models import F
+from apps.app.models import App
+from apps.host.models import Host
+from apps.schedule.models import Task
+from apps.monitor.models import Detection
+from apps.alarm.models import Alarm
+from apps.deploy.models import Deploy, DeployRequest
+from libs.utils import json_response, human_date
+from datetime import datetime, timedelta
+import json
+
+
+def get_statistic(request):
+    data = {
+        'app': App.objects.count(),
+        'host': Host.objects.count(),
+        'task': Task.objects.count(),
+        'detection': Detection.objects.count()
+    }
+    return json_response(data)
+
+
+def get_alarm(request):
+    now = datetime.now()
+    data = {human_date(now - timedelta(days=x + 1)): 0 for x in range(14)}
+    for alarm in Alarm.objects.filter(status='1', created_at__gt=human_date(now - timedelta(days=14))):
+        date = alarm.created_at[:10]
+        if date in data:
+            data[date] += 1
+    data = [{'date': k, 'value': v} for k, v in data.items()]
+    return json_response(data)
+
+
+def get_request(request):
+    data = {x.id: {'name': x.name, 'count': 0} for x in App.objects.all()}
+    for req in DeployRequest.objects.filter(created_at__gt=human_date()):
+        data[req.deploy.app_id]['count'] += 1
+    data = sorted(data.values(), key=lambda x: x['count'], reverse=True)[:5]
+    return json_response(data)
+
+
+def get_deploy(request):
+    host = Host.objects.count()
+    data = {x.id: {'name': x.name, 'count': 0} for x in App.objects.all()}
+    for dep in Deploy.objects.all():
+        data[dep.app_id]['count'] += len(json.loads(dep.host_ids))
+    data = filter(lambda x: x['count'], data.values())
+    return json_response({'host': host, 'res': list(data)})
+

spug_api/spug/urls.py

@@ -26,5 +26,6 @@ urlpatterns = [
     path('config/', include('apps.config.urls')),
     path('app/', include('apps.app.urls')),
     path('deploy/', include('apps.deploy.urls')),
+    path('home/', include('apps.home.urls')),
     path('apis/', include('apps.apis.urls')),
 ]