fix attempt
This commit is contained in:
		
							parent
							
								
									a535d25714
								
							
						
					
					
						commit
						b250109736
					
				@ -79,7 +79,27 @@ class ForensicLogger:
 | 
			
		||||
            console_handler = logging.StreamHandler()
 | 
			
		||||
            console_handler.setFormatter(formatter)
 | 
			
		||||
            self.logger.addHandler(console_handler)
 | 
			
		||||
    
 | 
			
		||||
 | 
			
		||||
    def __getstate__(self):
 | 
			
		||||
        state = self.__dict__.copy()
 | 
			
		||||
        # Exclude the unpickleable 'logger' attribute
 | 
			
		||||
        if 'logger' in state:
 | 
			
		||||
            del state['logger']
 | 
			
		||||
        return state
 | 
			
		||||
 | 
			
		||||
    def __setstate__(self, state):
 | 
			
		||||
        self.__dict__.update(state)
 | 
			
		||||
        # Re-initialize the 'logger' attribute
 | 
			
		||||
        self.logger = logging.getLogger(f'dnsrecon.{self.session_id}')
 | 
			
		||||
        self.logger.setLevel(logging.INFO)
 | 
			
		||||
        formatter = logging.Formatter(
 | 
			
		||||
            '%(asctime)s - %(name)s - %(levelname)s - %(message)s'
 | 
			
		||||
        )
 | 
			
		||||
        if not self.logger.handlers:
 | 
			
		||||
            console_handler = logging.StreamHandler()
 | 
			
		||||
            console_handler.setFormatter(formatter)
 | 
			
		||||
            self.logger.addHandler(console_handler)
 | 
			
		||||
 | 
			
		||||
    def _generate_session_id(self) -> str:
 | 
			
		||||
        """Generate unique session identifier."""
 | 
			
		||||
        return f"dnsrecon_{datetime.now(timezone.utc).strftime('%Y%m%d_%H%M%S')}"
 | 
			
		||||
 | 
			
		||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user