File size: 968 Bytes
e3b27f8
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
import logging

def setup_logger(name, chat_id=None):
    logger = logging.getLogger(name)
    if not logger.handlers:
        logger.setLevel(logging.INFO)
        
        # Modify the formatter to include chat_id if available
        if chat_id:
            formatter = logging.Formatter(f'%(asctime)s - %(name)s - %(levelname)s - [Chat ID: {chat_id}] - %(message)s')
        else:
            formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
        
        # Console handler
        console_handler = logging.StreamHandler()
        console_handler.setFormatter(formatter)
        logger.addHandler(console_handler)
        
        # File Handler - Error Level (uncomment if needed)
        # error_file_handler = logging.FileHandler('error.log')
        # error_file_handler.setFormatter(formatter)
        # error_file_handler.setLevel(logging.ERROR)
        # logger.addHandler(error_file_handler)
    
    return logger