Możesz użyć pakietu sqlparse i sqlparse.format(sql, reindent=True, keyword_case='upper') powinien robić, co chcesz?