diff --git a/passbook/core/api/applications.py b/passbook/core/api/applications.py index 52d6a41d3..79a207bf8 100644 --- a/passbook/core/api/applications.py +++ b/passbook/core/api/applications.py @@ -1,11 +1,14 @@ """Application API Views""" from django.db.models import QuerySet +from rest_framework.decorators import action from rest_framework.request import Request from rest_framework.response import Response from rest_framework.serializers import ModelSerializer from rest_framework.viewsets import ModelViewSet from rest_framework_guardian.filters import ObjectPermissionsFilter +from passbook.admin.api.overview_metrics import get_events_per_1h +from passbook.audit.models import EventAction from passbook.core.models import Application from passbook.policies.engine import PolicyEngine @@ -59,3 +62,14 @@ class ApplicationViewSet(ModelViewSet): allowed_applications.append(application) serializer = self.get_serializer(allowed_applications, many=True) return self.get_paginated_response(serializer.data) + + @action(detail=True) + def metrics(self, request: Request, slug: str): + # TODO: Check app read and audit read perms + app = Application.objects.get(slug=slug) + return Response( + get_events_per_1h( + action=EventAction.AUTHORIZE_APPLICATION, + context__authorized_application__pk=app.pk.hex, + ) + ) diff --git a/passbook/flows/api.py b/passbook/flows/api.py index 93e6f6d67..d1aaa71fd 100644 --- a/passbook/flows/api.py +++ b/passbook/flows/api.py @@ -1,17 +1,35 @@ """Flow API Views""" +from django.core.cache import cache from rest_framework.serializers import ModelSerializer, SerializerMethodField from rest_framework.viewsets import ModelViewSet, ReadOnlyModelViewSet from passbook.flows.models import Flow, FlowStageBinding, Stage +from passbook.flows.planner import cache_key class FlowSerializer(ModelSerializer): """Flow Serializer""" + cache_count = SerializerMethodField() + + # pylint: disable=invalid-name + def get_cache_count(self, flow: Flow): + """Get count of cached flows""" + return len(cache.keys(f"{cache_key(flow)}*")) + class Meta: model = Flow - fields = ["pk", "name", "slug", "title", "designation", "stages", "policies"] + fields = [ + "pk", + "name", + "slug", + "title", + "designation", + "stages", + "policies", + "cache_count", + ] class FlowViewSet(ModelViewSet): diff --git a/passbook/policies/api.py b/passbook/policies/api.py index 6a47cd8de..3a67ec42d 100644 --- a/passbook/policies/api.py +++ b/passbook/policies/api.py @@ -1,4 +1,5 @@ """policy API Views""" +import django_filters.rest_framework from django.core.exceptions import ObjectDoesNotExist from rest_framework.serializers import ( ModelSerializer, @@ -62,6 +63,7 @@ class PolicyBindingViewSet(ModelViewSet): queryset = PolicyBinding.objects.all() serializer_class = PolicyBindingSerializer + filterset_fields = ["policy", "target", "enabled", "order", "timeout"] class PolicySerializer(ModelSerializer):