diff --git a/src/spaceone/identity/interface/grpc/workspace_group_user.py b/src/spaceone/identity/interface/grpc/workspace_group_user.py index 2830d9f..f5b24a9 100644 --- a/src/spaceone/identity/interface/grpc/workspace_group_user.py +++ b/src/spaceone/identity/interface/grpc/workspace_group_user.py @@ -1,4 +1,6 @@ +from google.protobuf.json_format import ParseDict from spaceone.api.identity.v2 import ( + workspace_group_pb2, workspace_group_user_pb2, workspace_group_user_pb2_grpc, ) @@ -19,19 +21,19 @@ def add(self, request, context): params, metadata = self.parse_request(request, context) workspace_group_user_svc = WorkspaceGroupUserService(metadata) response: dict = workspace_group_user_svc.add(params) - return self.dict_to_message(response) + return ParseDict(response, workspace_group_pb2.WorkspaceGroupInfo()) def remove(self, request, context): params, metadata = self.parse_request(request, context) workspace_group_user_svc = WorkspaceGroupUserService(metadata) response: dict = workspace_group_user_svc.remove(params) - return self.dict_to_message(response) + return ParseDict(response, workspace_group_pb2.WorkspaceGroupInfo()) def update_role(self, request, context): params, metadata = self.parse_request(request, context) workspace_group_user_svc = WorkspaceGroupUserService(metadata) response: dict = workspace_group_user_svc.update_role(params) - return self.dict_to_message(response) + return ParseDict(response, workspace_group_pb2.WorkspaceGroupInfo()) def find(self, request, context): params, metadata = self.parse_request(request, context) @@ -43,13 +45,13 @@ def get(self, request, context): params, metadata = self.parse_request(request, context) workspace_group_user_svc = WorkspaceGroupUserService(metadata) response: dict = workspace_group_user_svc.get(params) - return self.dict_to_message(response) + return ParseDict(response, workspace_group_pb2.WorkspaceGroupInfo()) def list(self, request, context): params, metadata = self.parse_request(request, context) workspace_group_user_svc = WorkspaceGroupUserService(metadata) response: dict = workspace_group_user_svc.list(params) - return self.dict_to_message(response) + return ParseDict(response, workspace_group_pb2.WorkspaceGroupsInfo()) def stat(self, request, context): params, metadata = self.parse_request(request, context)