Highlight self and cls attributes

hdima · Jan 1, 2016 · aa7aba6 · aa7aba6
1 parent 49fd76c
commit aa7aba6
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 0 deletions.
diff --git a/README.rst b/README.rst
@@ -103,6 +103,8 @@ Options used by the script
   Highlight builtin objects only
 ``python_highlight_builtin_funcs``
   Highlight builtin functions only
+``python_highlight_self``
+  Highlight self and cls attributes
 ``python_highlight_exceptions``
   Highlight standard exceptions
 ``python_highlight_string_formatting``

diff --git a/syntax.txt b/syntax.txt
@@ -40,6 +40,8 @@ Options used by the script
     :let python_highlight_builtin_objs = 1
 <  Highlight builtin functions only >
     :let python_highlight_builtin_funcs = 1
+<  Highlight self and cls attributes >
+    :let python_highlight_self = 1
 <  Highlight standard exceptions >
     :let python_highlight_exceptions = 1
 <  Highlight `%` string formatting >

diff --git a/syntax/python.vim b/syntax/python.vim
@@ -72,6 +72,7 @@
 "                                           objects
 "      python_highlight_builtin_objs        Highlight builtin objects only
 "      python_highlight_builtin_funcs       Highlight builtin functions only
+"    python_highlight_self                  Highlight self and cls attributes
 "    python_highlight_exceptions            Highlight standard exceptions
 "    python_highlight_string_formatting     Highlight % string formatting
 "    python_highlight_string_format         Highlight str.format syntax
@@ -139,6 +140,7 @@ if s:Enabled("g:python_highlight_all")
     call s:EnableByDefault("g:python_highlight_builtin_objs")
     call s:EnableByDefault("g:python_highlight_builtin_funcs")
   endif
+  call s:EnableByDefault("g:python_highlight_self")
   call s:EnableByDefault("g:python_highlight_exceptions")
   call s:EnableByDefault("g:python_highlight_string_formatting")
   call s:EnableByDefault("g:python_highlight_string_format")
@@ -417,6 +419,14 @@ if s:Enabled("g:python_highlight_builtin_objs")
   syn keyword pythonBuiltinObj	__debug__ __doc__ __file__ __name__ __package__
 endif
 
+"
+" Self and cls
+"
+
+if s:Enabled("g:python_highlight_self")
+  syn keyword pythonSelf self cls
+endif
+
 "
 " Builtin functions
 "
@@ -573,6 +583,8 @@ if version >= 508 || !exists("did_python_syn_inits")
   HiLink pythonBuiltinObj       Structure
   HiLink pythonBuiltinFunc      Function
 
+  HiLink pythonSelf             Identifier
+
   HiLink pythonExClass          Structure
 
   delcommand HiLink