def user_activation(request, code,
template_name="users/user_activation.html"):
try:
activation = UserActivationCode.objects.get(code=code)
active = activation.is_active()
except UserActivationCode.DoesNotExist:
activation = active = False
if active:
user = activation.user
user.is_active = True
user.save()
try:
consumer = Consumer.objects.get(user=user)
except Consumer.DoesNotExist:
consumer = Consumer(name=user.username, user=user,
status='accepted')
consumer.generate_random_codes()
consumer.save()
try:
token = Token.objects.get(user=user, consumer=consumer)
except Token.DoesNotExist:
token = Token(user=user, consumer=consumer, is_approved=True,
timestamp=time.time(),
token_type=Token.ACCESS)
token.generate_random_codes()
token.save()
extra_context = {
'active': active,
'activation': activation
}
return direct_to_template(request, template_name,
extra_context=extra_context)
def setUp(self):
self.user = User.objects.create_user('username', '*****@*****.**',
'userpassword')
self.user.save()
self.client = Client()
self.client.login(username='******', password='******')
self.consumer = Consumer(name=self.user.username,
status='accepted',
user=self.user)
self.consumer.generate_random_codes()
self.consumer.save()
self.token = Token(token_type=Token.ACCESS,
timestamp=time.time(),
is_approved=True,
user=self.user,
consumer=self.consumer)
self.token.generate_random_codes()
self.token.save()
