def get_domain_id()

in auto-installer/lambda/main.py [0:0]

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

def get_domain_id() -> str:
    if ENV_DOMAIN_ID:
        return ENV_DOMAIN_ID

    logger.debug("Auto-discovering SMStudio domain ID via ListDomains API")
    domains_resp = smclient.list_domains()
    domains = domains_resp["Domains"]
    if len(domains) < 0:
        raise ValueError(f"No SageMaker Studio domains in this region!")
    elif len(domains) > 1:
        raise ValueError(
            f"Cannot automatically select SageMaker Studio domain: multiple ({len(domains)}) were found"
        )

    return domains[0]["DomainId"]