33 lines
987 B
Python
33 lines
987 B
Python
import os, socket, logging
|
|
|
|
host = os.environ.get('VERSION_MANAGEMENT_SERVICE_HOST', default='10.10.43.240')
|
|
service = os.environ.get('VERSION_MANAGEMENT_SERVICE_PORT', default=31090)
|
|
|
|
class PrintHandler(logging.Handler):
|
|
def emit(self, record):
|
|
msg = self.format(record)
|
|
print(msg)
|
|
|
|
def get_sock_name():
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
sock.connect((host, service))
|
|
return sock.getsockname()[0]
|
|
|
|
def setup_logger():
|
|
logger = logging.getLogger(__name__)
|
|
logger.setLevel(logging.INFO)
|
|
log_file='dashboard/version-management.log'
|
|
formatter = logging.Formatter(f'%(asctime)s %(levelname)s {get_sock_name()} : %(message)s')
|
|
|
|
file_handler = logging.FileHandler(log_file)
|
|
file_handler.setFormatter(formatter)
|
|
|
|
print_handler = PrintHandler()
|
|
print_handler.setFormatter(formatter)
|
|
|
|
logger.addHandler(file_handler)
|
|
logger.addHandler(print_handler)
|
|
|
|
return logger
|
|
|
|
logger = setup_logger() |