gh-103935: Use `io.open_code()` when executing code in trace and profile modules (GH-103947)

This commit is contained in:
Tian Gao 2023-04-27 13:29:35 -07:00 committed by GitHub
parent bf0b8a9f8d
commit d50f01ad0a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 7 additions and 3 deletions

View File

@ -8,6 +8,7 @@ __all__ = ["run", "runctx", "Profile"]
import _lsprof
import importlib.machinery
import io
import profile as _pyprofile
# ____________________________________________________________
@ -168,7 +169,7 @@ def main():
else:
progname = args[0]
sys.path.insert(0, os.path.dirname(progname))
with open(progname, 'rb') as fp:
with io.open_code(progname) as fp:
code = compile(fp.read(), progname, 'exec')
spec = importlib.machinery.ModuleSpec(name='__main__', loader=None,
origin=progname)

View File

@ -25,6 +25,7 @@
import importlib.machinery
import io
import sys
import time
import marshal
@ -588,7 +589,7 @@ def main():
else:
progname = args[0]
sys.path.insert(0, os.path.dirname(progname))
with open(progname, 'rb') as fp:
with io.open_code(progname) as fp:
code = compile(fp.read(), progname, 'exec')
spec = importlib.machinery.ModuleSpec(name='__main__', loader=None,
origin=progname)

View File

@ -49,6 +49,7 @@ Sample use, programmatically
"""
__all__ = ['Trace', 'CoverageResults']
import io
import linecache
import os
import sys
@ -716,7 +717,7 @@ def main():
sys.argv = [opts.progname, *opts.arguments]
sys.path[0] = os.path.dirname(opts.progname)
with open(opts.progname, 'rb') as fp:
with io.open_code(opts.progname) as fp:
code = compile(fp.read(), opts.progname, 'exec')
# try to emulate __main__ namespace as much as possible
globs = {

View File

@ -0,0 +1 @@
Use :func:`io.open_code` for files to be executed instead of raw :func:`open`