Add syntax highlight for walrus operator

fpob · fpob · commit 0beb5f853890 · 2021-11-14T15:15:52.000+01:00
`:=` operator (PEP 572)
diff --git a/doc/pymode.txt b/doc/pymode.txt
@@ -674,6 +674,10 @@ Highlight '=' operator               *'g:pymode_syntax_highlight_equal_operator'
 >
     let g:pymode_syntax_highlight_equal_operator = g:pymode_syntax_all
 
+Highlight ':=' operator             *'g:pymode_syntax_highlight_walrus_operator'*
+>
+    let g:pymode_syntax_highlight_walrus_operator = g:pymode_syntax_all
+
 Highlight '*' operator               *'g:pymode_syntax_highlight_stars_operator'*
 >
     let g:pymode_syntax_highlight_stars_operator = g:pymode_syntax_all
diff --git a/syntax/python.vim b/syntax/python.vim
@@ -23,6 +23,9 @@ 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)
 
+" Highlight ':=' operator
+call pymode#default('g:pymode_syntax_highlight_walrus_operator', g:pymode_syntax_all)
+
 " Highlight '*' operator
 call pymode#default('g:pymode_syntax_highlight_stars_operator', g:pymode_syntax_all)
 
@@ -114,6 +117,10 @@ endif
         syn match pythonExtraOperator "\%(=\)"
     endif
 
+    if g:pymode_syntax_highlight_walrus_operator
+        syn match pythonExtraOperator "\%(:=\)"
+    endif
+
     if g:pymode_syntax_highlight_stars_operator
         syn match pythonExtraOperator "\%(\*\|\*\*\)"
     endif

Original file line number	Diff line number	Diff line change
`@@ -674,6 +674,10 @@ Highlight '=' operator *'g:pymode_syntax_highlight_equal_operator'`
`674`	`674`	`>`
`675`	`675`	`let g:pymode_syntax_highlight_equal_operator = g:pymode_syntax_all`
`676`	`676`
	`677`	`+Highlight ':=' operator 'g:pymode_syntax_highlight_walrus_operator'`
	`678`	`+>`
	`679`	`+ let g:pymode_syntax_highlight_walrus_operator = g:pymode_syntax_all`
	`680`	`+`
`677`	`681`	`Highlight '' operator 'g:pymode_syntax_highlight_stars_operator'*`
`678`	`682`	`>`
`679`	`683`	`let g:pymode_syntax_highlight_stars_operator = g:pymode_syntax_all`