p115client 0.0.5.11.10.2__tar.gz → 0.0.5.11.11__tar.gz

{p115client-0.0.5.11.10.2 → p115client-0.0.5.11.11}/PKG-INFO

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

{p115client-0.0.5.11.10.2 → p115client-0.0.5.11.11}/p115client/client.py

@@ -377,6 +377,9 @@ def check_response(resp: dict | Awaitable[dict], /) -> dict | Coroutine[Any, Any
                 # {"state": false, "errno": 911, "error": "请验证账号"}
                 case 911:
                     raise AuthenticationError(errno.EAUTH, resp)
+                # {"state": false, "errno": 10004, "error": "错误的链接"}
+                case 10004:
+                    raise OperationalError(errno.EINVAL, resp)
                 # {"state": false, "errno": 20001, "error": "目录名称不能为空"}
                 case 20001:
                     raise OperationalError(errno.EINVAL, resp)
@@ -3180,8 +3183,8 @@ class P115OpenClient(ClientRequestMixin):
             - `c`: 允许同时打开次数，如果为 0，则是无限次数
             - `f`: 请求时要求携带请求头
                 - 如果为空，则无要求
-                - 如果为 1，则需要 User-Agent（和请求直链时的一致）
-                - 如果为 3，则需要 User-Agent（和请求直链时的一致） 和 Cookie（由请求直链时的响应所返回的 Set-Cookie 响应头）
+                - 如果为 1，则需要 user-agent（和请求直链时的一致）
+                - 如果为 3，则需要 user-agent（和请求直链时的一致） 和 Cookie（由请求直链时的响应所返回的 Set-Cookie 响应头）
 
         :param pickcode: 提取码
         :param strict: 如果为 True，当目标是目录时，会抛出 IsADirectoryError 异常
@@ -7857,8 +7860,8 @@ class P115Client(P115OpenClient):
             - `c`: 允许同时打开次数，如果为 0，则是无限次数
             - `f`: 请求时要求携带请求头
                 - 如果为空，则无要求
-                - 如果为 1，则需要 User-Agent（和请求直链时的一致）
-                - 如果为 3，则需要 User-Agent（和请求直链时的一致） 和 Cookie（由请求直链时的响应所返回的 Set-Cookie 响应头）
+                - 如果为 1，则需要 user-agent（和请求直链时的一致）
+                - 如果为 3，则需要 user-agent（和请求直链时的一致） 和 Cookie（由请求直链时的响应所返回的 Set-Cookie 响应头）
 
         :param pickcode: 提取码
         :param strict: 如果为 True，当目标是目录时，会抛出 IsADirectoryError 异常
@@ -18043,6 +18046,7 @@ class P115Client(P115OpenClient):
         POST https://lixian.115.com/lixianssp/?ac=add_task_urls
 
         :payload:
+            - url: str    💡 链接，支持HTTP、HTTPS、FTP、磁力链和电驴链接
             - url[0]: str 💡 链接，支持HTTP、HTTPS、FTP、磁力链和电驴链接
             - url[1]: str
             - ...
@@ -18052,7 +18056,7 @@ class P115Client(P115OpenClient):
         if isinstance(payload, str):
             payload = payload.strip("\n").split("\n")
         if not isinstance(payload, dict):
-            payload = {f"url[{i}]": url for i, url in enumerate(payload)}
+            payload = {f"url[{i}]": url for i, url in enumerate(payload) if url}
             if not payload:
                 raise ValueError("no `url` specified")
         return self._offline_post(payload, "add_task_urls", use_web_api=use_web_api, async_=async_, **request_kwargs)
@@ -21021,7 +21025,7 @@ class P115Client(P115OpenClient):
         request_kwargs.update(make_upload_payload(data))
         request_kwargs["headers"] = {
             **(request_kwargs.get("headers") or {}), 
-            "Content-Type": "application/x-www-form-urlencoded", 
+            "content-type": "application/x-www-form-urlencoded", 
             "user-agent": "Mozilla/5.0 115disk/99.99.99.99 115Browser/99.99.99.99 115wangpan_android/99.99.99.99", 
         }
         request_kwargs.setdefault("parse", parse_upload_init_response)

{p115client-0.0.5.11.10.2 → p115client-0.0.5.11.11}/p115client/tool/__init__.py

@@ -12,6 +12,7 @@ from .fs_files import *
 from .history import *
 from .iterdir import *
 from .life import *
+from .offline import *
 from .pool import *
 from .request import *
 from .upload import *

p115client-0.0.5.11.11/p115client/tool/offline.py

@@ -0,0 +1,164 @@
+#!/usr/bin/env python3
+# encoding: utf-8
+
+__author__ = "ChenyangGao <https://chenyanggao.github.io>"
+__all__ = ["offline_iter", "offline_restart_iter"]
+__doc__ = "这个模块提供了一些和离线下载有关的函数"
+
+from asyncio import sleep as async_sleep
+from collections.abc import AsyncIterator, Callable, Iterable, Iterator
+from itertools import count
+from time import sleep, time
+from typing import overload, Literal
+
+from iterutils import run_gen_step_iter, with_iter_next, Yield, YieldFrom
+from p115client import check_response, P115Client, P115OpenClient
+
+
+@overload
+def offline_iter(
+    client: str | P115Client | P115OpenClient, 
+    /, 
+    page_start: int = 1, 
+    page_stop: int = -1, 
+    cooldown: float = 0, 
+    use_open_api: bool = False, 
+    *, 
+    async_: Literal[False] = False, 
+    **request_kwargs, 
+) -> Iterator[dict]:
+    ...
+@overload
+def offline_iter(
+    client: str | P115Client | P115OpenClient, 
+    /, 
+    page_start: int = 1, 
+    page_stop: int = -1, 
+    cooldown: float = 0, 
+    use_open_api: bool = False, 
+    *, 
+    async_: Literal[True], 
+    **request_kwargs, 
+) -> AsyncIterator[dict]:
+    ...
+def offline_iter(
+    client: str | P115Client | P115OpenClient, 
+    /, 
+    page_start: int = 1, 
+    page_stop: int = -1, 
+    cooldown: float = 0, 
+    use_open_api: bool = False, 
+    *, 
+    async_: Literal[False, True] = False, 
+    **request_kwargs, 
+) -> Iterator[dict] | AsyncIterator[dict]:
+    """遍历任务列表，获取任务信息
+
+    :param client: 115 客户端或 cookies
+    :param page_start: 开始页数
+    :param page_stop: 结束页数（不含），如果 <= 0，则不限
+    :param cooldown: 接口调用冷却时间，单位：秒
+    :param use_open_api: 是否使用 open api
+    :param async_: 是否异步
+    :param request_kwargs: 其它请求参数
+
+    :return: 迭代器，返回任务信息
+    """
+    if isinstance(client, str):
+        client = P115Client(client, check_for_relogin=True)
+    if page_start < 1:
+        page_start = 1
+    if page_stop > 0:
+        pages: Iterable[int] = range(page_start, page_stop)
+    else:
+        pages = count(page_start)
+    if not isinstance(client, P115Client):
+        use_open_api = True
+    def gen_step():
+        if use_open_api:
+            offline_list = client.offline_list_open
+        else:
+            offline_list = client.offline_list
+        if cooldown > 0:
+            do_sleep = async_sleep if async_ else sleep
+        last_t: float = 0
+        for page in pages:
+            if last_t and (diff := last_t + cooldown - time()) > 0:
+                yield do_sleep(diff)
+            last_t = time()
+            resp = yield offline_list(page, async_=async_, **request_kwargs)
+            check_response(resp)
+            if use_open_api:
+                resp = resp["data"]
+            tasks = resp["tasks"]
+            if not tasks:
+                break
+            yield YieldFrom(resp["tasks"])
+            if len(tasks) < 30 or page >= resp["page_count"]:
+                break
+    return run_gen_step_iter(gen_step, async_=async_)
+
+
+@overload
+def offline_restart_iter(
+    client: str | P115Client, 
+    /, 
+    predicate: None | Callable[[dict], bool] = None, 
+    *, 
+    async_: Literal[False] = False, 
+    **request_kwargs, 
+) -> Iterator[dict]:
+    ...
+@overload
+def offline_restart_iter(
+    client: str | P115Client, 
+    /, 
+    predicate: None | Callable[[dict], bool] = None, 
+    *, 
+    async_: Literal[True], 
+    **request_kwargs, 
+) -> AsyncIterator[dict]:
+    ...
+def offline_restart_iter(
+    client: str | P115Client, 
+    /, 
+    predicate: None | Callable[[dict], bool] = None, 
+    *, 
+    async_: Literal[False, True] = False, 
+    **request_kwargs, 
+) -> Iterator[dict] | AsyncIterator[dict]:
+    """重试任务：重试那些因为空间不足而转存失败的任务
+
+    :param client: 115 客户端或 cookies
+    :param predicate: 断言，用于筛选
+    :param async_: 是否异步
+    :param request_kwargs: 其它请求参数
+
+    :return: 迭代器，逐个任务返回执行重试后的响应
+    """
+    if isinstance(client, str):
+        client = P115Client(client, check_for_relogin=True)
+    def gen_step():
+        left_no_space: list[dict] = []
+        add_task = left_no_space.append
+        with with_iter_next(offline_iter(
+            client, 
+            async_=async_, 
+            **request_kwargs, 
+        )) as do_next:
+            while True:
+                task = yield do_next()
+                if task["move"] == -1:
+                    add_task(task)
+                elif task["status"] == 2:
+                    break
+        for task in filter(predicate, left_no_space):
+            resp = yield client.offline_restart(
+                task["info_hash"], 
+                async_=async_, 
+                **request_kwargs, 
+            )
+            resp["task"] = task
+            yield Yield(resp)
+    return run_gen_step_iter(gen_step, async_=async_)
+

{p115client-0.0.5.11.10.2 → p115client-0.0.5.11.11}/pyproject.toml

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