我想将别名拆分为diff数组,这是我正在使用的代码
def parse_sql_columns_alias(self, sql):
columns = []
parsed = sqlparse.parse(sql)
stmt = parsed[0]
for token in stmt.tokens:
if isinstance(token, IdentifierList):
for identifier in token.get_identifiers():
columns.append(str(identifier))
if isinstance(token, Identifier):
columns.append(str(token))
if token.ttype is Keyword: # from
break
return columns
得到下面的输出
['empid as eid', 'empname as ename']
想要输出
[eid, ename]