Skip to content

Commit

Permalink
feat: list data_source users #1159
Browse files Browse the repository at this point in the history
  • Loading branch information
Canway-shiisa committed Aug 17, 2023
1 parent 68f6f92 commit 594f5af
Show file tree
Hide file tree
Showing 6 changed files with 108 additions and 0 deletions.
10 changes: 10 additions & 0 deletions src/bk-user/bkuser/apis/web/data_source/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
# -*- coding: utf-8 -*-
"""
TencentBlueKing is pleased to support the open source community by making 蓝鲸智云-用户管理(Bk-User) available.
Copyright (C) 2017-2021 THL A29 Limited, a Tencent company. All rights reserved.
Licensed under the MIT License (the "License"); you may not use this file except in compliance with the License.
You may obtain a copy of the License at http://opensource.org/licenses/MIT
Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
specific language governing permissions and limitations under the License.
"""
34 changes: 34 additions & 0 deletions src/bk-user/bkuser/apis/web/data_source/serializers.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,34 @@
# -*- coding: utf-8 -*-
"""
TencentBlueKing is pleased to support the open source community by making 蓝鲸智云-用户管理(Bk-User) available.
Copyright (C) 2017-2021 THL A29 Limited, a Tencent company. All rights reserved.
Licensed under the MIT License (the "License"); you may not use this file except in compliance with the License.
You may obtain a copy of the License at http://opensource.org/licenses/MIT
Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
specific language governing permissions and limitations under the License.
"""
from rest_framework import serializers

from bkuser.apps.data_source.models import DataSourceDepartmentUserRelation


class UserSearchInputSLZ(serializers.Serializer):
username = serializers.CharField(required=False, help_text="用户名", allow_blank=True)


class UserSearchOutput(serializers.Serializer):
username = serializers.CharField()
full_name = serializers.CharField()
phone = serializers.CharField()
email = serializers.CharField()
departments = serializers.SerializerMethodField()

def get_departments(self, obj):
department_user_relation_objs = DataSourceDepartmentUserRelation.objects.filter(user=obj)
if department_user_relation_objs:
return [
{"id": relation_obj.department.id, "name": relation_obj.department.name}
for relation_obj in department_user_relation_objs
]
return []
17 changes: 17 additions & 0 deletions src/bk-user/bkuser/apis/web/data_source/urls.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,17 @@
# -*- coding: utf-8 -*-
"""
TencentBlueKing is pleased to support the open source community by making 蓝鲸智云-用户管理(Bk-User) available.
Copyright (C) 2017-2021 THL A29 Limited, a Tencent company. All rights reserved.
Licensed under the MIT License (the "License"); you may not use this file except in compliance with the License.
You may obtain a copy of the License at http://opensource.org/licenses/MIT
Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
specific language governing permissions and limitations under the License.
"""
from django.urls import path

from bkuser.apis.web.data_source import views

urlpatterns = [
path("<int:id>/users/", views.DataSourceUserListCreateApi.as_view(), name="data_source.list_create"),
]
45 changes: 45 additions & 0 deletions src/bk-user/bkuser/apis/web/data_source/views.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,45 @@
# -*- coding: utf-8 -*-
"""
TencentBlueKing is pleased to support the open source community by making 蓝鲸智云-用户管理(Bk-User) available.
Copyright (C) 2017-2021 THL A29 Limited, a Tencent company. All rights reserved.
Licensed under the MIT License (the "License"); you may not use this file except in compliance with the License.
You may obtain a copy of the License at http://opensource.org/licenses/MIT
Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
specific language governing permissions and limitations under the License.
"""
from drf_yasg.utils import swagger_auto_schema
from rest_framework import generics, status

from bkuser.apis.web.data_source.serializers import UserSearchInputSLZ, UserSearchOutput
from bkuser.apps.data_source.models import DataSource, DataSourceUser
from bkuser.common.error_codes import error_codes


class DataSourceUserListCreateApi(generics.ListCreateAPIView):
pagination_class = None
serializer_class = UserSearchOutput

@swagger_auto_schema(
operation_description="数据源用户列表",
query_serializer=UserSearchInputSLZ(),
responses={status.HTTP_200_OK: UserSearchOutput(many=True)},
)
def get(self, request, *args, **kwargs):
slz = UserSearchInputSLZ(data=self.request.query_params)
slz.is_valid(raise_exception=True)
data = slz.validated_data
data_source_id = kwargs["id"]

# 校验数据源是否存在
try:
data_source = DataSource.objects.get(id=data_source_id)
except Exception:
raise error_codes.DATA_SOURCE_NOT_EXIST

self.queryset = DataSourceUser.objects.filter(data_source=data_source)

if data.get("username"):
self.queryset = DataSourceUser.objects.filter(username__icontains=data["username"])

return self.list(request, *args, **kwargs)
1 change: 1 addition & 0 deletions src/bk-user/bkuser/apis/web/urls.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,4 +15,5 @@
path("basic/", include("bkuser.apis.web.basic.urls")),
# 租户
path("tenants/", include("bkuser.apis.web.tenant.urls")),
path("data_source/", include("bkuser.apis.web.data_source.urls")),
]
1 change: 1 addition & 0 deletions src/bk-user/bkuser/common/error_codes.py
Original file line number Diff line number Diff line change
Expand Up @@ -70,6 +70,7 @@ class ErrorCodes:
REMOTE_REQUEST_ERROR = ErrorCode(_("调用外部系统API异常"))
# 数据源
DATA_SOURCE_TYPE_NOT_SUPPORTED = ErrorCode(_("数据源类型不支持"))
DATA_SOURCE_NOT_EXIST = ErrorCode(_("数据源不存在"))
# 租户
CREATE_TENANT_FAILED = ErrorCode(_("租户创建失败"))
UPDATE_TENANT_FAILED = ErrorCode(_("租户更新失败"))
Expand Down

0 comments on commit 594f5af

Please sign in to comment.