Skip to content

Commit 19a266c

Browse files
authored
[3.11] gh-111366: Correctly show custom syntax error messages in the codeop module functions (GH-111384). (#111516)
(cherry picked from commit cd6e0a0)
1 parent bc9f470 commit 19a266c

File tree

3 files changed

+30
-6
lines changed

3 files changed

+30
-6
lines changed

Lib/codeop.py

+13-6
Original file line numberDiff line numberDiff line change
@@ -70,8 +70,7 @@ def _maybe_compile(compiler, source, filename, symbol):
7070
return None
7171
# fallthrough
7272

73-
return compiler(source, filename, symbol)
74-
73+
return compiler(source, filename, symbol, incomplete_input=False)
7574

7675
def _is_syntax_error(err1, err2):
7776
rep1 = repr(err1)
@@ -82,8 +81,12 @@ def _is_syntax_error(err1, err2):
8281
return True
8382
return False
8483

85-
def _compile(source, filename, symbol):
86-
return compile(source, filename, symbol, PyCF_DONT_IMPLY_DEDENT | PyCF_ALLOW_INCOMPLETE_INPUT)
84+
def _compile(source, filename, symbol, incomplete_input=True):
85+
flags = 0
86+
if incomplete_input:
87+
flags |= PyCF_ALLOW_INCOMPLETE_INPUT
88+
flags |= PyCF_DONT_IMPLY_DEDENT
89+
return compile(source, filename, symbol, flags)
8790

8891
def compile_command(source, filename="<input>", symbol="single"):
8992
r"""Compile a command and determine whether it is incomplete.
@@ -114,8 +117,12 @@ class Compile:
114117
def __init__(self):
115118
self.flags = PyCF_DONT_IMPLY_DEDENT | PyCF_ALLOW_INCOMPLETE_INPUT
116119

117-
def __call__(self, source, filename, symbol):
118-
codeob = compile(source, filename, symbol, self.flags, True)
120+
def __call__(self, source, filename, symbol, **kwargs):
121+
flags = self.flags
122+
if kwargs.get('incomplete_input', True) is False:
123+
flags &= ~PyCF_DONT_IMPLY_DEDENT
124+
flags &= ~PyCF_ALLOW_INCOMPLETE_INPUT
125+
codeob = compile(source, filename, symbol, flags, True)
119126
for feature in _features:
120127
if codeob.co_flags & feature.compiler_flag:
121128
self.flags |= feature.compiler_flag

Lib/test/test_codeop.py

+14
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
import warnings
88
from test import support
99
from test.support import warnings_helper
10+
from textwrap import dedent
1011

1112
from codeop import compile_command, PyCF_DONT_IMPLY_DEDENT
1213
import io
@@ -341,6 +342,19 @@ def test_invalid_warning(self):
341342
self.assertRegex(str(w[0].message), 'invalid escape sequence')
342343
self.assertEqual(w[0].filename, '<input>')
343344

345+
def assertSyntaxErrorMatches(self, code, message):
346+
with self.subTest(code):
347+
with self.assertRaisesRegex(SyntaxError, message):
348+
compile_command(code, symbol='exec')
349+
350+
def test_syntax_errors(self):
351+
self.assertSyntaxErrorMatches(
352+
dedent("""\
353+
def foo(x,x):
354+
pass
355+
"""), "duplicate argument 'x' in function definition")
356+
357+
344358

345359
if __name__ == "__main__":
346360
unittest.main()
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
Fix an issue in the :mod:`codeop` that was causing :exc:`SyntaxError`
2+
exceptions raised in the presence of invalid syntax to not contain precise
3+
error messages. Patch by Pablo Galindo

0 commit comments

Comments
 (0)