My tools of the trade for python programming.
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 

62 lines
1.3 KiB

#
# wpylib.file.tee module
# File-manipulation utilities: tee output splitter
#
# Wirawan Purwanto
# Created: 20120514
#
# Routines put here are commonly used in my own scripts.
# They are not necessarily suitable for general-purpose uses; evaluate
# your needs and see if they can them as well.
#
#
"""
wpylib.file.tee module
File-manipulation utilities: tee output splitter
This module is part of wpylib project.
"""
class tee_output(object):
"""
Tee class for output file streams.
Supports file-like objects and auto-opened files (if the argument
is a string).
See also: http://shallowsky.com/blog/programming/python-tee.html
"""
def __init__(self, *files, **opts):
fd = []
auto_close = []
mode = opts.get("mode", "w")
for f in files:
if isinstance(f, basestring):
F = open(f, mode=mode)
fd.append(F)
auto_close.append(True)
else:
fd.append(f)
auto_close.append(False)
self.fd = fd
self.auto_close = auto_close
def __del__(self):
self.close()
def close(self):
for (auto,f) in zip(self.auto_close,self.fd):
if auto:
f.close()
else:
f.flush()
self.fd = []
self.auto_close = []
def write(self, s):
for f in self.fd:
f.write(s)
def flush(self):
for f in self.fd:
f.flush()