Support more Python keywords

2025-05-08 12:34:06 +02:00 · 2021-08-22 18:01:55 +08:00 · 2021-08-22 18:01:55 +08:00 · ed2aead6f6
commit ed2aead6f6
parent f7cce387c1
1 changed files with 13 additions and 0 deletions
--- a/src/main/kotlin/com/github/izhangzhihao/rainbow/brackets/visitor/PythonRainbowVisitor.kt
+++ b/src/main/kotlin/com/github/izhangzhihao/rainbow/brackets/visitor/PythonRainbowVisitor.kt
@ -46,6 +46,19 @@ class PythonRainbowVisitor : RainbowHighlightVisitor() {
            IF_KEYWORD,
            ELSE_KEYWORD,
            ELIF_KEYWORD,
+            BREAK_KEYWORD,
+            CONTINUE_KEYWORD,
+            CLASS_KEYWORD,
+            DEF_KEYWORD,
+            EXCEPT_KEYWORD,
+            FINALLY_KEYWORD,
+            FOR_KEYWORD,
+            RAISE_KEYWORD,
+            RETURN_KEYWORD,
+            TRY_KEYWORD,
+            WITH_KEYWORD,
+            WHILE_KEYWORD,
+            YIELD_KEYWORD,
        )

        private fun LeafPsiElement.getBracketLevel(): Int =