From f81ac9f4d47d978bf1209cbd4fd7826d3dbd3582 Mon Sep 17 00:00:00 2001 From: chenzx Date: Thu, 6 Apr 2023 11:16:41 +0800 Subject: [PATCH] [FIX] update logger --- learnware/logger.py | 5 ++++- learnware/market/database_ops.py | 9 ++++----- 2 files changed, 8 insertions(+), 6 deletions(-) diff --git a/learnware/logger.py b/learnware/logger.py index e838dc9..ddf0534 100644 --- a/learnware/logger.py +++ b/learnware/logger.py @@ -4,7 +4,7 @@ import logging.handlers from .config import C -def get_module_logger(module_name, level=None): +def get_module_logger(module_name, level=logging.INFO): """ Get a logger for a specific module. :param module_name: str @@ -20,6 +20,9 @@ def get_module_logger(module_name, level=None): level = C.logging_level # Get logger. + console_handler = logging.StreamHandler() + console_handler.setLevel('INFO') module_logger = logging.getLogger(module_name) module_logger.setLevel(level) + module_logger.addHandler(console_handler) return module_logger diff --git a/learnware/market/database_ops.py b/learnware/market/database_ops.py index e88108a..9ecd173 100644 --- a/learnware/market/database_ops.py +++ b/learnware/market/database_ops.py @@ -7,7 +7,7 @@ import json ROOT_PATH = os.path.dirname(os.path.abspath(__file__)) DB_PATH = os.path.join(ROOT_PATH, "market.db") -LOGGER = get_module_logger("market", level="INFO") +LOGGER = get_module_logger("market") def init_empty_db(func): @@ -27,8 +27,8 @@ def init_empty_db(func): LOGGER.info("Database Built!") conn.commit() conn.close() - else: - pass + func() + return wrapper @init_empty_db @@ -41,9 +41,8 @@ def delete_learnware_from_db(id:str): @init_empty_db def load_market_from_db(): - if not os.path.exists(DB_PATH): - init_empty_db() conn = sqlite3.connect(DB_PATH) + LOGGER.info("Reload from Database") c = conn.cursor() cursor = c.execute("SELECT id, name, semantic_spec, model_path, stat_spec_path from LEARNWARE")