ztee Namespace Reference

Classes
class	GZipLog

Functions
def	capture
def	log

Variables
tuple	args = parser.parse_args()
tuple	fd = sys.stdout.fileno()
tuple	parser = argparse.ArgumentParser(description="Take all input and write a compressed log file.")

Function Documentation

def ztee.capture	(		fd,
			args
	)

Definition at line 94 of file ztee.py.

def capture(fd, args):
    log_handler = GZipLog(log_file=args.log)

    def sigusr1_handle(*kargs, **kwargs):
        log_handler._sigusr_interrupt = True
        if log_handler.file:
            log_handler.flush_block()

    signal.signal(signal.SIGUSR1, sigusr1_handle)

    try:
        while True:
            try:
                log_handler._sigusr_interrupt = False
                rlist, wlist, xlist = select.select([fd], [], [], 5)
            except select.error as e:
                if e[0] != errno.EINTR: raise
                if not log_handler._sigusr_interrupt: raise

                continue

            if not rlist:
                log_handler.handle_timeout()
                continue

            bytes = os.read(fd, 4096)
            if (bytes == ''):
                break

            log_handler.write(bytes)
            if not args.q:
                sys.stdout.write(bytes)
    finally:
        log_handler.finish()