def __init__()

in src/s3_helper.py [0:0]

• 10 lines of code
• 1 McCabe index (conditional complexity)

    def __init__(self, role_arn=None):
        """
        Initialize the S3 client.
        :param role_arn: Role which have access to consolidation request S3 payload file.
        :return:

        """
        default_session = "SSIE_Annotation_Consolidation_Lambda_Session"
        sts_connection = boto3.client('sts')
        assume_role_object = sts_connection.assume_role(RoleArn=role_arn, RoleSessionName=default_session)
        session = boto3.Session(
            aws_access_key_id=assume_role_object['Credentials']['AccessKeyId'],
            aws_secret_access_key=assume_role_object['Credentials']['SecretAccessKey'],
            aws_session_token=assume_role_object['Credentials']['SessionToken'])
        self.s3 = session.resource('s3')
        self.s3_client = session.client('s3')