p115client 0.0.5.8.4__tar.gz → 0.0.5.8.6__tar.gz

{p115client-0.0.5.8.4 → p115client-0.0.5.8.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: p115client
-Version: 0.0.5.8.4
+Version: 0.0.5.8.6
 Summary: Python 115 webdisk client.
 Home-page: https://github.com/ChenyangGao/p115client
 License: MIT

{p115client-0.0.5.8.4 → p115client-0.0.5.8.6}/p115client/client.py

@@ -3139,6 +3139,64 @@ class P115OpenClient(ClientRequestMixin):
             }
         return self.request(url=api, params=payload, async_=async_, **request_kwargs)
 
+    @overload
+    def fs_star_set(
+        self, 
+        payload: int | str | Iterable[int | str] | dict, 
+        /, 
+        star: bool = True, 
+        base_url: bool | str | Callable[[], str] = False, 
+        *, 
+        async_: Literal[False] = False, 
+        **request_kwargs, 
+    ) -> dict:
+        ...
+    @overload
+    def fs_star_set(
+        self, 
+        payload: int | str | Iterable[int | str] | dict, 
+        /, 
+        star: bool = True, 
+        base_url: bool | str | Callable[[], str] = False, 
+        *, 
+        async_: Literal[True], 
+        **request_kwargs, 
+    ) -> Coroutine[Any, Any, dict]:
+        ...
+    def fs_star_set(
+        self, 
+        payload: int | str | Iterable[int | str] | dict, 
+        /, 
+        star: bool = True, 
+        base_url: bool | str | Callable[[], str] = False, 
+        *, 
+        async_: Literal[False, True] = False, 
+        **request_kwargs, 
+    ) -> dict | Coroutine[Any, Any, dict]:
+        """为文件或目录设置或取消星标，此接口是对 `fs_update_open` 的封装
+
+        .. note::
+            即使其中任何一个 id 目前已经被删除，也可以操作成功
+
+        :payload:
+            - file_id: int | str    💡 只能传入 1 个
+            - file_id[0]: int | str 💡 如果有多个，则按顺序给出
+            - file_id[1]: int | str
+            - ...
+            - star: 0 | 1 = 1
+        """
+        api = complete_webapi("/files/star", base_url=base_url)
+        if isinstance(payload, (int, str)):
+            payload = {"file_id": payload, "star": int(star)}
+        elif not isinstance(payload, dict):
+            payload = {f"file_id[{i}]": id for i, id in enumerate(payload)}
+            if not payload:
+                return {"state": False, "message": "no op"}
+            payload["star"] = int(star)
+        else:
+            payload = {"star": int(star), **payload}
+        return self.fs_update(payload, async_=async_, **request_kwargs)
+
     @overload
     def fs_update(
         self, 
@@ -3175,13 +3233,16 @@ class P115OpenClient(ClientRequestMixin):
         POST https://proapi.115.com/open/ufile/update
 
         .. hint::
-            类似于 `P115Client.fs_edit_app`
+            即使文件已经被删除，也可以操作成功
 
         .. note::
             https://www.yuque.com/115yun/open/gyrpw5a0zc4sengm
 
         :payload:
-            - file_id: int | str
+            - file_id: int | str    💡 只能传入 1 个
+            - file_id[0]: int | str 💡 如果有多个，则按顺序给出
+            - file_id[1]: int | str
+            - ...
             - file_name: str = <default> 💡 文件名
             - star: 0 | 1 = <default> 💡 是否星标：0:取消星标 1:设置星标
             - ...
@@ -3970,6 +4031,7 @@ class P115OpenClient(ClientRequestMixin):
     fs_mkdir_open = fs_mkdir
     fs_move_open = fs_move
     fs_search_open = fs_search
+    fs_star_set_open = fs_star_set
     fs_update_open = fs_update
     recyclebin_clean_open = recyclebin_clean
     recyclebin_list_open = recyclebin_list

{p115client-0.0.5.8.4 → p115client-0.0.5.8.6}/p115client/tool/download.py

@@ -25,6 +25,7 @@ from shutil import rmtree
 from threading import Lock
 from time import time
 from typing import cast, overload, Any, Final, Literal, TypedDict
+from types import EllipsisType
 from urllib.parse import quote, urlsplit
 from urllib.request import urlopen, Request
 from uuid import uuid4
@@ -257,7 +258,7 @@ def iter_files_with_url(
     use_star: None | bool = False, 
     escape: None | bool | Callable[[str], str] = True, 
     normalize_attr: Callable[[dict], dict] = normalize_attr, 
-    id_to_dirnode: None | dict[int, tuple[str, int] | DirNode] = None, 
+    id_to_dirnode: None | EllipsisType | dict[int, tuple[str, int] | DirNode] = None, 
     app: str = "web", 
     raise_for_changed_count: bool = False, 
     user_agent: str = "", 
@@ -278,7 +279,7 @@ def iter_files_with_url(
     use_star: None | bool = False, 
     escape: None | bool | Callable[[str], str] = True, 
     normalize_attr: Callable[[dict], dict] = normalize_attr, 
-    id_to_dirnode: None | dict[int, tuple[str, int] | DirNode] = None, 
+    id_to_dirnode: None | EllipsisType | dict[int, tuple[str, int] | DirNode] = None, 
     app: str = "web", 
     raise_for_changed_count: bool = False, 
     user_agent: str = "", 
@@ -298,7 +299,7 @@ def iter_files_with_url(
     use_star: None | bool = False, 
     escape: None | bool | Callable[[str], str] = True, 
     normalize_attr: Callable[[dict], dict] = normalize_attr, 
-    id_to_dirnode: None | dict[int, tuple[str, int] | DirNode] = None, 
+    id_to_dirnode: None | EllipsisType | dict[int, tuple[str, int] | DirNode] = None, 
     app: str = "web", 
     raise_for_changed_count: bool = False, 
     user_agent: str = "", 
@@ -423,7 +424,7 @@ def iter_images_with_url(
     use_star: None | bool = False, 
     escape: None | bool | Callable[[str], str] = True, 
     normalize_attr: Callable[[dict], dict] = normalize_attr, 
-    id_to_dirnode: None | dict[int, tuple[str, int] | DirNode] = None, 
+    id_to_dirnode: None | EllipsisType | dict[int, tuple[str, int] | DirNode] = None, 
     app: str = "web", 
     raise_for_changed_count: bool = False, 
     *, 
@@ -442,7 +443,7 @@ def iter_images_with_url(
     use_star: None | bool = False, 
     escape: None | bool | Callable[[str], str] = True, 
     normalize_attr: Callable[[dict], dict] = normalize_attr, 
-    id_to_dirnode: None | dict[int, tuple[str, int] | DirNode] = None, 
+    id_to_dirnode: None | EllipsisType | dict[int, tuple[str, int] | DirNode] = None, 
     app: str = "web", 
     raise_for_changed_count: bool = False, 
     *, 
@@ -460,7 +461,7 @@ def iter_images_with_url(
     use_star: None | bool = False, 
     escape: None | bool | Callable[[str], str] = True, 
     normalize_attr: Callable[[dict], dict] = normalize_attr, 
-    id_to_dirnode: None | dict[int, tuple[str, int] | DirNode] = None, 
+    id_to_dirnode: None | EllipsisType | dict[int, tuple[str, int] | DirNode] = None, 
     app: str = "web", 
     raise_for_changed_count: bool = False, 
     *, 
@@ -578,7 +579,7 @@ def iter_subtitles_with_url(
     use_star: None | bool = False, 
     escape: None | bool | Callable[[str], str] = True, 
     normalize_attr: Callable[[dict], dict] = normalize_attr, 
-    id_to_dirnode: None | dict[int, tuple[str, int] | DirNode] = None, 
+    id_to_dirnode: None | EllipsisType | dict[int, tuple[str, int] | DirNode] = None, 
     app: str = "web", 
     raise_for_changed_count: bool = False, 
     *, 
@@ -597,7 +598,7 @@ def iter_subtitles_with_url(
     use_star: None | bool = False, 
     escape: None | bool | Callable[[str], str] = True, 
     normalize_attr: Callable[[dict], dict] = normalize_attr, 
-    id_to_dirnode: None | dict[int, tuple[str, int] | DirNode] = None, 
+    id_to_dirnode: None | EllipsisType | dict[int, tuple[str, int] | DirNode] = None, 
     app: str = "web", 
     raise_for_changed_count: bool = False, 
     *, 
@@ -615,7 +616,7 @@ def iter_subtitles_with_url(
     use_star: None | bool = False, 
     escape: None | bool | Callable[[str], str] = True, 
     normalize_attr: Callable[[dict], dict] = normalize_attr, 
-    id_to_dirnode: None | dict[int, tuple[str, int] | DirNode] = None, 
+    id_to_dirnode: None | EllipsisType | dict[int, tuple[str, int] | DirNode] = None, 
     app: str = "web", 
     raise_for_changed_count: bool = False, 
     *, 
@@ -855,7 +856,7 @@ def make_strm(
     suffix: str = "", 
     type: Literal[1, 2, 3, 4, 5, 6, 7, 99] = 4, 
     max_workers: None | int = None, 
-    id_to_dirnode: None | dict[int, tuple[str, int] | DirNode] = None, 
+    id_to_dirnode: None | EllipsisType | dict[int, tuple[str, int] | DirNode] = None, 
     path_already: bool = False, 
     app: str = "android", 
     fs_files_cooldown: int | float = 0.5, 
@@ -881,7 +882,7 @@ def make_strm(
     suffix: str = "", 
     type: Literal[1, 2, 3, 4, 5, 6, 7, 99] = 4, 
     max_workers: None | int = None, 
-    id_to_dirnode: None | dict[int, tuple[str, int] | DirNode] = None, 
+    id_to_dirnode: None | EllipsisType | dict[int, tuple[str, int] | DirNode] = None, 
     path_already: bool = False, 
     app: str = "android", 
     fs_files_cooldown: int | float = 0.5, 
@@ -906,7 +907,7 @@ def make_strm(
     suffix: str = "", 
     type: Literal[1, 2, 3, 4, 5, 6, 7, 99] = 4, 
     max_workers: None | int = None, 
-    id_to_dirnode: None | dict[int, tuple[str, int] | DirNode] = None, 
+    id_to_dirnode: None | EllipsisType | dict[int, tuple[str, int] | DirNode] = None, 
     path_already: bool = False, 
     app: str = "android", 
     fs_files_cooldown: int | float = 0.5, 
@@ -1170,40 +1171,36 @@ def iter_download_nodes(
         method = client.download_files
     else:
         method = client.download_folders
+    get_nodes = partial(method, async_=async_, **{"base_url": get_base_url, **request_kwargs})
     if max_workers == 1:
         def gen_step(pickcode):
             if isinstance(pickcode, int):
                 resp = yield client.fs_file_skim(pickcode, async_=async_, **request_kwargs)
                 check_response(resp)
                 pickcode = resp["data"][0]["pick_code"]
-            request_kwargs.setdefault("base_url", get_base_url)
             for i in count(1):
                 payload = {"pickcode": pickcode, "page": i}
-                resp = yield method(payload, async_=async_, **request_kwargs)
+                resp = yield get_nodes(payload)
                 check_response(resp)
                 data = resp["data"]
                 yield YieldFrom(data["list"], identity=True)
                 if not data["has_next_page"]:
                     break
     else:
+        max_page = 0
         get_next_page = count(1).__next__
         if async_:
             q: Any = AsyncQueue()
         else:
             q = SimpleQueue()
         get, put = q.get, q.put_nowait
-        max_page = 0
         def request(pickcode):
             nonlocal max_page
             while True:
                 page = get_next_page()
                 if max_page and page > max_page:
                     return
-                resp: dict = yield method(
-                    {"pickcode": pickcode, "page": page}, 
-                    async_=async_, # type: ignore
-                    **request_kwargs, 
-                )
+                resp: dict = yield get_nodes({"pickcode": pickcode, "page": page})
                 try:
                     check_response(resp)
                 except BaseException as e:
@@ -1214,7 +1211,9 @@ def iter_download_nodes(
                 if not data["has_next_page"]:
                     max_page = page
         def gen_step(pickcode):
-            nonlocal max_workers
+            nonlocal max_workers, max_page, get_next_page
+            max_page = 0
+            get_next_page = count(1).__next__
             if async_:
                 if max_workers is None or max_workers <= 0:
                     max_workers = 20
@@ -1239,7 +1238,6 @@ def iter_download_nodes(
                 )
                 check_response(resp)
                 pickcode = resp["data"][0]["pick_code"]
-            request_kwargs.setdefault("base_url", cycle(("http://proapi.115.com", "https://proapi.115.com")).__next__)
             try:
                 sentinel = object()
                 countdown: Callable
@@ -1282,6 +1280,11 @@ def iter_download_nodes(
             )) as get_next:
                 while True:
                     attr = yield get_next
+                    if not files:
+                        yield Yield(
+                            {"fid": str(attr["id"]), "pid": "0", "fn": attr["name"]}, 
+                            identity=True, 
+                        )
                     yield YieldFrom(
                         run_gen_step_iter(gen_step(attr["pickcode"]), async_=async_), 
                         identity=True, 
@@ -1293,7 +1296,7 @@ def iter_download_nodes(
 def iter_download_files(
     client: str | P115Client, 
     cid: int = 0, 
-    id_to_dirnode: None | dict[int, tuple[str, int] | DirNode] = None, 
+    id_to_dirnode: None | EllipsisType | dict[int, tuple[str, int] | DirNode] = None, 
     escape: None | bool | Callable[[str], str] = True, 
     with_ancestors: bool = True, 
     max_workers: None | int = None, 
@@ -1307,7 +1310,7 @@ def iter_download_files(
 def iter_download_files(
     client: str | P115Client, 
     cid: int = 0, 
-    id_to_dirnode: None | dict[int, tuple[str, int] | DirNode] = None, 
+    id_to_dirnode: None | EllipsisType | dict[int, tuple[str, int] | DirNode] = None, 
     escape: None | bool | Callable[[str], str] = True, 
     with_ancestors: bool = True, 
     max_workers: None | int = None, 
@@ -1320,7 +1323,7 @@ def iter_download_files(
 def iter_download_files(
     client: str | P115Client, 
     cid: int = 0, 
-    id_to_dirnode: None | dict[int, tuple[str, int] | DirNode] = None, 
+    id_to_dirnode: None | EllipsisType | dict[int, tuple[str, int] | DirNode] = None, 
     escape: None | bool | Callable[[str], str] = True, 
     with_ancestors: bool = True, 
     max_workers: None | int = None, 

{p115client-0.0.5.8.4 → p115client-0.0.5.8.6}/p115client/tool/edit.py

@@ -103,6 +103,7 @@ def update_desc(
     desc: str = "", 
     batch_size: int = 10_000, 
     max_workers: None | int = None, 
+    app: str = "web", 
     *, 
     async_: Literal[False] = False, 
     **request_kwargs, 
@@ -116,6 +117,7 @@ def update_desc(
     desc: str = "", 
     batch_size: int = 10_000, 
     max_workers: None | int = None, 
+    app: str = "web", 
     *, 
     async_: Literal[True], 
     **request_kwargs, 
@@ -128,6 +130,7 @@ def update_desc(
     desc: str = "", 
     batch_size: int = 10_000, 
     max_workers: None | int = None, 
+    app: str = "web", 
     *, 
     async_: Literal[False, True] = False, 
     **request_kwargs, 
@@ -139,13 +142,19 @@ def update_desc(
     :param desc: 备注文本
     :param batch_size: 批次大小，分批次，每次提交的 id 数
     :param max_workers: 并发工作数，如果为 None 或者 <= 0，则自动确定
+    :param app: 使用此设备的接口
     :param async_: 是否异步
     :param request_kwargs: 其它请求参数
     """
+    if app in ("", "web", "desktop", "harmony"):
+        method = "fs_desc_set"
+    else:
+        method = "fs_desc_set_app"
+        request_kwargs["app"] = app
     return update_abstract(
         client, 
         ids, # type: ignore
-        method="fs_desc_set", 
+        method=method, 
         value=desc, 
         batch_size=batch_size, 
         max_workers=max_workers, 
@@ -162,6 +171,7 @@ def update_star(
     star: bool = True, 
     batch_size: int = 10_000, 
     max_workers: None | int = None, 
+    app: str = "web", 
     *, 
     async_: Literal[False] = False, 
     **request_kwargs, 
@@ -175,6 +185,7 @@ def update_star(
     star: bool = True, 
     batch_size: int = 10_000, 
     max_workers: None | int = None, 
+    app: str = "web", 
     *, 
     async_: Literal[True], 
     **request_kwargs, 
@@ -187,6 +198,7 @@ def update_star(
     star: bool = True, 
     batch_size: int = 10_000, 
     max_workers: None | int = None, 
+    app: str = "web", 
     *, 
     async_: Literal[False, True] = False, 
     **request_kwargs, 
@@ -201,13 +213,23 @@ def update_star(
     :param star: 是否设置星标
     :param batch_size: 批次大小，分批次，每次提交的 id 数
     :param max_workers: 并发工作数，如果为 None 或者 <= 0，则自动确定
+    :param app: 使用此设备的接口
     :param async_: 是否异步
     :param request_kwargs: 其它请求参数
     """
+    if isinstance(client, str):
+        client = P115Client(client, check_for_relogin=True)
+    if not isinstance(client, P115Client) or app == "open":
+        method = "fs_star_set_open"
+    elif app in ("", "web", "desktop", "harmony"):
+        method = "fs_star_set"
+    else:
+        method = "fs_star_set_app"
+        request_kwargs["app"] = app
     return update_abstract(
         client, 
         ids, # type: ignore
-        method="fs_star_set", 
+        method=method, 
         value=star, 
         batch_size=batch_size, 
         max_workers=max_workers, 
@@ -224,6 +246,7 @@ def update_label(
     label: int | str = 1, 
     batch_size: int = 10_000, 
     max_workers: None | int = None, 
+    app: str = "web", 
     *, 
     async_: Literal[False] = False, 
     **request_kwargs, 
@@ -237,6 +260,7 @@ def update_label(
     label: int | str = 1, 
     batch_size: int = 10_000, 
     max_workers: None | int = None, 
+    app: str = "web", 
     *, 
     async_: Literal[True], 
     **request_kwargs, 
@@ -249,6 +273,7 @@ def update_label(
     label: int | str = 1, 
     batch_size: int = 10_000, 
     max_workers: None | int = None, 
+    app: str = "web", 
     *, 
     async_: Literal[False, True] = False, 
     **request_kwargs, 
@@ -260,13 +285,19 @@ def update_label(
     :param label: 标签 id，多个用逗号 "," 隔开，如果用一个根本不存在的 id，效果就是清空标签列表
     :param batch_size: 批次大小，分批次，每次提交的 id 数
     :param max_workers: 并发工作数，如果为 None 或者 <= 0，则自动确定
+    :param app: 使用此设备的接口
     :param async_: 是否异步
     :param request_kwargs: 其它请求参数
     """
+    if app in ("", "web", "desktop", "harmony"):
+        method = "fs_label_set"
+    else:
+        method = "fs_label_set_app"
+        request_kwargs["app"] = app
     return update_abstract(
         client, 
         ids, # type: ignore
-        method="fs_label_set", 
+        method=method, 
         value=label, 
         batch_size=batch_size, 
         max_workers=max_workers, 
@@ -401,6 +432,7 @@ def update_show_play_long(
     show: bool = True, 
     batch_size: int = 10_000, 
     max_workers: None | int = None, 
+    app: str = "web", 
     *, 
     async_: Literal[False] = False, 
     **request_kwargs, 
@@ -414,6 +446,7 @@ def update_show_play_long(
     show: bool = True, 
     batch_size: int = 10_000, 
     max_workers: None | int = None, 
+    app: str = "web", 
     *, 
     async_: Literal[True], 
     **request_kwargs, 
@@ -426,6 +459,7 @@ def update_show_play_long(
     show: bool = True, 
     batch_size: int = 10_000, 
     max_workers: None | int = None, 
+    app: str = "web", 
     *, 
     async_: Literal[False, True] = False, 
     **request_kwargs, 
@@ -437,13 +471,19 @@ def update_show_play_long(
     :param show: 是否显示时长
     :param batch_size: 批次大小，分批次，每次提交的 id 数
     :param max_workers: 并发工作数，如果为 None 或者 <= 0，则自动确定
+    :param app: 使用此设备的接口
     :param async_: 是否异步
     :param request_kwargs: 其它请求参数
     """
+    if app in ("", "web", "desktop", "harmony"):
+        method = "fs_show_play_long_set"
+    else:
+        method = "fs_show_play_long_set_app"
+        request_kwargs["app"] = app
     return update_abstract(
         client, 
         ids, # type: ignore
-        method="fs_show_play_long_set", 
+        method=method, 
         value=show, 
         batch_size=batch_size, 
         max_workers=max_workers, 
@@ -518,6 +558,7 @@ def batch_unstar(
     batch_size: int = 10_000, 
     ensure_file: None | bool = None, 
     max_workers: None | int = None, 
+    app: str = "web", 
     *, 
     async_: Literal[False] = False, 
     **request_kwargs, 
@@ -530,6 +571,7 @@ def batch_unstar(
     batch_size: int = 10_000, 
     ensure_file: None | bool = None, 
     max_workers: None | int = None, 
+    app: str = "web", 
     *, 
     async_: Literal[True], 
     **request_kwargs, 
@@ -541,6 +583,7 @@ def batch_unstar(
     batch_size: int = 10_000, 
     ensure_file: None | bool = None, 
     max_workers: None | int = None, 
+    app: str = "web", 
     *, 
     async_: Literal[False, True] = False, 
     **request_kwargs, 
@@ -556,6 +599,7 @@ def batch_unstar(
         - None: 可以是目录或文件
 
     :param max_workers: 并发工作数，如果为 None 或者 <= 0，则自动确定
+    :param app: 使用此设备的接口
     :param async_: 是否异步
     :param request_kwargs: 其它请求参数
     """
@@ -572,7 +616,7 @@ def batch_unstar(
             client, 
             payload={"cid": 0, "count_folders": 1, "cur": 0, "fc_mix": 0, "offset": 0, "show_dir": 1, "star": 1}, 
             ensure_file=ensure_file, 
-            app="android", 
+            app=app, 
             cooldown=0.5, 
             async_=async_, 
             **request_kwargs, 
@@ -587,6 +631,7 @@ def batch_unstar(
             star=False, 
             batch_size=batch_size, 
             max_workers=max_workers, 
+            app=app, 
             async_=async_, # type: ignore
             **request_kwargs, 
         )

{p115client-0.0.5.8.4 → p115client-0.0.5.8.6}/p115client/tool/iterdir.py

@@ -8,9 +8,9 @@ __all__ = [
     "get_ancestors_to_cid", "get_id_to_path", "get_id_to_sha1", "get_id_to_pickcode", 
     "iter_nodes_skim", "iter_stared_dirs_raw", "iter_stared_dirs", "ensure_attr_path", 
     "ensure_attr_path_by_category_get", "iterdir_raw", "iterdir", "iterdir_limited", 
-    "iter_files_raw", "iter_files", "traverse_files", "iter_dupfiles", "iter_image_files", 
-    "share_extract_payload", "share_iterdir", "share_iter_files", "iter_selected_nodes", 
-    "iter_selected_nodes_by_pickcode", "iter_selected_nodes_using_category_get", 
+    "iter_files_raw", "iter_files", "traverse_files", "iter_dirs", "iter_dupfiles", 
+    "iter_image_files", "share_extract_payload", "share_iterdir", "share_iter_files", 
+    "iter_selected_nodes", "iter_selected_nodes_by_pickcode", "iter_selected_nodes_using_category_get", 
     "iter_selected_nodes_using_edit", "iter_selected_nodes_using_star_event", 
     "iter_selected_dirs_using_star", "iter_files_with_dirname", "iter_files_with_path", 
     "iter_files_with_path_by_export_dir", "iter_parents_3_level", "iter_dir_nodes", 
@@ -41,9 +41,9 @@ from weakref import WeakValueDictionary
 from asynctools import async_chain, async_filter, async_map, to_list
 from concurrenttools import run_as_thread, taskgroup_map, threadpool_map
 from iterutils import (
-    bfs_gen, chunked, async_foreach, ensure_aiter, foreach, flatten, 
-    iter_unique, run_gen_step, run_gen_step_iter, through, async_through, 
-    with_iter_next, Yield, YieldFrom, 
+    as_gen_step, bfs_gen, chunked, async_foreach, ensure_aiter, foreach, 
+    flatten, iter_unique, run_gen_step, run_gen_step_iter, through, 
+    async_through, with_iter_next, Yield, YieldFrom, 
 )
 from iter_collect import iter_keyed_dups, SupportsLT
 from orjson import loads
@@ -2789,6 +2789,101 @@ def traverse_files(
     return run_gen_step_iter(gen_step, async_=async_)
 
 
+@overload
+def iter_dirs(
+    client: str | P115Client, 
+    cid: int | str = 0, 
+    id_to_dirnode: None | EllipsisType | dict[int, tuple[str, int] | DirNode] = None, 
+    with_pickcode: bool = False, 
+    max_workers: None | int = None, 
+    *, 
+    async_: Literal[False] = False, 
+    **request_kwargs, 
+) -> Iterator[dict]:
+    ...
+@overload
+def iter_dirs(
+    client: str | P115Client, 
+    cid: int | str = 0, 
+    id_to_dirnode: None | EllipsisType | dict[int, tuple[str, int] | DirNode] = None, 
+    with_pickcode: bool = False, 
+    max_workers: None | int = None, 
+    *, 
+    async_: Literal[True], 
+    **request_kwargs, 
+) -> AsyncIterator[dict]:
+    ...
+def iter_dirs(
+    client: str | P115Client, 
+    cid: int | str = 0, 
+    id_to_dirnode: None | EllipsisType | dict[int, tuple[str, int] | DirNode] = None, 
+    with_pickcode: bool = False, 
+    max_workers: None | int = None, 
+    *, 
+    async_: Literal[False, True] = False, 
+    **request_kwargs, 
+) -> Iterator[dict] | AsyncIterator[dict]:
+    """遍历目录树，获取目录信息
+
+    :param client: 115 客户端或 cookies
+    :param cid: 目录 id（如果是 int） 或者 pickcode（如果是 str）
+    :param id_to_dirnode: 字典，保存 id 到对应文件的 `DirNode(name, parent_id)` 命名元组的字典
+    :param with_pickcode: 是否需要包含提取码
+    :param max_workers: 最大并发数
+    :param async_: 是否异步
+    :param request_kwargs: 其它请求参数
+
+    :return: 迭代器，返回此目录内的（仅目录）文件信息
+    """
+    from .download import iter_download_nodes
+    if isinstance(client, str):
+        client = P115Client(client, check_for_relogin=True)
+    if id_to_dirnode is None:
+        id_to_dirnode = ID_TO_DIRNODE_CACHE[client.user_id]
+    it = iter_download_nodes(
+        client, 
+        cid, 
+        files=False, 
+        max_workers=max_workers, 
+        async_=async_, # type: ignore
+        **request_kwargs, 
+    )
+    do_map: Callable = async_map if async_ else map
+    def project(info: dict, /) -> dict:
+        attr = {"id": int(info["fid"]), "parent_id": int(info["pid"]), "name": info["fn"]}
+        if id_to_dirnode is not ...:
+            id_to_dirnode[attr["id"]] = DirNode(attr["name"], attr["parent_id"])
+        return attr
+    it = do_map(project, it)
+    if with_pickcode:
+        file_skim = client.fs_file_skim
+        @as_gen_step(async_=async_)
+        def batch_load_pickcode(batch: Sequence[dict], /):
+            resp = yield file_skim(
+                (a["id"] for a in batch), 
+                method="POST", 
+                async_=async_, 
+                **request_kwargs, 
+            )
+            check_response(resp)
+            maps = {int(a["file_id"]): a["pick_code"] for a in resp["data"]}
+            for attr in batch:
+                attr["pickcode"] = maps[attr["id"]]
+            return batch
+        def gen_step(iterable):
+            batch_map = taskgroup_map if async_ else threadpool_map
+            with with_iter_next(batch_map(
+                batch_load_pickcode, 
+                chunked(iterable, 3000), 
+                max_workers=max_workers, 
+            )) as get_next:
+                while True:
+                    batch = yield get_next
+                    yield YieldFrom(batch, identity=True)
+        it = run_gen_step_iter(gen_step(it), async_=async_)
+    return it
+
+
 @overload
 def iter_dupfiles[K](
     client: str | P115Client, 

{p115client-0.0.5.8.4 → p115client-0.0.5.8.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "p115client"
-version = "0.0.5.8.4"
+version = "0.0.5.8.6"
 description = "Python 115 webdisk client."
 authors = ["ChenyangGao <wosiwujm@gmail.com>"]
 license = "MIT"