"""Source API Views"""
from rest_framework.serializers import ModelSerializer
from rest_framework.viewsets import ModelViewSet
from passbook.policies.forms import GENERAL_SERIALIZER_FIELDS
from passbook.policies.reputation.models import ReputationPolicy
class ReputationPolicySerializer(ModelSerializer):
"""Reputation Policy Serializer"""
class Meta:
model = ReputationPolicy
fields = GENERAL_SERIALIZER_FIELDS + ["check_ip", "check_username", "threshold"]
class ReputationPolicyViewSet(ModelViewSet):
"""Source Viewset"""
queryset = ReputationPolicy.objects.all()
serializer_class = ReputationPolicySerializer