Advertisement
Guest User

Untitled

a guest
Mar 24th, 2017
63
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.76 KB | None | 0 0
  1. class LoginView(FormView):
  2. template_name = 'login.html'
  3. form_class = LoginForm
  4. success_url = '/dashboard'
  5.  
  6. def dispatch(self, request, *args, **kwargs):
  7. if self.request.user.is_authenticated:
  8. messages.add_message(self.request, messages.INFO,
  9. "User already logged in")
  10. return redirect('/dashboard')
  11. else:
  12. return super().dispatch(request, *args, **kwargs)
  13.  
  14. def form_valid(self, form):
  15. user = form.login(self.request)
  16. print(user.two_factor_auth)
  17. if user.two_factor_auth is False:
  18. login(self.request, user)
  19. return redirect('/dashboard')
  20. else:
  21. try:
  22. response = send_verfication_code(user)
  23. pass
  24. except Exception as e:
  25. messages.add_message(self.request, messages.ERROR,
  26. 'verification code not sent. \n'
  27. 'Please retry logging in.')
  28. return redirect('/login')
  29. data = json.loads(response.text)
  30.  
  31. if data['success'] == False:
  32. messages.add_message(self.request, messages.ERROR,
  33. data['message'])
  34. return redirect('/login')
  35.  
  36. print(response.status_code, response.reason)
  37. print(response.text)
  38. if data['success'] == True:
  39. self.request.method = "GET"
  40. print(self.request.method)
  41. kwargs = {'user':user}
  42. return PhoneVerificationView(self.request, **kwargs)
  43. else:
  44. messages.add_message(self.request, messages.ERROR,
  45. data['message'])
  46. return redirect('/login')
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement