tytcli 0.2.0__py3-none-any.whl

tytcli/commands/wechat.py

@@ -0,0 +1,253 @@
+"""``tytcli wechat`` —— 公众号文章（WechatPublic 模块）。
+
+住在第二个 API 根（``MIDDLE_BASE_URLS``）上，不是核心根；实测得到的契约、以及内部
+交接文档与实测不符的那些地方，都记在 ``client.py`` 的模块 docstring 里。
+"""
+
+from __future__ import annotations
+
+from collections.abc import Sequence
+from typing import Any
+
+import click
+
+from ..client import (
+    WECHAT_ARTICLE_ID_MAX,
+    WECHAT_PAGE_SIZE_MAX,
+    WECHAT_PAGE_SIZE_MIN,
+    WECHAT_POST_VIEWS,
+)
+from ..errors import UsageError
+from ..output import HINT_KEY, emit_ok
+from .shared import (
+    middle_client,
+    parse_fields_option,
+    raise_for_envelope,
+    require_config,
+    trim_page,
+    unwrap_data,
+)
+
+#: ``wechat articles`` 默认（裁剪后）输出保留的字段。
+#:
+#: 2026-11-06 在 test 上实测：20 条记录的一页是 66.8 KiB，其中**光 ``cleanBody``
+#: 一个字段就占 74%**（去掉后只剩 17.5 KiB）。``cleanBody`` 是文章正文——那是
+#: *详情*命令的职责，而且交接文档根本没把它列为列表字段。所以默认输出是索引，
+#: 不是内容。
+#:
+#: **绝不静默丢字段**：文档里带着 ``trimmed``、``record_fields`` 与
+#: ``__tytcli_hint__``；``--fields`` 用于显式自选，``--full`` 返回服务端发来的
+#: 每一个字段。
+WECHAT_DEFAULT_FIELDS: tuple[str, ...] = (
+    "title",
+    "nickname",
+    "summary",
+    "articleId",
+    "publishTime",
+)
+
+
+
+
+
+
+def _wechat_hint(default_fields: Sequence[str], observed: list[str]) -> str:
+    """裁剪过的文章页所对应的 ``__tytcli_hint__`` 文案。
+
+    措辞是模块专属的，所以留在这里而不是放进 ``shared``；它周围的机制（观测到了
+    哪些字段、什么时候该输出提示）是共享的。
+    """
+    extra = [field for field in observed if field not in default_fields]
+    parts = [f"默认每条文章仅输出：{', '.join(default_fields)}。"]
+    if extra:
+        parts.append(f"该接口本次还返回了：{', '.join(extra)}。")
+    parts.append(
+        "用 --fields a,b,c 自选字段，或 --full 取服务端原样全量"
+        "（含 cleanBody 正文，20 条约 67 KiB）。"
+    )
+    parts.append("articleId 可直接用于：tytcli wechat article --article-id <ID>。")
+    # 用 "" 而不是 " " 拼接：每一段本来就以全角句号结尾，句号前多一个空格在
+    # 中文里看起来像错字。
+    return "".join(parts)
+
+
+@click.group()
+def wechat() -> None:
+    """公众号文章。
+
+    \b
+    注意：该模块可能未在所有环境部署。若命令返回 HTTP 404，
+    错误的 details.hint 会说明原因——这不是路由拼错。
+    """
+
+
+@wechat.command("articles")
+@click.option(
+    "--view",
+    type=click.Choice(list(WECHAT_POST_VIEWS)),
+    default="ALL",
+    show_default=True,
+    help="ALL 全部文章 / SUBSCRIBED 我的订阅 / FAVORITE 我的收藏。",
+)
+@click.option(
+    "--fake-id",
+    default=None,
+    metavar="ID",
+    help="只看某个公众号（列表里的 fakeId）；省略表示不限。",
+)
+@click.option("--page", "page_no", type=int, default=1, show_default=True, help="页码，从 1 开始。")
+@click.option(
+    "--page-size",
+    type=int,
+    default=20,
+    show_default=True,
+    help=f"每页条数，{WECHAT_PAGE_SIZE_MIN}–{WECHAT_PAGE_SIZE_MAX}。",
+)
+@click.option(
+    "--fields",
+    "fields_option",
+    default=None,
+    metavar="A,B,C",
+    help="只输出这些文章字段（逗号分隔）。与 --full 互斥。",
+)
+@click.option(
+    "--full",
+    "full",
+    is_flag=True,
+    help="输出服务端返回的全部字段，不裁剪。与 --fields 互斥。",
+)
+@click.pass_context
+def wechat_articles(
+    ctx: click.Context,
+    view: str,
+    fake_id: str | None,
+    page_no: int,
+    page_size: int,
+    fields_option: str | None,
+    full: bool,
+) -> None:
+    """分页查询公众号文章。
+
+    \b
+    默认输出是精简索引（title/nickname/summary/articleId/publishTime），
+    并在 __tytcli_hint__ 里说明还能取到哪些字段。
+    20 条时：默认约 5 KiB，--full 约 67 KiB。
+
+    \b
+    拿到 articleId 后查详情：
+      tytcli wechat article --article-id <ID>
+    """
+    pretty: bool = ctx.obj["pretty"]
+
+    if full and fields_option is not None:
+        raise UsageError(
+            "--full 与 --fields 互斥：前者要全量，后者要自选。",
+            code="USAGE_FIELDS_CONFLICT",
+        )
+    if page_no < 1:
+        raise UsageError(
+            f"--page 必须 >= 1（当前 {page_no}）。",
+            code="USAGE_PAGE_OUT_OF_RANGE",
+            details={"page": page_no, "min": 1},
+        )
+    if not WECHAT_PAGE_SIZE_MIN <= page_size <= WECHAT_PAGE_SIZE_MAX:
+        raise UsageError(
+            f"--page-size 必须在 {WECHAT_PAGE_SIZE_MIN}–{WECHAT_PAGE_SIZE_MAX} 之间"
+            f"（当前 {page_size}）。",
+            code="USAGE_PAGE_SIZE_OUT_OF_RANGE",
+            details={
+                "page_size": page_size,
+                "min": WECHAT_PAGE_SIZE_MIN,
+                "max": WECHAT_PAGE_SIZE_MAX,
+            },
+        )
+    selected = parse_fields_option(fields_option) if fields_option is not None else None
+
+    config = require_config()
+    client = middle_client(config)
+    envelope = client.wechat_articles(
+        view=view,
+        fake_id=fake_id,
+        page_no=page_no,
+        page_size=page_size,
+    )
+    raise_for_envelope(
+        envelope, env_name=config.active_env, base_url=config.middle_base_url
+    )
+
+    inner_raw = unwrap_data(envelope.payload)
+
+    meta: dict[str, Any] = {
+        "env": config.active_env,
+        "base_url": config.middle_base_url,
+        "api_root": "middleApi",
+        "http_status": envelope.http_status,
+        "view": view,
+        "page_no": page_no,
+        "page_size": page_size,
+    }
+    if fake_id is not None:
+        meta["fake_id"] = fake_id
+
+    page = trim_page(
+        inner_raw=inner_raw,
+        list_key="records",
+        default_fields=WECHAT_DEFAULT_FIELDS,
+        selected=selected,
+        full=full,
+        make_hint=_wechat_hint,
+    )
+    document_extra: dict[str, Any] = {}
+    if page.hint is not None:
+        document_extra[HINT_KEY] = page.hint
+    emit_ok(page.payload, pretty=pretty, **meta, **page.meta, **document_extra)
+
+
+@wechat.command("article")
+@click.option(
+    "--article-id",
+    required=True,
+    metavar="HEX64",
+    help=f"文章的 articleId（列表返回的 {WECHAT_ARTICLE_ID_MAX} 位十六进制）。",
+)
+@click.pass_context
+def wechat_article(ctx: click.Context, article_id: str) -> None:
+    """查询单篇公众号文章详情。
+
+    \b
+    输出为服务端原样全量（单篇约 14 KiB），含正文与 AI 分析结果
+    （观点、段落、话题、关键词）。
+
+    \b
+    articleId 来自 `tytcli wechat articles` 的输出。
+    """
+    pretty: bool = ctx.obj["pretty"]
+
+    # argv 里的空白是 shell 造成的，去掉是安全的。而 identity 标题**内部**的
+    # 空白不是——这正是 identity 不适合当定位符的原因之一。
+    article_id = article_id.strip()
+    if not article_id:
+        raise UsageError("--article-id 不能为空。", code="USAGE_ARTICLE_ID_EMPTY")
+    if len(article_id) > WECHAT_ARTICLE_ID_MAX:
+        raise UsageError(
+            f"--article-id 超过 {WECHAT_ARTICLE_ID_MAX} 字符（当前 {len(article_id)}）。",
+            code="USAGE_ARTICLE_ID_TOO_LONG",
+            details={"length": len(article_id), "max": WECHAT_ARTICLE_ID_MAX},
+        )
+
+    config = require_config()
+    client = middle_client(config)
+    envelope = client.wechat_article(article_id=article_id)
+    raise_for_envelope(
+        envelope, env_name=config.active_env, base_url=config.middle_base_url
+    )
+
+    emit_ok(
+        unwrap_data(envelope.payload),
+        pretty=pretty,
+        env=config.active_env,
+        base_url=config.middle_base_url,
+        api_root="middleApi",
+        http_status=envelope.http_status,
+        trimmed=False,
+    )

tytcli/commands/whoami.py

@@ -0,0 +1,76 @@
+"""``tytcli whoami`` —— 当前登录身份。"""
+
+from __future__ import annotations
+
+from typing import Any, cast
+
+import click
+
+from ..client import TytClient
+from ..output import emit_ok
+from .shared import raise_for_envelope, require_config
+
+#: `whoami` 额外发布的身份字段，这样 agent 不必去翻完整 payload——光 `Menus`
+#: 权限树就有好几 KB。
+#:
+#: **刻意排除 `Phone`**：它是 PII，而这份摘要正是最容易被记进对话或日志的部分。
+#: 它仍然存在于 `data` 里，`data` 始终是服务端响应的原样透传。
+IDENTITY_FIELDS = (
+    "UserId",
+    "UserName",
+    "JjtId",
+    "CompanyId",
+    "CompanyName",
+    "Department",
+    "Duty",
+    "EmpNo",
+    "AccountType",
+    "EstPerm",
+    "EmotionPerm",
+)
+
+
+@click.command()
+@click.pass_context
+def whoami(ctx: click.Context) -> None:
+    """查询当前 Token 对应的登录身份。
+
+    \b
+    输出中 `identity` 是关键字段的摘要（不含手机号）；
+    `data` 是服务端原样透传的完整信息（含权限菜单树）。
+    """
+    pretty: bool = ctx.obj["pretty"]
+    config = require_config()
+    token = config.require_token()
+
+    client = TytClient(config.base_url, token, timeout=config.timeout_seconds)
+    envelope = client.whoami()
+
+    raise_for_envelope(envelope, env_name=config.active_env, base_url=config.base_url)
+
+    emit_ok(
+        envelope.payload,
+        pretty=pretty,
+        env=config.active_env,
+        base_url=config.base_url,
+        http_status=envelope.http_status,
+        identity=extract_identity(envelope.payload),
+    )
+
+
+def extract_identity(payload: dict[str, Any]) -> dict[str, Any] | None:
+    """从 GetLoginInfoV2 的 payload 里取出标量身份字段。
+
+    形状与预期不符时返回 ``None``，这样服务端的改动静默降级成"没有摘要"，而不是
+    让命令崩掉。`data` 里依然带着全部字段。
+    """
+    raw_data = payload.get("Data")
+    if not isinstance(raw_data, dict):
+        return None
+    data = cast("dict[str, Any]", raw_data)
+    identity: dict[str, Any] = {key: data[key] for key in IDENTITY_FIELDS if key in data}
+    menus = data.get("Menus")
+    # isinstance() 会把 Any 窄化成 list[Unknown]；cast 成具体的 list，
+    # 这样 len() 的参数类型才是已知的。
+    identity["MenuCount"] = len(cast("list[Any]", menus)) if isinstance(menus, list) else None
+    return identity