Files
dingtalk-workspace-cli/skills/mono/scripts/extract_media_id.py
539d10f80f feat: 合并 pre-mcp-discovery 分支能力到 main(服务发现保持生产) (#391)
将 test/pre-mcp-discovery 分支上多位同学的能力改动整体合入 main,
服务发现端点保持生产 https://mcp.dingtalk.com,未携带分支上的预发改动。

主要内容:
- 修雨域(chat / report / todo / contact / mail)对齐 wukong 基线 (#355)
- DWS 内容生产能力增强 (#387)、文档导出与 OA 对齐 (#362 #388 #390)
- envelope 注册 cli.Aliases 为 cobra aliases、命令结构与 JSON 解析优化
- skills 多包同步 wukong 对齐内容(attendance/calendar/minutes/oa/sheet 等)
- product/oa 进一步对齐 wukong

排除项:未携带 566b4e1(test: point service discovery to pre-mcp.dingtalk.com);
以下 4 个常量保持生产:skill_command.go / auth/endpoints.go / cli/loader.go / market/registry.go。

Co-authored-by: john <32427341+wqyenjoy@users.noreply.github.com>
Co-authored-by: wxianfeng <wang.fl1429@gmail.com>
2026-06-02 21:04:12 +08:00

88 lines
2.6 KiB
Python
Raw Permalink Blame History

This file contains ambiguous Unicode characters
This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.
#!/usr/bin/env python3
"""
从 dt_media_upload 返回的 URL 中提取 mediaId。
用法:
python extract_media_id.py <url>
python extract_media_id.py "https://down.dingtalk.com/media/lQLPD4JNnliqBq3NBQDNA8Cw_960_1280.png"
# 输出: @lQLPD4JNnliqBq3NBQDNA8Cw
依赖: 无(纯 Python 标准库)
典型工作流(发送图片/语音/视频消息):
# 1. 用 dt_media_upload 上传文件,获得 URL
# 2. 用本脚本从 URL 提取 mediaId
python extract_media_id.py "<dt_media_upload 返回的 URL>"
# 输出: @lQLPxxx(直接用于 --media-id 或 --pic-url 参数)
"""
import argparse
import re
import sys
from urllib.parse import urlparse
def extract_media_id(url: str) -> str:
"""从 dt_media_upload 返回的 URL 中提取 mediaId。
支持的 URL 格式示例:
https://down.dingtalk.com/media/lQLPD4JNnliqBq3NBQDNA8Cw_960_1280.png
https://down.dingtalk.com/media/lQLPD4JNnliqBq3NBQDNA8Cw.png
https://down.dingtalk.com/media/lQLPD4JNnliqBq3NBQDNA8Cw
提取规则:
1. 取 URL 路径中 /media/ 之后的部分
2. 去除尾部的 _数字_数字.扩展名 后缀(如 _960_1280.png)
3. 如果没有尺寸后缀,去除尾部的 .扩展名
4. 加上 @ 前缀
返回: @lQLPD4JNnliqBq3NBQDNA8Cw
"""
url = url.strip()
# 提取路径部分
parsed = urlparse(url)
path = parsed.path # e.g. /media/lQLPxxx_960_1280.png
# 取 /media/ 之后的部分
media_prefix = "/media/"
idx = path.find(media_prefix)
if idx == -1:
print(f"错误:URL 中未找到 /media/ 路径: {url}", file=sys.stderr)
sys.exit(1)
raw = path[idx + len(media_prefix):] # e.g. lQLPxxx_960_1280.png
if not raw:
print(f"错误:/media/ 后无内容: {url}", file=sys.stderr)
sys.exit(1)
# 尝试去除 _数字_数字.扩展名 后缀(如 _960_1280.png)
match = re.match(r'^(.+?)(_\d+_\d+\.\w+)$', raw)
if match:
media_id = match.group(1)
else:
# 没有尺寸后缀,尝试去除 .扩展名
match2 = re.match(r'^(.+)\.\w+$', raw)
if match2:
media_id = match2.group(1)
else:
# 无后缀,直接使用
media_id = raw
return f"@{media_id}"
def main():
parser = argparse.ArgumentParser(
description="从 dt_media_upload 返回的 URL 中提取 mediaId"
)
parser.add_argument("url", help="dt_media_upload 返回的文件 URL")
args = parser.parse_args()
media_id = extract_media_id(args.url)
print(media_id)
if __name__ == "__main__":
main()