当前位置：首页 > news >正文

【效率提升教程】飞书自动化上传图片和文字

news 2025/7/5 6:37:15

最近总是需要可视化模型结果，手动复制图片又慢又麻烦～

核心就是上传图片和文字这两个函数，替换这三个地方就可以实现啦～

SHEET_NAME = "" #子表名称
SPREADSHEET_TOKEN = "" #表格名

HEADERS = {
"Authorization": "Bearer u-eSfStf4a1cvWVkcyX03FHl5gjg11gk2XUgG0ggSy8JcL", #替换为自己的key
"Content-Type": "application/json",
}

import os
import json
import base64
import time
import requests# 飞书表格配置
SHEET_NAME = "" #子表名称
SPREADSHEET_TOKEN = "" #表格名
IMG_URL = f"https://open.feishu.cn/open-apis/sheets/v2/spreadsheets/{SPREADSHEET_TOKEN}/values_image"
TEXT_URL = f"https://open.feishu.cn/open-apis/sheets/v2/spreadsheets/{SPREADSHEET_TOKEN}/values"
HEADERS = {"Authorization": "Bearer u-eSfStf4a1cvWVkcyX03FHl5gjg11gk2XUgG0ggSy8JcL", #替换为自己的key"Content-Type": "application/json",
}def upload_image(image_path, column_index, row_index, max_retries=3):for attempt in range(max_retries):try:with open(image_path, "rb") as f:fb = f.read()fb = base64.b64encode(fb).decode("utf-8")data = {"range": f"{SHEET_NAME}!{chr(65 + column_index)}{row_index}:{chr(65 + column_index)}{row_index}","image": fb,"name": os.path.basename(image_path),}rsp = requests.post(IMG_URL, data=json.dumps(data), headers=HEADERS)rsp.raise_for_status()print(f"✅ 上传图片成功: {image_path} -> 行{row_index}, 列{column_index}")return Trueexcept Exception as e:print(f"❌ 上传图片失败: {e}（尝试 {attempt + 1}/{max_retries}）")time.sleep(2)return Falsedef upload_text(text, column_index, row_index):try:data = {"valueRange": {"range": f"{SHEET_NAME}!{chr(65 + column_index)}{row_index}:{chr(65 + column_index)}{row_index}","values": [[text]],}}text_rsp = requests.put(TEXT_URL, data=json.dumps(data), headers=HEADERS)text_rsp.raise_for_status()print(f"📝 上传文字成功 -> 行{row_index}, 列{column_index}")return Trueexcept Exception as e:print(f"❌ 上传文字失败: {e}")return Falsedef process_jsonl(jsonl_path, column_img=0, column_txt=1):start_row = 2row_index = start_rowwith open(jsonl_path, "r", encoding="utf-8") as f:for line in f:try:data = json.loads(line)image_path = data["image"]score_text = str(data.get("score", "N/A"))if not os.path.exists(image_path):print(f"❌ 缺失图像: {image_path}")continueupload_image(image_path, column_img, row_index)upload_text(score_text, column_txt, row_index)row_index += 1except Exception as e:print(f"❌ 解析失败: {e}")continueif __name__ == "__main__":jsonl_file = "/mnt/data_vlm/jianai/edit_data/image_score.jsonl"  # 你自己的 jsonl 路径process_jsonl(jsonl_file, column_img=6*2, column_txt=6*2 + 1)  # 根据表格实际列设置起始列