From 16c7d20b882cc988a2890d05ee545a8fbbfe044f Mon Sep 17 00:00:00 2001 From: ShenYuhan Date: Mon, 9 Nov 2020 11:54:02 +0800 Subject: [PATCH] fix file logger bugs --- paddlehub/serving/device.py | 6 +++--- paddlehub/serving/http_server.py | 7 +++---- paddlehub/utils/log.py | 6 +++--- 3 files changed, 9 insertions(+), 10 deletions(-) diff --git a/paddlehub/serving/device.py b/paddlehub/serving/device.py index d7dd7b96..dae05f7c 100644 --- a/paddlehub/serving/device.py +++ b/paddlehub/serving/device.py @@ -22,8 +22,8 @@ import subprocess from paddlehub.utils import log -filename = 'HubServing-%s.log' % time.strftime("%Y_%m_%d", time.localtime()) -log.logger = log.log_to_file(log.logger, filename) +filename = 'HubServing-%s.log' % time.strftime("%Y_%m_%d_%H%M%S", time.localtime()) +logger = log.get_file_logger(filename) class InferenceDevice(object): @@ -51,7 +51,7 @@ class InferenceDevice(object): zmq.device(zmq.QUEUE, self.frontend, self.backend) except Exception as e: - log.logger.error(traceback.format_exc()) + logger.error(traceback.format_exc()) finally: self.frontend.close() self.backend.close() diff --git a/paddlehub/serving/http_server.py b/paddlehub/serving/http_server.py index 9bf0a592..5aad5818 100644 --- a/paddlehub/serving/http_server.py +++ b/paddlehub/serving/http_server.py @@ -15,7 +15,6 @@ import time import os -import logging import multiprocessing import platform @@ -25,8 +24,8 @@ from paddlehub.serving.device import InferenceServer from paddlehub.serving.client import InferenceClientProxy from paddlehub.utils import utils, log -filename = 'HubServing-%s.log' % time.strftime("%Y_%m_%d", time.localtime()) -log.logger = log.log_to_file(log.logger, filename) +filename = 'HubServing-%s.log' % time.strftime("%Y_%m_%d_%H%M%S", time.localtime()) +logger = log.get_file_logger(filename) if platform.system() == "Windows": @@ -184,7 +183,7 @@ def run(port: int = 8866, client_port: int = 5559, names: list = [], workers: in options = {"bind": "0.0.0.0:%s" % port, "workers": workers, "worker_class": "sync"} StandaloneApplication(create_app(client_port, modules_name=names), options).run() - log.logger.info("PaddleHub-Serving has been stopped.") + logger.info("PaddleHub-Serving has been stopped.") def run_http_server(port: int = 8866, client_port: int = 5559, names: list = [], workers: int = 1): diff --git a/paddlehub/utils/log.py b/paddlehub/utils/log.py index 73090e7d..e58e79fc 100644 --- a/paddlehub/utils/log.py +++ b/paddlehub/utils/log.py @@ -475,20 +475,20 @@ class Table(object): return seprow + content -def log_to_file(logger, filename): +def get_file_logger(filename): ''' Set logger.handler to FileHandler. Args: - logger(Logger): modules info, include module name, version filename(str): filename to logging Examples: .. code-block:: python - logger = log_to_file(logger, 'test.log') + logger = get_file_logger('test.log') logger.logger.info('test_1') ''' + logger = Logger() old_handlers = logger.logger.handlers for handler in old_handlers: logger.logger.removeHandler(handler) -- GitLab