Skip to content


Subversion checkout URL

You can clone with
Download ZIP
Tree: e5d1a21332
Fetching contributors…

Cannot retrieve contributors at this time

127 lines (104 sloc) 4.448 kB
from mongoengine import *
from django.utils.hashcompat import md5_constructor, sha_constructor
from django.utils.encoding import smart_str
from django.contrib.auth.models import AnonymousUser
import datetime
def get_hexdigest(algorithm, salt, raw_password):
raw_password, salt = smart_str(raw_password), smart_str(salt)
if algorithm == 'md5':
return md5_constructor(salt + raw_password).hexdigest()
elif algorithm == 'sha1':
return sha_constructor(salt + raw_password).hexdigest()
raise ValueError('Got unknown password algorithm type in password')
class User(Document):
"""A User document that aims to mirror most of the API specified by Django
username = StringField(max_length=30, required=True)
first_name = StringField(max_length=30)
last_name = StringField(max_length=30)
chapter = StringField(max_length=60, default='')
comment = StringField(max_length=100, default='')
activated_by = StringField(max_length=30, default='')
email = StringField()
password = StringField(max_length=128)
is_staff = BooleanField(default=False)
is_active = BooleanField(default=False)
is_superuser = BooleanField(default=False)
last_login = DateTimeField(
date_joined = DateTimeField(
tags = ListField(StringField(max_length=50))
subscribe = ListField(StringField())
publish = ListField(StringField())
def __unicode__(self):
return self.username
def get_full_name(self):
"""Returns the users first and last names, separated by a space.
full_name = u'%s %s' % (self.first_name or '', self.last_name or '')
return full_name.strip()
def is_anonymous(self):
return False
def is_authenticated(self):
return True
def set_password(self, raw_password):
"""Sets the user's password - always use this rather than directly
assigning to :attr:`~mongoengine.django.auth.User.password` as the
password is hashed before storage.
from random import random
algo = 'sha1'
salt = get_hexdigest(algo, str(random()), str(random()))[:5]
hash = get_hexdigest(algo, salt, raw_password)
self.password = '%s$%s$%s' % (algo, salt, hash)
return self
def check_password(self, raw_password):
"""Checks the user's password against a provided password - always use
this rather than directly comparing to
:attr:`~mongoengine.django.auth.User.password` as the password is
hashed before storage.
algo, salt, hash = self.password.split('$')
return hash == get_hexdigest(algo, salt, raw_password)
def create_user(cls, username, password, email=None):
"""Create (and save) a new user with the given username, password and
email address.
now =
# Normalize the address by lowercasing the domain part of the email
# address.
if email is not None:
email_name, domain_part = email.strip().split('@', 1)
except ValueError:
email = '@'.join([email_name, domain_part.lower()])
user = User(username=username, email=email, date_joined=now)
return user
def get_and_delete_messages(self):
return []
class MongoEngineBackend(object):
"""Authenticate using MongoEngine and mongoengine.django.auth.User.
def authenticate(self, username=None, password=None):
user = User.objects(username=username).first()
if user:
if password and user.check_password(password):
return user
return None
def get_user(self, user_id):
return User.objects.with_id(user_id)
def get_user(userid):
"""Returns a User object from an id ( Django's equivalent takes
request, but taking an id instead leaves it up to the developer to store
the id in any way they want (session, signed cookie, etc.)
if not userid:
return AnonymousUser()
return MongoEngineBackend().get_user(userid) or AnonymousUser()
Jump to Line
Something went wrong with that request. Please try again.