from django.db import models
from django.utils.encoding import python_2_unicode_compatible
from scrubpii import allow_sensitive_fields
with allow_sensitive_fields():
@python_2_unicode_compatible
class Person(models.Model):
first_name = models.CharField(max_length=30)
last_name = models.CharField(max_length=30)
biography = models.TextField()
date_of_birth = models.DateField()
email = models.EmailField()
def __str__(self):
return "{0} {1}".format(self.first_name, self.last_name)
class Meta:
sensitive_fields = {'biography', 'last_name', 'first_name', 'email', 'date_of_birth'}
@python_2_unicode_compatible
class Book(models.Model):
title = models.CharField(max_length=30)
author = models.ForeignKey(Person)
def __str__(self):
return "{0} by {1}".format(self.title, self.author)
with allow_sensitive_fields():
def test_futureproofing(self):
import django.db.models.options as options
options.DEFAULT_NAMES = options.DEFAULT_NAMES + ('sensitive_fields',)
with pytest.raises(Exception):
with allow_sensitive_fields():
pass
