from django.shortcuts import render
from django.views import View

from country.models import Districts, States

class StateView(View):
    
    def get(self, request):
        states = States.objects.all()
        return render(request, "states.html", {'states': states})
    

class DistrictView(View):
    
    def get(self, request, id):
        districts = Districts.objects.filter(state_id = id)
        return render(request, "districts.html", {'districts': districts})