diff options
author | Cheryl Sabella <cheryl.sabella@gmail.com> | 2020-06-04 23:40:24 (GMT) |
---|---|---|
committer | GitHub <noreply@github.com> | 2020-06-04 23:40:24 (GMT) |
commit | 052d3fc0907be253cfd64b2c737a0b0aca586011 (patch) | |
tree | 3d50f8cb5dbbdab51518e443c456b24f7172aa0d /Lib | |
parent | 3744ed2c9c0b3905947602fc375de49533790cb9 (diff) | |
download | cpython-052d3fc0907be253cfd64b2c737a0b0aca586011.zip cpython-052d3fc0907be253cfd64b2c737a0b0aca586011.tar.gz cpython-052d3fc0907be253cfd64b2c737a0b0aca586011.tar.bz2 |
bpo-40807: Show warnings once from codeop._maybe_compile (#20486)
* bpo-40807: Show warnings once from codeop._maybe_compile
* Move catch_warnings
* news
Co-authored-by: Terry Jan Reedy <tjreedy@udel.edu>
Diffstat (limited to 'Lib')
-rw-r--r-- | Lib/codeop.py | 22 | ||||
-rw-r--r-- | Lib/test/test_codeop.py | 5 |
2 files changed, 18 insertions, 9 deletions
diff --git a/Lib/codeop.py b/Lib/codeop.py index 835e68c..7e192ea 100644 --- a/Lib/codeop.py +++ b/Lib/codeop.py @@ -57,6 +57,7 @@ Compile(): """ import __future__ +import warnings _features = [getattr(__future__, fname) for fname in __future__.all_feature_names] @@ -83,15 +84,18 @@ def _maybe_compile(compiler, source, filename, symbol): except SyntaxError: pass - try: - code1 = compiler(source + "\n", filename, symbol) - except SyntaxError as e: - err1 = e - - try: - code2 = compiler(source + "\n\n", filename, symbol) - except SyntaxError as e: - err2 = e + # Suppress warnings after the first compile to avoid duplication. + with warnings.catch_warnings(): + warnings.simplefilter("ignore") + try: + code1 = compiler(source + "\n", filename, symbol) + except SyntaxError as e: + err1 = e + + try: + code2 = compiler(source + "\n\n", filename, symbol) + except SyntaxError as e: + err2 = e try: if code: diff --git a/Lib/test/test_codeop.py b/Lib/test/test_codeop.py index 0c5e362f..45cb1a7 100644 --- a/Lib/test/test_codeop.py +++ b/Lib/test/test_codeop.py @@ -303,6 +303,11 @@ class CodeopTests(unittest.TestCase): self.assertNotEqual(compile_command("a = 1\n", "abc").co_filename, compile("a = 1\n", "def", 'single').co_filename) + def test_warning(self): + # Test that the warning is only returned once. + with support.check_warnings((".*literal", SyntaxWarning)) as w: + compile_command("0 is 0") + self.assertEqual(len(w.warnings), 1) if __name__ == "__main__": unittest.main() |