from django.shortcuts import render, redirect
from django.contrib.auth import login
from .forms import CitizenRegistrationForm

def home(request):
    return render(request, 'core/index.html')

def register(request):
    if request.method == 'POST':
        form = CitizenRegistrationForm(request.POST)
        if form.is_valid():
            user = form.save()
            login(request, user)
            return redirect('home')
    else:
        form = CitizenRegistrationForm()
    return render(request, 'core/register.html', {'form': form})