from django.shortcuts import render
from django.http import HttpResponse, HttpResponseRedirect
from .models import Users, Country
import base64
# Create your views here.


def index(request):
    if request.method == "GET":
        return render(request, 'app/pages/main/login.html')
    elif request.method == "POST":
        return render(request, 'app/pages/main/index.html')


def register(request):
    if request.method == 'POST':
        if request.POST["password1"] == request.POST["password2"]:
            country = Country.objects.get(name=request.POST["country"])
            ecnrypted_pass = base64.b64encode(request.POST["password1"])
            ip = request.META.get('REMOTE_ADDR')
            user = Users.objects.create(user_name=request.POST["username"],
                                        user_surname=request.POST["surname"],
                                        email=request.POST["email"],
                                        password=ecnrypted_pass,
                                        country_name=country,
                                        has_book=False,
                                        can_exchange=False,
                                        books_count=0,
                                        ip_address=ip
                                        )
            user.save()
            return HttpResponseRedirect("/")
        else:
            return HttpResponse("Something gone wrong")
    elif request.method == 'GET':
        countries = Country.objects.all()
        context = {"countries": countries}
        return render(request, 'main/SignUp.html', context)