Python StreamHandler.__exit__の例

コード例 #1

ファイル: debugging.py プロジェクト: ukaluzhny/uutils

class ConciseLog(object):
    def __init__(self, f = sys.stdout, level = "info"):
        frmt = '{record.message}'
        if level == "info":
            self.debug = NullHandler(level = DEBUG)
            self.info = StreamHandler(f, level = INFO, format_string = frmt)
        else:    
            self.debug = StreamHandler(f, level = DEBUG, format_string = frmt)
            self.info = None
    def __enter__(self):
        self.debug.__enter__()
        if self.info: self.info.__enter__()
    def __exit__(self, exc_type, exc_value, traceback):
        if self.info: 
            self.info.__exit__(exc_type, exc_value, traceback)
        self.debug.__exit__(exc_type, exc_value, traceback)

コード例 #2

class ConciseLog(object):
    def __init__(self, f=sys.stdout, level="info"):
        frmt = '{record.message}'
        if level == "info":
            self.debug = NullHandler(level=DEBUG)
            self.info = StreamHandler(f, level=INFO, format_string=frmt)
        else:
            self.debug = StreamHandler(f, level=DEBUG, format_string=frmt)
            self.info = None

    def __enter__(self):
        self.debug.__enter__()
        if self.info: self.info.__enter__()

    def __exit__(self, exc_type, exc_value, traceback):
        if self.info:
            self.info.__exit__(exc_type, exc_value, traceback)
        self.debug.__exit__(exc_type, exc_value, traceback)