def _setup_boto_clients()

in src/vw-serving/src/vw_serving/model_manager.py [0:0]

• 18 lines of code
• 3 McCabe index (conditional complexity)

    def _setup_boto_clients(self):
        aws_region = os.getenv("AWS_REGION", "us-west-2")
        self.boto_session = boto3.Session(region_name=aws_region)
        
        # initialize resource clients 
        self.exp_ddb_table = os.getenv(environment.EXP_METADATA_DYNAMO_TABLE, "")
        if not self.exp_ddb_table:
            raise AlgorithmError(
                    f"Please specify a dynamo db table name as '{environment.EXP_METADATA_DYNAMO_TABLE}' environment variable.")
        self.model_ddb_table = os.getenv(environment.MODEL_METADATA_DYNAMO_TABLE, "")
        if not self.model_ddb_table:
            raise AlgorithmError(
                    f"Please specify a dynamo db table name as '{environment.MODEL_METADATA_DYNAMO_TABLE}' environment variable.")

        self.exp_ddb_table_resource = self.boto_session.resource('dynamodb').Table(self.exp_ddb_table)
        self._check_ddb_table_existence(self.exp_ddb_table_resource)

        self.model_ddb_table_resource = self.boto_session.resource('dynamodb').Table(self.model_ddb_table)
        self._check_ddb_table_existence(self.model_ddb_table_resource)

        self.exp_ddb_wrapper = ExperimentDBClient(self.exp_ddb_table_resource)
        self.model_ddb_wrapper = ModelDBClient(self.model_ddb_table_resource)
        self.s3_resource = self.boto_session.resource('s3')