hydrus/include/HydrusLogger.py

import HydrusConstants as HC
import HydrusData
import os
import sys
import threading
import time

class HydrusLogger( object ):
    
    def __init__( self, base_dir, prefix ):
        
        self._log_path_base = os.path.join( base_dir, prefix )
        self._lock = threading.Lock()
        
    
    def __enter__( self ):
        
        self._previous_sys_stdout = sys.stdout
        self._previous_sys_stderr = sys.stderr
        
        self._problem_with_previous_stdout = False
        
        self._OpenLog()
        
        sys.stdout = self
        sys.stderr = self
        
        return self
        
    
    def __exit__( self, exc_type, exc_val, exc_tb ):
        
        sys.stdout = self._previous_sys_stdout
        sys.stderr = self._previous_sys_stderr
        
        self._CloseLog()
        
        return False
        
    
    def _CloseLog( self ):
        
        self._log_file.close()
        
    
    def _GetLogPath( self ):
        
        current_time_struct = time.localtime()
        
        ( current_year, current_month ) = ( current_time_struct.tm_year, current_time_struct.tm_mon )
        
        log_path = self._log_path_base + ' - ' + str( current_year ) + '-' + str( current_month ) + '.log'
        
        return log_path
        
    
    def _OpenLog( self ):
        
        self._log_path = self._GetLogPath()
        
        self._log_file = open( self._log_path, 'a' )
        
    
    def _SwitchToANewLogFileIfDue( self ):
        
        correct_log_path = self._GetLogPath()
        
        if correct_log_path != self._log_path:
            
            self._CloseLog()
            
            self._OpenLog()
            
        
    
    def flush( self ):
        
        with self._lock:
            
            if not self._problem_with_previous_stdout:
                
                try:
                    
                    self._previous_sys_stdout.flush()
                    
                except IOError:
                    
                    self._problem_with_previous_stdout = True
                    
                
            
            self._log_file.flush()
            
            self._SwitchToANewLogFileIfDue()
            
        
    
    def isatty( self ):
        
        return False
        
    
    def write( self, value ):
        
        with self._lock:
            
            if value in ( os.linesep, '\n' ):
                
                prefix = ''
                
            else:
                
                prefix = time.strftime( '%Y/%m/%d %H:%M:%S: ' )
                
            
            message = prefix + value
            
            if not self._problem_with_previous_stdout:
                
                stdout_message = HydrusData.ToByteString( message.replace( u'\u2026', '...' ) )
                
                try:
                    
                    self._previous_sys_stdout.write( stdout_message )
                    
                except IOError:
                    
                    self._problem_with_previous_stdout = True
                    
                
            
            log_message = HydrusData.ToByteString( message )
            
            self._log_file.write( log_message )