mirror of https://github.com/sgoudham/Enso-Bot.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
44 lines
1.2 KiB
Python
44 lines
1.2 KiB
Python
5 years ago
|
from django import db
|
||
|
|
from django.contrib import auth
|
||
|
|
|
||
|
|
UserModel = auth.get_user_model()
|
||
|
|
|
||
|
|
|
||
|
|
def check_password(environ, username, password):
|
||
|
|
"""
|
||
|
|
Authenticate against Django's auth database.
|
||
|
|
|
||
|
|
mod_wsgi docs specify None, True, False as return value depending
|
||
|
|
on whether the user exists and authenticates.
|
||
|
|
"""
|
||
|
|
# 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:
|
||
|
|
user = UserModel._default_manager.get_by_natural_key(username)
|
||
|
|
except UserModel.DoesNotExist:
|
||
|
|
return None
|
||
|
|
if not user.is_active:
|
||
|
|
return None
|
||
|
|
return user.check_password(password)
|
||
|
|
finally:
|
||
|
|
db.close_old_connections()
|
||
|
|
|
||
|
|
|
||
|
|
def groups_for_user(environ, username):
|
||
|
|
"""
|
||
|
|
Authorize a user based on groups
|
||
|
|
"""
|
||
|
|
db.reset_queries()
|
||
|
|
try:
|
||
|
|
try:
|
||
|
|
user = UserModel._default_manager.get_by_natural_key(username)
|
||
|
|
except UserModel.DoesNotExist:
|
||
|
|
return []
|
||
|
|
if not user.is_active:
|
||
|
|
return []
|
||
|
|
return [group.name.encode() for group in user.groups.all()]
|
||
|
|
finally:
|
||
|
|
db.close_old_connections()
|