from rest_framework import viewsets
from ..models import CreditProduct, LoanApplication, Loan, RepaymentSchedule, Repayment
from .serializers import CreditProductSerializer, LoanApplicationSerializer, LoanSerializer, RepaymentScheduleSerializer, RepaymentSerializer

class CreditProductViewSet(viewsets.ModelViewSet):
    queryset = CreditProduct.objects.all()
    serializer_class = CreditProductSerializer

class LoanApplicationViewSet(viewsets.ModelViewSet):
    queryset = LoanApplication.objects.all()
    serializer_class = LoanApplicationSerializer

class LoanViewSet(viewsets.ModelViewSet):
    queryset = Loan.objects.all()
    serializer_class = LoanSerializer

class RepaymentScheduleViewSet(viewsets.ModelViewSet):
    queryset = RepaymentSchedule.objects.all()
    serializer_class = RepaymentScheduleSerializer

class RepaymentViewSet(viewsets.ModelViewSet):
    queryset = Repayment.objects.all()
    serializer_class = RepaymentSerializer