132 lines
3.9 KiB
Python
132 lines
3.9 KiB
Python
# authentic2 - versatile identity manager
|
|
# Copyright (C) 2010-2019 Entr'ouvert
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify it
|
|
# under the terms of the GNU Affero General Public License as published
|
|
# by the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import base64
|
|
import operator
|
|
import pickle
|
|
|
|
from django.contrib.auth import get_user_model
|
|
from django.core import signing
|
|
from django.utils import six
|
|
from django.utils.encoding import force_text
|
|
from django_select2.forms import ModelSelect2MultipleWidget, ModelSelect2Widget
|
|
|
|
from authentic2_idp_oidc.models import OIDCAuthorization
|
|
from django_rbac.utils import get_role_model
|
|
|
|
from . import utils
|
|
|
|
|
|
class SplitTermMixin(object):
|
|
split_term_operator = operator.__or__
|
|
|
|
def filter_queryset(self, term, queryset=None):
|
|
if queryset is not None:
|
|
qs = queryset
|
|
else:
|
|
qs = self.get_queryset()
|
|
if not term.strip():
|
|
return qs.all()
|
|
queries = []
|
|
for term in term.split():
|
|
queries.append(super(SplitTermMixin, self).filter_queryset(term, queryset=qs))
|
|
qs = six.moves.reduce(self.split_term_operator, queries)
|
|
return qs
|
|
|
|
|
|
class Select2Mixin(object):
|
|
def build_attrs(self, *args, **kwargs):
|
|
attrs = super(Select2Mixin, self).build_attrs(*args, **kwargs)
|
|
field_data = {
|
|
'class': self.__class__.__name__,
|
|
'where_clause': force_text(base64.b64encode(pickle.dumps(self.queryset.query.where))),
|
|
}
|
|
attrs['data-field_id'] = signing.dumps(field_data)
|
|
return attrs
|
|
|
|
@classmethod
|
|
def get_initial_queryset(cls):
|
|
return cls.model.objects.all()
|
|
|
|
|
|
class SimpleModelSelect2Widget(Select2Mixin, ModelSelect2Widget):
|
|
pass
|
|
|
|
|
|
class SimpleModelSelect2MultipleWidget(Select2Mixin, ModelSelect2MultipleWidget):
|
|
pass
|
|
|
|
|
|
class SearchUserWidgetMixin(SplitTermMixin):
|
|
model = get_user_model()
|
|
search_fields = [
|
|
'username__icontains',
|
|
'first_name__icontains',
|
|
'last_name__icontains',
|
|
'email__icontains',
|
|
]
|
|
|
|
def label_from_instance(self, user):
|
|
return utils.label_from_user(user)
|
|
|
|
|
|
class ChooseUserWidget(SearchUserWidgetMixin, SimpleModelSelect2Widget):
|
|
pass
|
|
|
|
|
|
class ChooseUsersWidget(SearchUserWidgetMixin, SimpleModelSelect2MultipleWidget):
|
|
pass
|
|
|
|
|
|
class SearchRoleWidgetMixin(SplitTermMixin):
|
|
model = get_role_model()
|
|
split_term_operator = operator.__and__
|
|
search_fields = [
|
|
'name__icontains',
|
|
'service__name__icontains',
|
|
'ou__name__icontains',
|
|
]
|
|
|
|
def label_from_instance(self, obj):
|
|
label = six.text_type(obj)
|
|
if obj.ou and utils.get_ou_count() > 1:
|
|
label = u'{ou} - {obj}'.format(ou=obj.ou, obj=obj)
|
|
return label
|
|
|
|
|
|
class ChooseRoleWidget(SearchRoleWidgetMixin, SimpleModelSelect2Widget):
|
|
@classmethod
|
|
def get_initial_queryset(cls):
|
|
return cls.model.objects.exclude(slug__startswith='_')
|
|
|
|
|
|
class ChooseRolesWidget(SearchRoleWidgetMixin, SimpleModelSelect2MultipleWidget):
|
|
@classmethod
|
|
def get_initial_queryset(cls):
|
|
return cls.model.objects.exclude(slug__startswith='_')
|
|
|
|
|
|
class ChooseManageableMemberRolesWidget(SearchRoleWidgetMixin, SimpleModelSelect2MultipleWidget):
|
|
perm = 'manage_members'
|
|
|
|
|
|
class ChooseManageableMemberRoleWidget(SearchRoleWidgetMixin, SimpleModelSelect2Widget):
|
|
perm = 'manage_members'
|
|
|
|
|
|
class ChooseUserAuthorizationsWidget(SimpleModelSelect2Widget):
|
|
model = OIDCAuthorization
|