修复import

Author: chenhao

snippets/decorators.py

@@ -170,7 +170,8 @@ def wrapped(data: Iterable, *args, **kwargs):
             def _func(x):
                 return func(x, *args, **kwargs)
             rs_iter = executors.map(_func, data)
-            rs = tqdm(rs_iter)
+            total = None if not hasattr(data, '__len__') else len(data)
+            rs = tqdm(rs_iter, total=total)
             if return_list:
                 return list(rs)
             return rs

snippets/logs.py

@@ -7,6 +7,7 @@
 '''
 import logging
 import os
+from logging.handlers import TimedRotatingFileHandler
 
 
 logger = logging.getLogger(__name__)
@@ -62,8 +63,9 @@ def getlog_detail(name, level, format_type: str = "simple", do_print=True, do_fi
     if do_file:
         log_dir = log_dir or os.environ.get("LOG_DIR", "/tmp/logs")
         file_path = os.path.join(log_dir, name + ".log")
+        os.makedirs(log_dir, exist_ok=True)
         if file_type == "time_rotate":
-            filehandler = logging.handlers.TimedRotatingFileHandler(file_path, **file_config)
+            filehandler = TimedRotatingFileHandler(file_path, **file_config)
             filehandler.suffix = "%Y-%m-%d_%H-%M-%S.log"  # 设置历史文件 后缀
             filehandler.setFormatter(fmt)
             rs_logger.addHandler(filehandler)