2012-10-02 19:19:44 +08:00
|
|
|
from django.contrib import auth
|
2012-09-24 13:48:13 +08:00
|
|
|
from django import db
|
|
|
|
from django.utils.encoding import force_bytes
|
|
|
|
|
|
|
|
|
|
|
|
def check_password(environ, username, password):
|
|
|
|
"""
|
|
|
|
Authenticates against Django's auth database
|
|
|
|
|
|
|
|
mod_wsgi docs specify None, True, False as return value depending
|
|
|
|
on whether the user exists and authenticates.
|
|
|
|
"""
|
|
|
|
|
2012-10-02 19:19:44 +08:00
|
|
|
UserModel = auth.get_user_model()
|
2012-09-24 13:48:13 +08:00
|
|
|
# db connection state is managed similarly to the wsgi handler
|
|
|
|
# as mod_wsgi may call these functions outside of a request/response cycle
|
|
|
|
db.reset_queries()
|
|
|
|
|
|
|
|
try:
|
|
|
|
try:
|
2013-01-22 19:47:34 +08:00
|
|
|
user = UserModel._default_manager.get_by_natural_key(username)
|
2012-10-02 19:19:44 +08:00
|
|
|
except UserModel.DoesNotExist:
|
|
|
|
return None
|
2012-10-03 00:16:37 +08:00
|
|
|
if not user.is_active:
|
2012-09-24 13:48:13 +08:00
|
|
|
return None
|
|
|
|
return user.check_password(password)
|
|
|
|
finally:
|
|
|
|
db.close_connection()
|
|
|
|
|
|
|
|
def groups_for_user(environ, username):
|
|
|
|
"""
|
|
|
|
Authorizes a user based on groups
|
|
|
|
"""
|
|
|
|
|
2012-10-02 19:19:44 +08:00
|
|
|
UserModel = auth.get_user_model()
|
2012-09-24 13:48:13 +08:00
|
|
|
db.reset_queries()
|
|
|
|
|
|
|
|
try:
|
|
|
|
try:
|
2013-01-22 19:47:34 +08:00
|
|
|
user = UserModel._default_manager.get_by_natural_key(username)
|
2012-10-02 19:19:44 +08:00
|
|
|
except UserModel.DoesNotExist:
|
|
|
|
return []
|
|
|
|
try:
|
|
|
|
if not user.is_active:
|
|
|
|
return []
|
|
|
|
except AttributeError as e:
|
|
|
|
# a custom user may not support is_active
|
2012-09-24 13:48:13 +08:00
|
|
|
return []
|
|
|
|
return [force_bytes(group.name) for group in user.groups.all()]
|
|
|
|
finally:
|
|
|
|
db.close_connection()
|