alejandroandreu / flask-minio Goto Github PK

def process_minio_string(minio_uri: str) -> dict[str, str]:
    """Turn a minio URI string into split usable variables.

    URI format is: minio+secure://username:password@hostname:port/bucket

    This works from the right and left separately to get to password. This
    ensures that there are no issues splitting strings when the password contains
    special characters."""
    # Work from left to password
    prefix, _, uri = minio_uri.partition("+")
    secure, _, uri = uri.partition("://")
    username, _, uri = uri.partition(":")

    # Work from right to password
    uri, _, bucket = uri.rpartition("/")
    uri, _, port = uri.rpartition(":")
    password, _, hostname = uri.rpartition("@")

    if prefix != "minio":
        raise TypeError("Minio URI appears to be of wrong type")

    if secure == "secure":
        sec = 1
    elif secure == "insecure":
        sec = 0
    else:
        raise TypeError("Minio string security must be 'secure' or 'insecure'.")

    return {
        "secure": sec,
        "username": username,
        "password": password,
        "hostname": hostname,
        "bucket": bucket,
        "port": port,
        "endpoint": f"{hostname}:{port}",
    }