Source code for xsdata.utils.text

from typing import List
from typing import Tuple


def prefix(string: str, sep: str = ":") -> str:
    return split(string, sep)[0]


def suffix(string: str, sep: str = ":") -> str:
    return split(string, sep)[1]


def split(string: str, sep: str = ":") -> Tuple:
    parts = string.split(sep, 1)
    if len(parts) == 1:
        return None, string

    return parts[0], parts[1]


def collapse_whitespace(string: str) -> str:
    return " ".join([part for part in string.split(" ") if part.strip()])


def capitalize(string: str) -> str:
    return string[0].upper() + string[1:]


def pascal_case(string: str) -> str:
    return "".join([capitalize(part) for part in snake_case(string).split("_") if part])


def snake_case(string: str) -> str:
    result: List[str] = []
    was_upper = False
    for char in string:

        if char.isalnum():
            if char.isupper():
                if not was_upper and result and result[-1] != "_":
                    result.append("_")
                was_upper = True
            else:
                was_upper = False
            result.append(char.lower())
        else:
            was_upper = False
            if result and result[-1] != "_":
                result.append("_")
    return "".join(result).strip("_")