code.grnet.gr Git - astakos/blob - snf-astakos-app/astakos/im/auth_backends.py

   1 from django.contrib.auth.backends import ModelBackend
   2 from django.core.validators import email_re
   3
   4 from astakos.im.models import AstakosUser
   5
   6 class TokenBackend(ModelBackend):
   7     """
   8     AuthenticationBackend used to authenticate using token instead
   9     """
  10     def authenticate(self, email=None, auth_token=None):
  11         try:
  12             user = AstakosUser.objects.get(email=email, is_active=True)
  13             if user.auth_token == auth_token:
  14                 return user
  15         except AstakosUser.DoesNotExist:
  16             return None
  17
  18     def get_user(self, user_id):
  19         try:
  20             return AstakosUser.objects.get(pk=user_id)
  21         except AstakosUser.DoesNotExist:
  22             return None
  23
  24 class EmailBackend(ModelBackend):
  25     """
  26     If the ``username`` parameter is actually an email uses email to authenticate
  27     the user else tries the username.
  28
  29     Used from ``astakos.im.forms.LoginForm`` to authenticate.
  30     """
  31     def authenticate(self, username=None, password=None):
  32         #If username is an email address, then try to pull it up
  33         if email_re.search(username):
  34             try:
  35                 user = AstakosUser.objects.get(email=username, is_active=True)
  36             except AstakosUser.DoesNotExist:
  37                 return None
  38         else:
  39             #We have a non-email address username we
  40             #should try username
  41             try:
  42                 user = AstakosUser.objects.get(username=username)
  43             except AstakosUser.DoesNotExist:
  44                 return None
  45         if user.check_password(password):
  46             return user
  47
  48     def get_user(self, user_id):
  49         try:
  50             return AstakosUser.objects.get(pk=user_id)
  51         except AstakosUser.DoesNotExist:
  52             return None