72 lines
2.2 KiB
Python
72 lines
2.2 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/>.
|
|
|
|
from django.contrib.auth import get_user_model
|
|
from django.utils import six
|
|
|
|
from import_export.resources import ModelResource
|
|
from import_export.fields import Field
|
|
from import_export.widgets import Widget
|
|
|
|
from authentic2.a2_rbac.models import Role
|
|
|
|
User = get_user_model()
|
|
|
|
|
|
class ListWidget(Widget):
|
|
def clean(self, value):
|
|
raise NotImplementedError
|
|
|
|
def render(self, value, object):
|
|
return u', '.join(map(six.text_type, value.all()))
|
|
|
|
|
|
class UserResource(ModelResource):
|
|
roles = Field()
|
|
|
|
def dehydrate_roles(self, instance):
|
|
result = set()
|
|
for role in instance.roles.all():
|
|
result.add(role)
|
|
for pr in role.parent_relation.all():
|
|
result.add(pr.parent)
|
|
return ', '.join(map(six.text_type, result))
|
|
|
|
class Meta:
|
|
model = User
|
|
exclude = ('password', 'user_permissions', 'is_staff',
|
|
'is_superuser', 'groups')
|
|
export_order = ('ou', 'uuid', 'id', 'username', 'email',
|
|
'first_name', 'last_name', 'last_login',
|
|
'date_joined', 'roles')
|
|
widgets = {
|
|
'roles': {
|
|
'field': 'name',
|
|
},
|
|
'ou': {
|
|
'field': 'name',
|
|
}
|
|
}
|
|
|
|
|
|
class RoleResource(ModelResource):
|
|
members = Field(attribute='members', widget=ListWidget())
|
|
|
|
class Meta:
|
|
model = Role
|
|
fields = ('name', 'slug', 'members')
|
|
export_order = fields
|