Use sql_metadata, which leverages sqlparse:

from sql_metadata import Parser


def get_query_columns(query) -> list[str]:
    return Parser(query).columns