Function _split_enum_values

lib/sqlalchemy/dialects/postgresql/array.py:489–519

(array_string: str)

def _split_enum_values(array_string: str) -> Sequence[Optional[str]]:
    if '"' not in array_string:
        # no escape char is present so it can just split on the comma
        return [
            r if r != "NULL" else None
            for r in (array_string.split(",") if array_string else [])
        ]

    # handles quoted strings from:
    # r'abc,"quoted","also\\\\quoted", "quoted, comma", "esc \" quot", qpr'
    # returns
    # ['abc', 'quoted', 'also\\quoted', 'quoted, comma', 'esc " quot', 'qpr']
    text = array_string.replace(r"\"", "_$ESC_QUOTE$_")
    text = text.replace(r"\\", "\\")
    result = []
    on_quotes = re.split(r'(")', text)
    in_quotes = False
    for tok in on_quotes:
        if tok == '"':
            in_quotes = not in_quotes
        elif in_quotes:
            result.append(tok.replace("_$ESC_QUOTE$_", '"'))
        else:
            # interpret NULL (without quotes!) as None
            result.extend(
                [
                    r if r != "NULL" else None
                    for r in re.findall(r"([^\s,]+),?", tok)
                ]
            )
    return result