Jul-12-2019, 12:01 AM
I've never used
sqlparse
before. Nevertheless, exploring its source code might lead to the following solution, e.g. import sqlparse from sqlparse.tokens import Token # look at source of tokens.py raw = 'select * from foo; select * from bar;' parsed = sqlparse.parse(raw) list(filter(lambda x: x.ttype in [Token.Keyword, Token.Keyword.DML], parsed[0].tokens))
Output:[<DML 'select' at 0x7F732037EEE8>, <Keyword 'from' at 0x7F732039F0A8>]