Můžete použít balíček sqlparse a sqlparse.format(sql, reindent=True, keyword_case='upper') měli byste dělat, co chcete?