142 lines
4.4 KiB
Python
142 lines
4.4 KiB
Python
import requests
|
||
import sys
|
||
import os
|
||
import subprocess
|
||
|
||
|
||
def get_api_key(key_name):
|
||
"""从api-key-vault读取Key"""
|
||
vault_path = os.path.join(
|
||
os.path.dirname(__file__),
|
||
"..", "..", # 从scripts/退到logistics-tracker/再退到OpenClaw/
|
||
"api-key-vault", "scripts", "vault.py"
|
||
)
|
||
vault_path = os.path.normpath(vault_path)
|
||
|
||
result = subprocess.run(
|
||
["python", vault_path, "get", key_name],
|
||
capture_output=True, text=True
|
||
)
|
||
key = result.stdout.strip()
|
||
|
||
if not key or key == "ERROR:KEY_NOT_FOUND":
|
||
return None
|
||
return key
|
||
|
||
|
||
def check_entitlement(skill_slug):
|
||
auth_base = (os.getenv("JIANGCHANG_AUTH_BASE_URL") or "").strip().rstrip("/")
|
||
if not auth_base:
|
||
return True, ""
|
||
|
||
user_id = (os.getenv("JIANGCHANG_USER_ID") or "").strip()
|
||
if not user_id:
|
||
return False, "鉴权失败：缺少用户身份（JIANGCHANG_USER_ID）"
|
||
|
||
auth_api_key = (os.getenv("JIANGCHANG_AUTH_API_KEY") or "").strip()
|
||
timeout = int((os.getenv("JIANGCHANG_AUTH_TIMEOUT_SECONDS") or "5").strip())
|
||
headers = {"Content-Type": "application/json"}
|
||
if auth_api_key:
|
||
headers["Authorization"] = f"Bearer {auth_api_key}"
|
||
|
||
payload = {
|
||
"user_id": user_id,
|
||
"skill_slug": skill_slug,
|
||
"trace_id": (os.getenv("JIANGCHANG_TRACE_ID") or "").strip(),
|
||
"context": {"entry": "main.py"},
|
||
}
|
||
try:
|
||
res = requests.post(
|
||
f"{auth_base}/api/entitlements/check",
|
||
json=payload,
|
||
headers=headers,
|
||
timeout=timeout,
|
||
)
|
||
except requests.RequestException as exc:
|
||
return False, f"鉴权请求失败：{exc}"
|
||
|
||
if res.status_code != 200:
|
||
return False, f"鉴权服务异常：HTTP {res.status_code}"
|
||
try:
|
||
body = res.json()
|
||
except ValueError:
|
||
return False, "鉴权服务异常：返回非 JSON"
|
||
|
||
code = body.get("code")
|
||
data = body.get("data") or {}
|
||
if code != 200:
|
||
return False, str(body.get("msg") or "鉴权失败")
|
||
if not data.get("allow", False):
|
||
return False, str(data.get("reason") or "未购买或已过期")
|
||
return True, ""
|
||
|
||
|
||
def query_tracking(tracking_number):
|
||
api_key = get_api_key("17track")
|
||
if not api_key:
|
||
return "错误：未找到17track的API Key，请先运行：\npython api-key-vault/scripts/vault.py set 17track 你的Key"
|
||
|
||
url = "https://api.17track.net/track/v2.2/gettrackinfo"
|
||
headers = {
|
||
"Content-Type": "application/json",
|
||
"17token": api_key
|
||
}
|
||
body = {
|
||
"data": [{"number": tracking_number}]
|
||
}
|
||
|
||
try:
|
||
response = requests.post(url, json=body, headers=headers, timeout=10)
|
||
result = response.json()
|
||
|
||
accepted = result.get("data", {}).get("accepted", [])
|
||
if not accepted:
|
||
return f"抱歉，单号 {tracking_number} 查询不到信息，可能还未入网，请稍后再试。"
|
||
|
||
track_info = accepted[0].get("track", {})
|
||
providers = track_info.get("tracking", {}).get("providers", [])
|
||
|
||
if not providers or not providers[0].get("events"):
|
||
return f"📦 单号：{tracking_number}\n该单号已入网，暂无轨迹更新，请稍后再查。"
|
||
|
||
events = providers[0]["events"]
|
||
latest = events[0]
|
||
|
||
recent_lines = []
|
||
for e in events[:3]:
|
||
time = e.get("time_iso", "")
|
||
location = e.get("location", "") or "未知"
|
||
desc = e.get("description", "")
|
||
recent_lines.append(f" · {time} {location} {desc}")
|
||
|
||
recent_text = "\n".join(recent_lines)
|
||
|
||
return (
|
||
f"📦 单号：{tracking_number}\n"
|
||
f"📍 最新状态：{latest.get('description', '未知')}\n"
|
||
f"🕐 更新时间：{latest.get('time_iso', '未知')}\n"
|
||
f"🗺 最新位置：{latest.get('location', '未知')}\n\n"
|
||
f"最近轨迹：\n{recent_text}"
|
||
)
|
||
|
||
except requests.exceptions.Timeout:
|
||
return "查询超时，请稍后重试。"
|
||
except Exception as e:
|
||
return f"查询失败：{str(e)}"
|
||
|
||
|
||
def main(argv=None) -> int:
|
||
args = argv if argv is not None else sys.argv[1:]
|
||
if len(args) < 1:
|
||
print("用法：python main.py <单号>")
|
||
return 1
|
||
ok, reason = check_entitlement("logistics-tracker")
|
||
if not ok:
|
||
print(f"❌ {reason}")
|
||
return 1
|
||
print(query_tracking(args[0]))
|
||
return 0
|
||
|
||
|
||
if __name__ == "__main__":
|
||
raise SystemExit(main()) |