| # Licensed to the Apache Software Foundation (ASF) under one or more |
| # contributor license agreements. See the NOTICE file distributed with |
| # this work for additional information regarding copyright ownership. |
| # The ASF licenses this file to You under the Apache License, Version 2.0 |
| # (the "License"); you may not use this file except in compliance with |
| # the License. You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # 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. |
| # |
| |
| import configparser |
| import logging |
| from typing import Optional |
| |
| from airavata_sdk.transport import utils |
| from airavata_sdk import Settings |
| |
| logger = logging.getLogger(__name__) |
| logger.setLevel(logging.DEBUG) |
| # create console handler with a higher log level |
| handler = logging.StreamHandler() |
| handler.setLevel(logging.DEBUG) |
| # create formatter and add it to the handler |
| formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s') |
| handler.setFormatter(formatter) |
| # add the handler to the logger |
| logger.addHandler(handler) |
| |
| |
| class IAMAdminClient(object): |
| |
| def __init__(self): |
| self.settings = Settings() |
| self.client = utils.initialize_iam_admin_client( |
| self.settings.PROFILE_SERVICE_HOST, |
| self.settings.PROFILE_SERVICE_PORT, |
| self.settings.PROFILE_SERVICE_SECURE, |
| ) |
| # expose the needed functions |
| self.set_up_gateway = self.client.setUpGateway |
| self.is_username_available = self.client.isUsernameAvailable |
| self.register_user = self.client.registerUser |
| self.enable_user = self.client.enableUser |
| self.is_user_enabled = self.client.isUserEnabled |
| self.is_user_exist = self.client.isUserExist |
| self.get_user = self.client.getUser |
| self.get_users = self.client.getUsers |
| self.reset_user_password = self.client.resetUserPassword |
| self.find_users = self.client.findUsers |
| self.update_user_profile = self.client.updateUserProfile |
| self.delete_user = self.client.deleteUser |
| self.add_role_to_user = self.client.addRoleToUser |
| self.remove_role_from_user = self.client.removeRoleFromUser |
| self.get_users_with_role = self.client.getUsersWithRole |
