allow to disable async/await hightlighting

andriykohut · andriykohut · commit 18fe49b4f782 · 2015-12-30T11:29:56.000+02:00
diff --git a/doc/pymode.txt b/doc/pymode.txt
@@ -628,6 +628,10 @@ Highlight "print" as a function            *'g:pymode_syntax_print_as_function'*
 >
     let g:pymode_syntax_print_as_function = 0
 
+Highlight "async/await" keywords       *'g:pymode_syntax_highlight_async_await'*
+>
+    let g:pymode_syntax_highlight_async_await = g:pymode_syntax_all
+
 Highlight '=' operator              *'g:pymode_syntax_highlight_equal_operator'*
 >
     let g:pymode_syntax_highlight_equal_operator = g:pymode_syntax_all
diff --git a/syntax/python.vim b/syntax/python.vim
@@ -16,6 +16,9 @@ call pymode#default('g:pymode_syntax_all', 1)
 
 " Highlight 'print' as function
 call pymode#default("g:pymode_syntax_print_as_function", 0)
+"
+" Highlight 'async/await' keywords
+call pymode#default("g:pymode_syntax_highlight_async_await", g:pymode_syntax_all)
 
 " Highlight '=' operator
 call pymode#default('g:pymode_syntax_highlight_equal_operator', g:pymode_syntax_all)
@@ -75,12 +78,8 @@ endif
     syn keyword pythonStatement yield
     syn keyword pythonLambdaExpr lambda
     syn keyword pythonStatement with as
-    syn keyword pythonStatement async await
 
     syn keyword pythonStatement def nextgroup=pythonFunction skipwhite
-    syn match pythonStatement "\<async\s\+def\>" nextgroup=pythonFunction skipwhite
-    syn match pythonStatement "\<async\s\+with\>" display
-    syn match pythonStatement "\<async\s\+for\>" nextgroup=pythonRepeat skipwhite
     syn match pythonFunction "\%(\%(def\s\|@\)\s*\)\@<=\h\%(\w\|\.\)*" contained nextgroup=pythonVars
     syn region pythonVars start="(" skip=+\(".*"\|'.*'\)+ end=")" contained contains=pythonParameters transparent keepend
     syn match pythonParameters "[^,]*" contained contains=pythonParam skipwhite
@@ -105,6 +104,13 @@ endif
         syn keyword pythonStatement print
     endif
 
+    if g:pymode_syntax_highlight_async_await
+        syn keyword pythonStatement async await
+        syn match pythonStatement "\<async\s\+def\>" nextgroup=pythonFunction skipwhite
+        syn match pythonStatement "\<async\s\+with\>" display
+        syn match pythonStatement "\<async\s\+for\>" nextgroup=pythonRepeat skipwhite
+    endif
+
     if g:pymode_syntax_highlight_equal_operator
         syn match pythonExtraOperator "\%(=\)"
     endif
diff --git a/t/syntax.vim b/t/syntax.vim
@@ -13,6 +13,7 @@ describe 'pymode-syntax'
         Expect g:pymode_syntax == 1
         Expect g:pymode_syntax_all == 1
         Expect g:pymode_syntax_print_as_function == 0
+	Expect g:pymode_syntax_highlight_async_await == 1
         Expect g:pymode_syntax_highlight_equal_operator == 1
         Expect g:pymode_syntax_highlight_stars_operator == 1
         Expect g:pymode_syntax_highlight_self == 1

Original file line number	Diff line number	Diff line change
`@@ -628,6 +628,10 @@ Highlight "print" as a function 'g:pymode_syntax_print_as_function'`
`628`	`628`	`>`
`629`	`629`	`let g:pymode_syntax_print_as_function = 0`
`630`	`630`
	`631`	`+Highlight "async/await" keywords 'g:pymode_syntax_highlight_async_await'`
	`632`	`+>`
	`633`	`+ let g:pymode_syntax_highlight_async_await = g:pymode_syntax_all`
	`634`	`+`
`631`	`635`	`Highlight '=' operator 'g:pymode_syntax_highlight_equal_operator'`
`632`	`636`	`>`
`633`	`637`	`let g:pymode_syntax_highlight_equal_operator = g:pymode_syntax_all`