dict key matching regex no longer matches beyond the end of a key

arian-deimling · arian-deimling · commit 3f20d8e2c27d · 2021-09-28T00:45:18.000-04:00
diff --git a/bpython/line.py b/bpython/line.py
@@ -34,7 +34,9 @@ def current_word(cursor_offset: int, line: str) -> Optional[LinePart]:
     return LinePart(start, end, word)
 
 
-_current_dict_key_re = LazyReCompile(r"""[\w_][\w0-9._]*\[(.*)""")
+_current_dict_key_re = LazyReCompile(
+    r"""[\w_][\w0-9._]*\[('[^']*|"[^"]*|[\w0-9._(), '"]*)"""
+)
 
 
 def current_dict_key(cursor_offset: int, line: str) -> Optional[LinePart]:
@@ -45,7 +47,9 @@ def current_dict_key(cursor_offset: int, line: str) -> Optional[LinePart]:
     return None
 
 
-_current_dict_re = LazyReCompile(r"""([\w_][\w0-9._]*)\[(.*)""")
+_current_dict_re = LazyReCompile(
+    r"""([\w_][\w0-9._]*)\[('[^']*|"[^"]*|[\w0-9._(), '"]*)"""
+)
 
 
 def current_dict(cursor_offset: int, line: str) -> Optional[LinePart]:
