import logging import geoip2 from django.contrib.gis.geoip2 import GeoIP2 from django.shortcuts import render, redirect from django.urls import reverse from django.db import IntegrityError from django.core.exceptions import ValidationError from django.db.utils import DatabaseError from lib.anonymize_ip import anonymize_ip from .models import Signup logger = logging.getLogger(__name__) def index(request): if request.method == "POST": x_forwarded_for = request.META.get('HTTP_X_FORWARDED_FOR') if x_forwarded_for: ip = x_forwarded_for.split(',')[0] else: ip = request.META.get('REMOTE_ADDR') anonymous_ip = anonymize_ip(ip) try: geoip = GeoIP2().city(anonymous_ip) except geoip2.errors.AddressNotFoundError: geoip = None s = Signup( email=request.POST.get("email"), anonymized_ip=anonymous_ip, user_agent=request.META["HTTP_USER_AGENT"], geoip=geoip, ) try: s.clean_fields() except ValidationError as e: print(e) return render(request, "signup/index.html", {"error_message": e.message}) logging.info("registering email={}".format(request.POST.get("email"))) try: s.save() except IntegrityError: # email already registered, presumably return redirect(reverse("index") + "?success=already_subscribed") except DatabaseError: logger.exception("database error when registering an email") err = "Sorry, database error. Please come back later." return render(request, "signup/index.html", {"error_message": err}) return redirect(reverse("index") + "?success=subscribed") return render(request, "signup/index.html", {})