chore: Add pylint for DB-GPT core lib (#1076)

dbgpt/util/utils.py

@@ -5,7 +5,7 @@ import asyncio
 import logging
 import logging.handlers
 import os
-from typing import Any, List
+from typing import Any, List, Optional, cast
 
 from dbgpt.configs.model_config import LOGDIR
 
@@ -28,19 +28,25 @@ def _get_logging_level() -> str:
     return os.getenv("DBGPT_LOG_LEVEL", "INFO")
 
 
-def setup_logging_level(logging_level=None, logger_name: str = None):
+def setup_logging_level(
+    logging_level: Optional[str] = None, logger_name: Optional[str] = None
+):
     if not logging_level:
         logging_level = _get_logging_level()
     if type(logging_level) is str:
         logging_level = logging.getLevelName(logging_level.upper())
     if logger_name:
         logger = logging.getLogger(logger_name)
-        logger.setLevel(logging_level)
+        logger.setLevel(cast(str, logging_level))
     else:
         logging.basicConfig(level=logging_level, encoding="utf-8")
 
 
-def setup_logging(logger_name: str, logging_level=None, logger_filename: str = None):
+def setup_logging(
+    logger_name: str,
+    logging_level: Optional[str] = None,
+    logger_filename: Optional[str] = None,
+):
     if not logging_level:
         logging_level = _get_logging_level()
     logger = _build_logger(logger_name, logging_level, logger_filename)
@@ -74,7 +80,11 @@ def get_gpu_memory(max_gpus=None):
     return gpu_memory
 
 
-def _build_logger(logger_name, logging_level=None, logger_filename: str = None):
+def _build_logger(
+    logger_name,
+    logging_level: Optional[str] = None,
+    logger_filename: Optional[str] = None,
+):
     global handler
 
     formatter = logging.Formatter(
@@ -111,14 +121,14 @@ def get_or_create_event_loop() -> asyncio.BaseEventLoop:
     try:
         loop = asyncio.get_event_loop()
         assert loop is not None
-        return loop
+        return cast(asyncio.BaseEventLoop, loop)
     except RuntimeError as e:
         if not "no running event loop" in str(e) and not "no current event loop" in str(
             e
         ):
             raise e
         logging.warning("Cant not get running event loop, create new event loop now")
-    return asyncio.get_event_loop_policy().new_event_loop()
+    return cast(asyncio.BaseEventLoop, asyncio.get_event_loop_policy().new_event_loop())
 
 
 def logging_str_to_uvicorn_level(log_level_str):
@@ -152,7 +162,7 @@ class EndpointFilter(logging.Filter):
         return record.getMessage().find(self._path) == -1
 
 
-def setup_http_service_logging(exclude_paths: List[str] = None):
+def setup_http_service_logging(exclude_paths: Optional[List[str]] = None):
     """Setup http service logging
 
     Now just disable some logs