File tree Expand file tree Collapse file tree 1 file changed +5
-4
lines changed Expand file tree Collapse file tree 1 file changed +5
-4
lines changed Original file line number Diff line number Diff line change 20
20
ZERO_WIDTH_BRACKET = re .compile (r"\x01.*?\x02" )
21
21
ZERO_WIDTH_TRANS = str .maketrans ({"\x01 " : "" , "\x02 " : "" })
22
22
IDENTIFIERS_AFTER = {"def" , "class" }
23
+ KEYWORD_CONSTANTS = {"True" , "False" , "None" }
23
24
BUILTINS = {str (name ) for name in dir (builtins ) if not name .startswith ('_' )}
24
25
25
26
@@ -196,12 +197,12 @@ def gen_colors_from_token_stream(
196
197
is_def_name = False
197
198
span = Span .from_token (token , line_lengths )
198
199
yield ColorSpan (span , "definition" )
199
- elif token .string in ("True" , "False" , "None" ):
200
- span = Span .from_token (token , line_lengths )
201
- yield ColorSpan (span , "keyword_constant" )
202
200
elif keyword .iskeyword (token .string ):
201
+ span_cls = "keyword"
202
+ if token .string in KEYWORD_CONSTANTS :
203
+ span_cls = "keyword_constant"
203
204
span = Span .from_token (token , line_lengths )
204
- yield ColorSpan (span , "keyword" )
205
+ yield ColorSpan (span , span_cls )
205
206
if token .string in IDENTIFIERS_AFTER :
206
207
is_def_name = True
207
208
elif (
You can’t perform that action at this time.
0 commit comments