fix: only the buildin words

Signed-off-by: yihong0618 <zouzou0208@gmail.com>

Author: yihong0618

Lib/_pyrepl/utils.py

@@ -196,8 +196,7 @@ def gen_colors_from_token_stream(
                     is_def_name = False
                     span = Span.from_token(token, line_lengths)
                     yield ColorSpan(span, "definition")
-                elif (keyword.iskeyword(token.string)
-                      and not (prev_token and prev_token.type == T.OP and prev_token.string == ".")):
+                elif keyword.iskeyword(token.string):
                     span = Span.from_token(token, line_lengths)
                     yield ColorSpan(span, "keyword")
                     if token.string in IDENTIFIERS_AFTER:

Lib/test/test_pyrepl/test_utils.py

@@ -63,11 +63,9 @@ def gen_raise():
 
     def test_gen_colors_keyword_highlighting(self):
         no_highlight_cases = [
-            ("a.set", [(".", "op")]),  # 'set' should not be highlighted after '.'
-            ("a.def", [(".", "op")]),  # 'def' should not be highlighted after '.'
-            ("obj.class", [(".", "op")]),  # 'class' should not be highlighted after '.'
-            ("obj.list", [(".", "op")]),  # 'list' should not be highlighted after '.'
-            ("obj.match", [(".", "op")]),  # 'match' should not be highlighted after '.'
+            ("a.set", [(".", "op")]),
+            ("obj.list", [(".", "op")]),
+            ("obj.match", [(".", "op")]),
         ]
         for code, expected_highlights in no_highlight_cases:
             with self.subTest(code=code):
@@ -82,8 +80,6 @@ def test_gen_colors_keyword_highlighting(self):
         highlight_cases = [
             ("set", [("set", "builtin")]),
             ("list", [("list", "builtin")]),
-            ("def func():", [("def", "keyword"), ("func", "definition"), ("(", "op"), (")", "op"), (":", "op")]),
-            ("class A:", [("class", "keyword"), ("A", "definition"), (":", "op")]),
         ]
         for code, expected_highlights in highlight_cases:
             with self.subTest(code=code):