Você pode usar o pacote sqlparse e sqlparse.format(sql, reindent=True, keyword_case='upper') deve fazer o que você quer?