Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- class UserFactory(factory.Factory):
- password = 'test'
- @classmethod
- def _prepare(cls, create, **kwargs):
- password = kwargs.pop('password', None)
- user = super(UserFactory, cls)._prepare(create, **kwargs)
- if password:
- user.raw_password = password
- user.set_password(password)
- if create:
- user.save()
- return user
Add Comment
Please, Sign In to add comment