def setUpClass(cls):
DBModel.metadata.drop_all(sqla_engine)
DBModel.metadata.create_all(sqla_engine)
r1, r2 = Role(name='admin'), Role(name='pikus')
DBSession.add(r1)
DBSession.add(r2)
a = Account(username='******',
fullName='Greg Burek',
email='*****@*****.**')
a.set_password('dupa')
a.roles.extend((r1, r2))
DBSession.add(a)
DBSession.commit()
cls.DBSession = DBSession
def save(self):
user = Account(email=self.validated_data['email'],
first_name=self.validated_data['first_name'],
last_name=self.validated_data['last_name'],
username=self.validated_data['username'],
address=self.validated_data['address'],
city=self.validated_data['city'],
zip_code=self.validated_data['zip_code'],
country=self.validated_data['country'])
password = self.validated_data['password']
password2 = self.validated_data['password2']
if password != password2:
raise serializers.ValidationError(
{'password': '******'})
user.set_password(password)
user.save()
Token.objects.get_or_create(user=user)
return user
