mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-07-19 01:05:26 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

bpo-45019: Add a tool to generate list of modules to include for frozen modules (gh-27980)

Browse source 
Frozen modules must be added to several files in order to work properly. Before this change this had to be done manually. Here we add a tool to generate the relevant lines in those files instead. This helps us avoid mistakes and omissions.

https://bugs.python.org/issue45019
This commit is contained in:
Eric Snow 2021-08-30 17:25:11 -06:00 • committed by GitHub
parent 5246dbc2a1
commit 044e8d866f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
19 changed files with 833 additions and 199 deletions
Download patch file Download diff file Expand all files Collapse all files

36
Tools/scripts/update_file.py
View file

@ -6,23 +6,47 @@ This avoids wholesale rebuilds when a code (re)generation phase does not
actually change the in-tree generated code.
"""
import contextlib
import os
import os.path
import sys
def main(old_path, new_path):
with open(old_path, 'rb') as f:
@contextlib.contextmanager
def updating_file_with_tmpfile(filename, tmpfile=None):
"""A context manager for updating a file via a temp file.
The context manager provides two open files: the source file open
for reading, and the temp file, open for writing.
Upon exiting: both files are closed, and the source file is replaced
with the temp file.
"""
# XXX Optionally use tempfile.TemporaryFile?
if not tmpfile:
tmpfile = filename + '.tmp'
elif os.path.isdir(tmpfile):
tmpfile = os.path.join(tmpfile, filename + '.tmp')
with open(tmpfile, 'w') as outfile:
with open(filename) as infile:
yield infile, outfile
update_file_with_tmpfile(filename, tmpfile)
def update_file_with_tmpfile(filename, tmpfile):
with open(filename, 'rb') as f:
old_contents = f.read()
with open(new_path, 'rb') as f:
with open(tmpfile, 'rb') as f:
new_contents = f.read()
if old_contents != new_contents:
os.replace(new_path, old_path)
os.replace(tmpfile, filename)
else:
os.unlink(new_path)
os.unlink(tmpfile)
if __name__ == '__main__':
if len(sys.argv) != 3:
print("Usage: %s <path to be updated> <path with new contents>" % (sys.argv[0],))
sys.exit(1)
main(sys.argv[1], sys.argv[2])
update_file_with_tmpfile(sys.argv[1], sys.argv[2])