| 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 | |