236 lines
8.9 KiB
Python
236 lines
8.9 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
英文台词优化 & 回填脚本 v2 — 使用精确行号
|
||
规则：MEMORY.md 对话台词优化通用规则
|
||
知识点：grandmother, grandfather, visit, today
|
||
句型知识点：... visit+sb, Today is...
|
||
"""
|
||
import json, subprocess
|
||
|
||
# (sheet_row, optimized_F_value) — exact sheet row numbers
|
||
rows = [
|
||
(3, "Open up!"),
|
||
(4, "Your letter is here!"),
|
||
(6, "Nobody answers?"),
|
||
(7, "Alright... I'll go in myself."),
|
||
(9, "Whoa! What a mess!"),
|
||
(11, "What's wrong?"),
|
||
(12, "Vicky? Can you hear me?"),
|
||
(14, "Today is a No-Mom-and-Dad day!"),
|
||
(15, "Our parents aren't home!"),
|
||
(16, "So we can do whatever we want!"),
|
||
(17, "Really?"),
|
||
(19, "Yes!"),
|
||
(20, "Today is a good day!"),
|
||
(21, "Huh?"),
|
||
(22, "Are you sure today is a good day?"),
|
||
(23, "So on a good day, nobody says sorry?"),
|
||
(24, "Eva, Peter, say sorry."),
|
||
(25, "Okay. We're sorry."),
|
||
(27, "Ah, kids."),
|
||
(28, "Come visit us!"),
|
||
(29, "Oh, I'm not here to visit."),
|
||
(30, "I'm here to give you something."),
|
||
(32, "Wait a moment."),
|
||
(34, "Hello? Grandfather and grandmother, how are you?"),
|
||
(35, "Good, good! We're great!"),
|
||
(36, "How are you all, my lovely grandchildren?"),
|
||
(37, "We're great! Today is so much fun!"),
|
||
(38, "Great to hear that!"),
|
||
(39, "Are you ready for us?"),
|
||
(40, "I'm so excited to visit you all!"),
|
||
(41, "Sure! When will you visit us?"),
|
||
(42, "Today is the day, of course!"),
|
||
(43, "What? You're both coming today?"),
|
||
(44, "Yes! Me and Justin. Your grandmother and grandfather."),
|
||
(45, "We sent you a letter. Remember?"),
|
||
(46, "See you later!"),
|
||
(47, "I love you! Bye!"),
|
||
(49, "Oh no! What did they say on the phone?"),
|
||
(50, "Your grandfather and grandmother are coming to visit you!"),
|
||
(51, "On your good day."),
|
||
(53, "Eva! Peter! Alex! We're in trouble!"),
|
||
(55, "What happened?"),
|
||
(56, "Grandfather and grandmother are coming to visit us!"),
|
||
(57, "So what?"),
|
||
(58, "Today! Right now!"),
|
||
(60, "No, not today..."),
|
||
(61, "Look at this place!"),
|
||
(62, "Yes, today is a total mess."),
|
||
(63, "It's not the best day for visitors."),
|
||
(64, "They could be here at any minute!"),
|
||
(65, "How will they feel about this?"),
|
||
(67, "Grandfather and grandmother will be so upset!"),
|
||
(69, "What should we do?"),
|
||
(70, "Don't worry! I'll help you clean up!"),
|
||
(71, "Okay, first pick up the toys from the floor."),
|
||
(73, "Oh no! No time to put these away!"),
|
||
(74, "Put them in the cupboard!"),
|
||
(75, "Your grandfather and grandmother won't see!"),
|
||
(76, "Good idea!"),
|
||
(79, "All done!"),
|
||
(80, "We need to clean the floor!"),
|
||
(81, "I'll do it!"),
|
||
(84, "Will Grandfather and Grandmother be happy?"),
|
||
(85, "Well... this will do for now."),
|
||
(86, "Before your grandfather and grandmother come to visit."),
|
||
(88, "Alex! The sofa is a mess!"),
|
||
(89, "I'll clean it."),
|
||
(91, "I can't get the chocolate off!"),
|
||
(92, "I'm good at many things. Just not housework!"),
|
||
(94, "Oh no! We can't clean up in time!"),
|
||
(95, "I wish we had more time for this!"),
|
||
(96, "Why today, of all days?"),
|
||
(97, "I don't want them to see our house like this."),
|
||
(98, "Why didn't they tell us earlier?"),
|
||
(100, "I think they did tell you..."),
|
||
(101, "In this letter."),
|
||
(102, "What is in this letter?"),
|
||
(104, "Hi, your grandfather is writing this letter."),
|
||
(105, "I will come to visit you guys."),
|
||
(106, "Your grandmother will visit you too!"),
|
||
(107, "We love you, see you soon!"),
|
||
(108, "They were so happy to visit you!"),
|
||
(109, "Yes. We should be happy too."),
|
||
(110, "We should not be grumpy like this."),
|
||
(111, "Let's do something to welcome them!"),
|
||
(112, "You can make something they like!"),
|
||
(113, "What is their favourite food?"),
|
||
(114, "Grandmother likes apples."),
|
||
(115, "Grandfather likes pie."),
|
||
(116, "Then you can make apple pie for them!"),
|
||
(117, "I remember the recipe."),
|
||
(118, "I can fix the cooking robot."),
|
||
(119, "Okay! Apple pie is today's welcome gift!"),
|
||
(121, "You did it!"),
|
||
(122, "Now we just wait for them to arrive."),
|
||
(124, "They're coming!"),
|
||
(126, "Who wants a big hug from Grandfather?"),
|
||
(127, "Me! Me!"),
|
||
(128, "Who wants Grandmother's homemade candy?"),
|
||
(129, "Me! Me!"),
|
||
(131, "We made something for you."),
|
||
(132, "Mmm! What's that smell? Is that a pie?"),
|
||
(133, "I smell apples!"),
|
||
(134, "Whoa! I know that smell! Apple pie!"),
|
||
(135, "That is our favourite!"),
|
||
(137, "Oh, you have a friend over!"),
|
||
(138, "Hello! I am here to visit Vicky."),
|
||
(139, "Nice to meet you again!"),
|
||
(140, "I guess you helped out today too."),
|
||
(142, "Today seems pretty busy."),
|
||
(143, "Yes, we made a little mess before you came."),
|
||
(144, "Today was not easy."),
|
||
(145, "That's okay."),
|
||
(146, "Grandfather and Grandmother can fix anything."),
|
||
(147, "That's what family is for!"),
|
||
(148, "Mess or not, we love seeing you!"),
|
||
]
|
||
|
||
print(f"Total dialogue lines: {len(rows)}")
|
||
|
||
# Build the complete column F data (one value per row, 1-based sheet rows 2-149)
|
||
# We need to clear the ENTIRE column first, then fill only dialogue rows.
|
||
# The API PUT overwrites the specified range. Write all rows 2-149.
|
||
# Use a dict for fast lookup
|
||
f_map = {r: v for r, v in rows}
|
||
|
||
# Get token
|
||
with open("/root/.openclaw/credentials/xiaoyan/config.json") as f:
|
||
cfg = json.load(f)
|
||
app_id = "cli_a931175d41799cc7"
|
||
app_secret = cfg["apps"][0]["appSecret"]
|
||
|
||
result = subprocess.run([
|
||
"curl", "-s", "-X", "POST",
|
||
"https://open.feishu.cn/open-apis/auth/v3/tenant_access_token/internal",
|
||
"-H", "Content-Type: application/json",
|
||
"-d", json.dumps({"app_id": app_id, "app_secret": app_secret})
|
||
], capture_output=True, text=True)
|
||
token = json.loads(result.stdout)["tenant_access_token"]
|
||
|
||
spreadsheet_token = "DA7csROG6hmnQMt59oScOLRfnWc"
|
||
sheet_id = "Nr3Ovs"
|
||
|
||
# Step 1: Clear entire column F (rows 2-149)
|
||
print("Clearing column F (rows 2-149)...")
|
||
empty_payload = {
|
||
"valueRange": {
|
||
"range": f"{sheet_id}!F2:F149",
|
||
"values": [[""] for _ in range(148)] # 149-2+1 = 148 rows
|
||
}
|
||
}
|
||
result = subprocess.run([
|
||
"curl", "-s", "-X", "PUT",
|
||
f"https://open.feishu.cn/open-apis/sheets/v2/spreadsheets/{spreadsheet_token}/values",
|
||
"-H", f"Authorization: Bearer {token}",
|
||
"-H", "Content-Type: application/json",
|
||
"-d", json.dumps(empty_payload)
|
||
], capture_output=True, text=True)
|
||
resp = json.loads(result.stdout)
|
||
print(f" Clear: code={resp.get('code')}, msg={resp.get('msg')}")
|
||
assert resp.get("code") == 0, f"Clear failed: {result.stdout[:200]}"
|
||
|
||
# Step 2: Write optimized values to each dialogue row
|
||
# Group consecutive rows for batching
|
||
batch_start = None
|
||
batch_end = None
|
||
batch_values = []
|
||
|
||
def flush_batch():
|
||
global batch_start, batch_end, batch_values
|
||
if not batch_values:
|
||
return
|
||
range_str = f"{sheet_id}!F{batch_start}:F{batch_end}"
|
||
payload = {"valueRange": {"range": range_str, "values": batch_values}}
|
||
result = subprocess.run([
|
||
"curl", "-s", "-X", "PUT",
|
||
f"https://open.feishu.cn/open-apis/sheets/v2/spreadsheets/{spreadsheet_token}/values",
|
||
"-H", f"Authorization: Bearer {token}",
|
||
"-H", "Content-Type: application/json",
|
||
"-d", json.dumps(payload)
|
||
], capture_output=True, text=True)
|
||
resp = json.loads(result.stdout)
|
||
print(f" Write {range_str}: code={resp.get('code')}, msg={resp.get('msg')}")
|
||
assert resp.get("code") == 0, f"Write failed: {result.stdout[:200]}"
|
||
batch_start = None
|
||
batch_end = None
|
||
batch_values = []
|
||
|
||
# Also need to fill empty rows between dialogue rows
|
||
# Strategy: write row by row, merging consecutive rows into batches
|
||
for sheet_row in range(2, 150):
|
||
value = f_map.get(sheet_row, "")
|
||
if batch_start is None:
|
||
batch_start = sheet_row
|
||
batch_end = sheet_row
|
||
batch_values = [[value]]
|
||
else:
|
||
batch_end = sheet_row
|
||
batch_values.append([value])
|
||
|
||
flush_batch()
|
||
|
||
print("Done! Verifying...")
|
||
|
||
# Step 3: Verify key rows
|
||
check_rows = [3, 7, 14, 23, 28, 34, 44, 49, 56, 64, 92, 97, 111, 121, 134, 147]
|
||
for r in check_rows:
|
||
result = subprocess.run([
|
||
"curl", "-s", "-X", "GET",
|
||
f"https://open.feishu.cn/open-apis/sheets/v2/spreadsheets/{spreadsheet_token}/values/{sheet_id}!D{r}:F{r}?valueRenderOption=ToString",
|
||
"-H", f"Authorization: Bearer {token}"
|
||
], capture_output=True, text=True)
|
||
vals = json.loads(result.stdout).get("data", {}).get("valueRange", {}).get("values", [[]])[0]
|
||
role = vals[0] if len(vals) > 0 else "-"
|
||
e_val = vals[1] if len(vals) > 1 else "-"
|
||
f_val = vals[2] if len(vals) > 2 else "-"
|
||
expected = f_map.get(r, "-")
|
||
status = "✓" if f_val == (expected or "") else "✗ MISMATCH"
|
||
if status != "✓":
|
||
print(f" Row {r}: [{role}] E='{e_val}' F='{f_val}' expected='{expected}' {status}")
|
||
else:
|
||
print(f" Row {r}: [{role}] F='{f_val}' {status}")
|
||
|
||
print("\nAll verified!")
|