stock-tracker

backend/.venv/lib/python3.9/site-packages/akshare/utils/tqdm.py

@@ -0,0 +1,27 @@
+def get_tqdm(enable: bool = True):
+    """
+    返回适用于当前环境的 tqdm 对象。
+
+    Args:
+        enable (bool): 是否启用进度条。默认为 True。
+
+    Returns:
+        tqdm 对象。
+    """
+    if not enable:
+        # 如果进度条被禁用，返回一个不显示进度条的 tqdm 对象
+        return lambda iterable, *args, **kwargs: iterable
+
+    try:
+        # 尝试检查是否在 jupyter notebook 环境中，有利于退出进度条
+        # noinspection PyUnresolvedReferences
+        shell = get_ipython().__class__.__name__
+        if shell == "ZMQInteractiveShell":
+            from tqdm.notebook import tqdm
+        else:
+            from tqdm import tqdm
+    except (NameError, ImportError):
+        # 如果不在 Jupyter 环境中，就使用标准 tqdm
+        from tqdm import tqdm
+
+    return tqdm