def _get_connection()

in modular_sdk/models/pynamodb_extension/base_role_access_model.py [0:0]

• 24 lines of code
• 4 McCabe index (conditional complexity)

    def _get_connection(cls) -> TableConnection:
        _modular = Modular()
        sts = _modular.sts_service()
        if sts.assure_modular_credentials_valid():
            env = _modular.environment_service()
            for model in BaseRoleAccessModel.__subclasses__():
                if model._connection:
                    # works as well but seems too tough
                    # model._connection = None
                    _LOG.warning(
                        f'Existing connection found in {model.__name__}. '
                        f'Updating credentials in botocore session and '
                        f'dropping the existing botocore client...')
                    model._connection.connection.session.set_credentials(
                        env.modular_aws_access_key_id(),
                        env.modular_aws_secret_access_key(),
                        env.modular_aws_session_token()
                    )
                    model._connection.connection._client = None
                else:
                    _LOG.info(
                        f'Existing connection not found in {model.__name__}'
                        f'. Probably the first request. Connection will be '
                        f'created using creds from envs which '
                        f'already have been updated')
        return super()._get_connection()