summaryrefslogtreecommitdiffstats
path: root/Tools/freeze
diff options
context:
space:
mode:
authorSerhiy Storchaka <storchaka@gmail.com>2015-03-20 07:21:59 (GMT)
committerSerhiy Storchaka <storchaka@gmail.com>2015-03-20 07:21:59 (GMT)
commit53c3fb186a587e01f88ec9c6b365fd614668d214 (patch)
treeafa6312f355ee29e22ea3c940119200bde1845cd /Tools/freeze
parent8490f5acfee7269ac0eb41257a3a214dfb31a655 (diff)
downloadcpython-53c3fb186a587e01f88ec9c6b365fd614668d214.zip
cpython-53c3fb186a587e01f88ec9c6b365fd614668d214.tar.gz
cpython-53c3fb186a587e01f88ec9c6b365fd614668d214.tar.bz2
Issue #22826: The result of open() in Tools/freeze/bkfile.py is now better
compatible with regular files (in particular it now supports the context management protocol).
Diffstat (limited to 'Tools/freeze')
-rw-r--r--Tools/freeze/bkfile.py67
-rwxr-xr-xTools/freeze/freeze.py17
-rw-r--r--Tools/freeze/makefreeze.py54
3 files changed, 51 insertions, 87 deletions
diff --git a/Tools/freeze/bkfile.py b/Tools/freeze/bkfile.py
index 6abacc9..20a70b0 100644
--- a/Tools/freeze/bkfile.py
+++ b/Tools/freeze/bkfile.py
@@ -1,49 +1,26 @@
from builtins import open as _orig_open
-class _BkFile:
- def __init__(self, file, mode, bufsize):
- import os
- self.__filename = file
- self.__backup = file + '~'
- try:
- os.unlink(self.__backup)
- except OSError:
- pass
- try:
- os.rename(file, self.__backup)
- except OSError:
- self.__backup = None
- self.__file = _orig_open(file, mode, bufsize)
- self.closed = self.__file.closed
- self.fileno = self.__file.fileno
- self.flush = self.__file.flush
- self.isatty = self.__file.isatty
- self.mode = self.__file.mode
- self.name = self.__file.name
- self.read = self.__file.read
- try:
- self.readinto = self.__file.readinto
- except AttributeError:
- pass
- self.readline = self.__file.readline
- self.readlines = self.__file.readlines
- self.seek = self.__file.seek
- self.tell = self.__file.tell
- self.truncate = self.__file.truncate
- self.write = self.__file.write
- self.writelines = self.__file.writelines
-
- def close(self):
- self.__file.close()
- if self.__backup is None:
- return
- import filecmp
- if filecmp.cmp(self.__backup, self.__filename, shallow = 0):
- import os
- os.unlink(self.__filename)
- os.rename(self.__backup, self.__filename)
-
-def open(file, mode = 'r', bufsize = -1):
+def open(file, mode='r', bufsize=-1):
if 'w' not in mode:
return _orig_open(file, mode, bufsize)
- return _BkFile(file, mode, bufsize)
+ import os
+ backup = file + '~'
+ try:
+ os.unlink(backup)
+ except OSError:
+ pass
+ try:
+ os.rename(file, backup)
+ except OSError:
+ return _orig_open(file, mode, bufsize)
+ f = _orig_open(file, mode, bufsize)
+ _orig_close = f.close
+ def close():
+ _orig_close()
+ import filecmp
+ if filecmp.cmp(backup, file, shallow=False):
+ import os
+ os.unlink(file)
+ os.rename(backup, file)
+ f.close = close
+ return f
diff --git a/Tools/freeze/freeze.py b/Tools/freeze/freeze.py
index e0c6c2c..15ddbe6 100755
--- a/Tools/freeze/freeze.py
+++ b/Tools/freeze/freeze.py
@@ -439,25 +439,17 @@ def main():
frozendllmain_c, os.path.basename(extensions_c)] + files
maindefn = checkextensions_win32.CExtension( '__main__', xtras )
frozen_extensions.append( maindefn )
- outfp = open(makefile, 'w')
- try:
+ with open(makefile, 'w') as outfp:
winmakemakefile.makemakefile(outfp,
locals(),
frozen_extensions,
os.path.basename(target))
- finally:
- outfp.close()
return
# generate config.c and Makefile
builtins.sort()
- infp = open(config_c_in)
- outfp = bkfile.open(config_c, 'w')
- try:
+ with open(config_c_in) as infp, bkfile.open(config_c, 'w') as outfp:
makeconfig.makeconfig(infp, outfp, builtins)
- finally:
- outfp.close()
- infp.close()
cflags = ['$(OPT)']
cppflags = defines + includes
@@ -475,11 +467,8 @@ def main():
files + supp_sources + addfiles + libs + \
['$(MODLIBS)', '$(LIBS)', '$(SYSLIBS)']
- outfp = bkfile.open(makefile, 'w')
- try:
+ with bkfile.open(makefile, 'w') as outfp:
makemakefile.makemakefile(outfp, somevars, files, base_target)
- finally:
- outfp.close()
# Done!
diff --git a/Tools/freeze/makefreeze.py b/Tools/freeze/makefreeze.py
index ef18ec7..64e3e6b 100644
--- a/Tools/freeze/makefreeze.py
+++ b/Tools/freeze/makefreeze.py
@@ -39,36 +39,34 @@ def makefreeze(base, dict, debug=0, entry_point=None, fail_import=()):
mangled = "__".join(mod.split("."))
if m.__code__:
file = 'M_' + mangled + '.c'
- outfp = bkfile.open(base + file, 'w')
- files.append(file)
- if debug:
- print("freezing", mod, "...")
- str = marshal.dumps(m.__code__)
- size = len(str)
- if m.__path__:
- # Indicate package by negative size
- size = -size
- done.append((mod, mangled, size))
- writecode(outfp, mangled, str)
- outfp.close()
+ with bkfile.open(base + file, 'w') as outfp:
+ files.append(file)
+ if debug:
+ print("freezing", mod, "...")
+ str = marshal.dumps(m.__code__)
+ size = len(str)
+ if m.__path__:
+ # Indicate package by negative size
+ size = -size
+ done.append((mod, mangled, size))
+ writecode(outfp, mangled, str)
if debug:
print("generating table of frozen modules")
- outfp = bkfile.open(base + 'frozen.c', 'w')
- for mod, mangled, size in done:
- outfp.write('extern unsigned char M_%s[];\n' % mangled)
- outfp.write(header)
- for mod, mangled, size in done:
- outfp.write('\t{"%s", M_%s, %d},\n' % (mod, mangled, size))
- outfp.write('\n')
- # The following modules have a NULL code pointer, indicating
- # that the frozen program should not search for them on the host
- # system. Importing them will *always* raise an ImportError.
- # The zero value size is never used.
- for mod in fail_import:
- outfp.write('\t{"%s", NULL, 0},\n' % (mod,))
- outfp.write(trailer)
- outfp.write(entry_point)
- outfp.close()
+ with bkfile.open(base + 'frozen.c', 'w') as outfp:
+ for mod, mangled, size in done:
+ outfp.write('extern unsigned char M_%s[];\n' % mangled)
+ outfp.write(header)
+ for mod, mangled, size in done:
+ outfp.write('\t{"%s", M_%s, %d},\n' % (mod, mangled, size))
+ outfp.write('\n')
+ # The following modules have a NULL code pointer, indicating
+ # that the frozen program should not search for them on the host
+ # system. Importing them will *always* raise an ImportError.
+ # The zero value size is never used.
+ for mod in fail_import:
+ outfp.write('\t{"%s", NULL, 0},\n' % (mod,))
+ outfp.write(trailer)
+ outfp.write(entry_point)
return files