fix: xx.keyword that keyword should not be highlight

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

Author: yihong0618

Lib/_pyrepl/utils.py

@@ -196,7 +196,8 @@ 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):
+                elif (keyword.iskeyword(token.string)
+                      and not (prev_token and prev_token.type == T.OP and prev_token.string == ".")):
                     span = Span.from_token(token, line_lengths)
                     yield ColorSpan(span, "keyword")
                     if token.string in IDENTIFIERS_AFTER:
@@ -208,7 +209,8 @@ def gen_colors_from_token_stream(
                 ):
                     span = Span.from_token(token, line_lengths)
                     yield ColorSpan(span, "soft_keyword")
-                elif token.string in BUILTINS:
+                elif (token.string in BUILTINS
+                      and not (prev_token and prev_token.type == T.OP and prev_token.string == ".")):
                     span = Span.from_token(token, line_lengths)
                     yield ColorSpan(span, "builtin")
 

Lib/test/test_pyrepl/test_utils.py

@@ -1,6 +1,6 @@
 from unittest import TestCase
 
-from _pyrepl.utils import str_width, wlen, prev_next_window
+from _pyrepl.utils import str_width, wlen, prev_next_window, gen_colors
 
 
 class TestUtils(TestCase):
@@ -60,3 +60,38 @@ def gen_raise():
         self.assertEqual(next(pnw), (3, 4, None))
         with self.assertRaises(ZeroDivisionError):
             next(pnw)
+
+    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 '.'
+        ]
+        for code, expected_highlights in no_highlight_cases:
+            with self.subTest(code=code):
+                colors = list(gen_colors(code))
+                # Extract (text, tag) pairs for comparison
+                actual_highlights = []
+                for color in colors:
+                    span_text = code[color.span.start:color.span.end + 1]
+                    actual_highlights.append((span_text, color.tag))
+                self.assertEqual(actual_highlights, expected_highlights,
+                               f"In '{code}', expected {expected_highlights}, got {actual_highlights}")
+        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):
+                colors = list(gen_colors(code))
+                # Extract (text, tag) pairs for comparison
+                actual_highlights = []
+                for color in colors:
+                    span_text = code[color.span.start:color.span.end + 1]
+                    actual_highlights.append((span_text, color.tag))
+                self.assertEqual(actual_highlights, expected_highlights,
+                               f"In '{code}', expected {expected_highlights}, got {actual_highlights}")

Misc/NEWS.d/next/Core_and_Builtins/2025-09-01-16-09-02.gh-issue-138318.t-WEN5.rst

@@ -0,0 +1 @@
+fix: xxx.keyword the keyword should not highlight in new repl