57 lines
1.8 KiB
Python
57 lines
1.8 KiB
Python
import factory
|
|
import faker
|
|
|
|
from unidecode import unidecode
|
|
|
|
from django.contrib.auth.models import User
|
|
from ietf.person.models import Person, Alias, Email
|
|
|
|
fake = faker.Factory.create()
|
|
|
|
class UserFactory(factory.DjangoModelFactory):
|
|
class Meta:
|
|
model = User
|
|
django_get_or_create = ('username',)
|
|
|
|
first_name = factory.Faker('first_name')
|
|
last_name = factory.Faker('last_name')
|
|
email = factory.LazyAttribute(lambda u: '%s.%s@%s'%(u.first_name,u.last_name,fake.domain_name()))
|
|
username = factory.LazyAttribute(lambda u: u.email)
|
|
|
|
@factory.post_generation
|
|
def set_password(self, create, extracted, **kwargs):
|
|
self.set_password( '%s+password' % self.username )
|
|
|
|
class PersonFactory(factory.DjangoModelFactory):
|
|
class Meta:
|
|
model = Person
|
|
|
|
user = factory.SubFactory(UserFactory)
|
|
name = factory.LazyAttribute(lambda p: '%s %s'%(p.user.first_name,p.user.last_name))
|
|
ascii = factory.LazyAttribute(lambda p: unidecode(p.name))
|
|
|
|
@factory.post_generation
|
|
def default_aliases(self, create, extracted, **kwargs):
|
|
make_alias = getattr(AliasFactory, 'create' if create else 'build')
|
|
make_alias(person=self,name=self.name)
|
|
make_alias(person=self,name=self.ascii)
|
|
|
|
@factory.post_generation
|
|
def default_emails(self, create, extracted, **kwargs):
|
|
make_email = getattr(EmailFactory, 'create' if create else 'build')
|
|
make_email(person=self,address=self.user.email)
|
|
|
|
class AliasFactory(factory.DjangoModelFactory):
|
|
class Meta:
|
|
model = Alias
|
|
django_get_or_create = ('name',)
|
|
|
|
name = factory.Faker('name')
|
|
|
|
class EmailFactory(factory.DjangoModelFactory):
|
|
class Meta:
|
|
model = Email
|
|
django_get_or_create = ('address',)
|
|
|
|
address = '%s.%s@%s' % (factory.Faker('first_name'),factory.Faker('last_name'),factory.Faker('domain_name'))
|