31 lines
832 B
Python
31 lines
832 B
Python
import geoip2
|
|
|
|
from django.db import models
|
|
from django.contrib.gis.geoip2 import GeoIP2
|
|
|
|
class Signup(models.Model):
|
|
email = models.EmailField()
|
|
created_at = models.DateTimeField(auto_now_add = True)
|
|
anonymized_ip = models.GenericIPAddressField()
|
|
user_agent = models.CharField(max_length = 255)
|
|
|
|
class Meta:
|
|
constraints = [
|
|
models.UniqueConstraint(name = "unique_email", fields = ["email"]),
|
|
]
|
|
|
|
def __str__(self):
|
|
return self.email
|
|
|
|
def _country(self):
|
|
try:
|
|
return GeoIP2().country(self.anonymized_ip)
|
|
except geoip2.errors.AddressNotFoundError:
|
|
return {}
|
|
|
|
def country_name(self):
|
|
return self._country().get("country_name", "??")
|
|
|
|
def country_code(self):
|
|
return self._country().get("country_code", "??")
|