def __init__()

in sagemaker_studio_image_build/codebuild.py [0:0]

• 32 lines of code
• 8 McCabe index (conditional complexity)

    def __init__(self, s3_location, role, repository=None, compute_type=None, vpc_config=None):
        self.s3_location = s3_location
        self.role = role

        self.session = boto3.session.Session()
        self.domain_id, self.user_profile_name = self._get_studio_metadata()
        self.repo_name = None
        self.compute_type = compute_type or "BUILD_GENERAL1_SMALL"
        self.vpc_config = vpc_config

        if repository:
            self.repo_name, self.tag = repository.split(":", maxsplit=1)

        if self.domain_id and self.user_profile_name:
            project_name_prefix = (
                f"sagemaker-studio-{self.domain_id}-{self.user_profile_name}-"
            )
            project_name_prefix = (
                project_name_prefix[:239]
                if len(project_name_prefix) > 239
                else project_name_prefix
            )
            self.project_name = project_name_prefix + "".join(
                random.choices(string.ascii_letters, k=16)
            )
            if not self.repo_name:
                self.repo_name = f"sagemaker-studio-{self.domain_id}"
                self.tag = self.user_profile_name

        else:
            self.project_name = "sagemaker-studio-image-build-" + "".join(
                random.choices(string.ascii_letters, k=16)
            )
            if not self.repo_name:
                self.repo_name = "sagemaker-studio"
                self.tag = "latest"