diff options
Diffstat (limited to 'Doc/library/atexit.rst')
-rw-r--r-- | Doc/library/atexit.rst | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/Doc/library/atexit.rst b/Doc/library/atexit.rst index cc1051b..fc2b5a7 100644 --- a/Doc/library/atexit.rst +++ b/Doc/library/atexit.rst @@ -61,17 +61,22 @@ from a file when it is imported and save the counter's updated value automatically when the program terminates without relying on the application making an explicit call into this module at termination. :: + infile = open("/tmp/counter") try: - _count = int(open("/tmp/counter").read()) + _count = int(infile.read()) except IOError: _count = 0 + finally: + infile.close() + def incrcounter(n): global _count _count = _count + n def savecounter(): - open("/tmp/counter", "w").write("%d" % _count) + with open("/tmp/counter", "w") as outfile: + outfile.write("%d" % _count) import atexit atexit.register(savecounter) |