def format_database_user()

in google/cloud/sql/connector/utils.py [0:0]

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

def format_database_user(database_version: str, user: str) -> str:
    """Format database `user` param for Cloud SQL automatic IAM authentication.

    Args:
        database_version (str): Cloud SQL database version.
        user (str): Database username to connect to Cloud SQL database with.

    Returns:
        str: Formatted database username.
    """
    # remove suffix for Postgres service accounts
    if database_version.startswith("POSTGRES"):
        suffix = ".gserviceaccount.com"
        user = user[: -len(suffix)] if user.endswith(suffix) else user
        return user

    # remove everything after and including the @ for MySQL
    if database_version.startswith("MYSQL") and "@" in user:
        return user.split("@")[0]

    return user