update_sheet_cell

app.py

@@ -2270,12 +2270,30 @@ def refresh_video_LLM_all_content_by_sheet(sheet_url, sheet_qa_column, video_ids
         video_id_col_index = header_row.index(target_id_column_name)
         qa_col_index = header_row.index(target_qa_column_name)
         # Google Sheet 的欄位字母 (A=0, B=1, ...)
-        qa_col_letter = chr(ord('A') + qa_col_index)
+        qa_col_letter = chr(ord('A') + qa_col_index) # <--- 使用這個變數
         print(f"'{target_id_column_name}' 欄位索引: {video_id_col_index}")
         print(f"'{target_qa_column_name}' 欄位索引: {qa_col_index} (欄位 {qa_col_letter})")
+
+        # *** 新增：嘗試取得工作表名稱 ***
+        sheet_name = SHEET_SERVICE.get_sheet_name_by_url(sheet_url) # 假設 SHEET_SERVICE 有此方法
+        if not sheet_name:
+            print("錯誤：無法從 URL 獲取工作表名稱。將無法執行即時更新。")
+            # 可以選擇在這裡 return 或設定一個標誌來跳過更新
+            # return {"success_video_ids": [], "failed_video_ids": list(set(video_ids_list))} # 範例：直接返回失敗
+            can_update_sheet = False
+        else:
+            print(f"取得工作表名稱: {sheet_name}")
+            can_update_sheet = True
+
     except ValueError as e:
         print(f"錯誤：在標頭列 {header_row} 中找不到必要的欄位: {e}")
         return {"success_video_ids": [], "failed_video_ids": []}
+    except AttributeError:
+         print("錯誤：SHEET_SERVICE 可能沒有 get_sheet_name_by_url 方法。請確認或實作。")
+         can_update_sheet = False
+    except Exception as e: # 捕捉其他可能的錯誤
+        print(f"取得工作表名稱時發生未預期錯誤: {e}")
+        can_update_sheet = False
 
     video_id_to_row_map = {}
     for i, row in enumerate(data_rows):
@@ -2314,26 +2332,29 @@ def refresh_video_LLM_all_content_by_sheet(sheet_url, sheet_qa_column, video_ids
             ids_to_refresh.append(video_id) # 如果檢查出錯，也歸類為需要刷新
 
     # 5. 批次更新 "OO" (Phase 1: GCS 已存在)
-    if ids_to_set_oo_phase1:
+    if ids_to_set_oo_phase1 and can_update_sheet: # <--- 加上 can_update_sheet 檢查
         print(f"\n準備批次更新 {len(ids_to_set_oo_phase1)} 個 Video ID 的 QA 為 '{sheet_qa_success_tag}' (GCS 已存在)...")
         update_data_oo1 = []
-        sheet_name = SHEET_SERVICE.get_sheet_name_by_url(sheet_url) # 需要 SHEET_SERVICE 提供此方法
-        if not sheet_name:
-             print("錯誤：無法獲取工作表名稱，無法執行批次更新。")
-             # 或者可以嘗試從 sheet_data 推斷，但不夠通用
-        else:
-            for vid, row_num in ids_to_set_oo_phase1:
-                update_data_oo1.append({
-                    'range': f"{sheet_name}!{qa_col_letter}{row_num}",
-                    'value': sheet_qa_success_tag
-                })
+        # sheet_name = SHEET_SERVICE.get_sheet_name_by_url(sheet_url) # 已在前面取得
+        # if not sheet_name:
+        #      print("錯誤：無法獲取工作表名稱，無法執行批次更新。")
+        # else:
+        for vid, row_num in ids_to_set_oo_phase1:
+            update_data_oo1.append({
+                'range': f"{sheet_name}!{qa_col_letter}{row_num}", # <--- 使用 sheet_name 和 qa_col_letter
+                'values': [[sheet_qa_success_tag]] # Google API batchUpdate 通常需要二維列表
+            })
+        if update_data_oo1: # 確保列表不是空的
             try:
                 # *** 假設 SHEET_SERVICE.batch_update_cells 存在 ***
+                # *** 注意：確認 batch_update_cells 的 value 格式是否需要 [[value]] ***
                 SHEET_SERVICE.batch_update_cells(sheet_url, update_data_oo1)
                 print(f"成功批次更新 {len(update_data_oo1)} 個儲存格為 '{sheet_qa_success_tag}'。")
             except Exception as e:
                 print(f"批次更新 QA 為 '{sheet_qa_success_tag}' 時發生錯誤: {e}")
                 # 這裡可以考慮是否要將這些 ID 移到失敗列表
+    elif ids_to_set_oo_phase1 and not can_update_sheet:
+        print(f"\n警告：無法取得工作表名稱，跳過批次更新 {len(ids_to_set_oo_phase1)} 個 GCS 已存在的狀態。")
 
     # 6. 逐一處理需要刷新的 Video ID，並即時更新 Sheet 狀態
     if ids_to_refresh:
@@ -2342,67 +2363,59 @@ def refresh_video_LLM_all_content_by_sheet(sheet_url, sheet_qa_column, video_ids
         failed_refresh_ids = []       # 初始化失敗列表
         # 這個迴圈負責執行刷新，並在每次完成後立即更新 Sheet
         for video_id in ids_to_refresh:
-            row_number = video_id_to_row_map[video_id] # 必定存在，前面已檢查
+            row_number = video_id_to_row_map[video_id] # 必定存在，前面已檢查 (1-based sheet row number)
             print(f"  正在刷新 {video_id} (第 {row_number} 列)...")
-            update_request = None
+            current_status_tag = sheet_qa_failed_tag # 預設失敗
             try:
                 refresh_video_LLM_all_content_by_id(video_id)
                 print(f"    - {video_id} 刷新成功。")
                 successfully_refreshed_ids.append((video_id, row_number)) # 仍然記錄成功，以備後續統計
-                # 準備單一成功更新請求
-                # *** 注意: 您需要確認 SHEET_NAME 和 STATUS_COL_LETTER ***
-                update_request = [{
-                    'range': f'{SHEET_NAME}!{STATUS_COL_LETTER}{row_number}', # 例如: '工作表1!G5'
-                    'values': [['OO']]
-                }]
+                current_status_tag = sheet_qa_success_tag # 設為成功
                 time.sleep(1) # 避免過於頻繁觸發其他 API (例如 GCS 刪除/上傳)
             except Exception as e:
                 print(f"    - {video_id} 刷新失敗: {str(e)}")
                 failed_refresh_ids.append((video_id, row_number)) # 仍然記錄失敗，以備後續統計
-                # 準備單一失敗更新請求
-                # *** 注意: 您需要確認 SHEET_NAME 和 STATUS_COL_LETTER ***
-                update_request = [{
-                    'range': f'{SHEET_NAME}!{STATUS_COL_LETTER}{row_number}', # 例如: '工作表1!G5'
-                    'values': [['XX']]
-                }]
+                current_status_tag = sheet_qa_failed_tag # 確認是失敗
                 time.sleep(5) # 失敗時稍作停頓
             finally:
-                # 無論成功或失敗，都嘗試更新該行的狀態
-                if update_request:
+                # 無論成功或失敗，都嘗試更新該行的狀態 (如果可以更新 Sheet)
+                if can_update_sheet: # <--- 加上 can_update_sheet 檢查
                     try:
-                        print(f"    - 更新 Sheet 第 {row_number} 列狀態...")
-                        # *** 假設 batch_update_cells 可以接受單一更新請求的列表 ***
-                        # *** 注意: 您需要確認 batch_update_cells 函式存在且能處理此格式 ***
-                        SHEET_SERVICE.batch_update_cells(sheet_url, update_request) # 假設使用 SHEET_SERVICE
-                        print(f"      - 第 {row_number} 列狀態更新成功。")
+                        # *** 使用 update_sheet_cell 更新單一儲存格 ***
+                        # row_number 是 1-based 的實際列號
+                        # update_sheet_cell 需要 0-based 的索引，所以傳遞 row_number - 1
+                        target_row_index = row_number - 1
+                        print(f"    - 更新 Sheet 第 {row_number} 列, 欄位索引 {qa_col_index} 狀態為 '{current_status_tag}'...")
+
+                        # 呼叫 update_sheet_cell
+                        update_success = SHEET_SERVICE.update_sheet_cell(
+                            sheet_url=sheet_url,
+                            target_row_index_in_data=target_row_index, # 傳遞 0-based 索引
+                            qa_col_index=qa_col_index,                 # 傳遞 0-based 索引
+                            qa_result=current_status_tag               # 傳遞要寫入的值
+                        )
+
+                        if update_success:
+                            print(f"      - 第 {row_number} 列狀態更新成功。")
+                        else:
+                            # update_sheet_cell 內部已經記錄了錯誤，這裡可以選擇性地再記錄一次
+                            print(f"    - !! 更新 Sheet 第 {row_number} 列狀態失敗 (詳見先前日誌)。")
+
                     except Exception as update_e:
-                        print(f"    - !! 更新 Sheet 第 {row_number} 列狀態失敗: {str(update_e)}")
-                        # 即使更新 Sheet 失敗，也要記錄下來，避免影響下一個影片的處理
-                        # 可以在這裡加入更詳細的錯誤記錄機制
-
-    # 7. 批次更新刷新結果 (OO for success, XX for failure) <-- 這段程式碼現在不需要了，因為已在迴圈內即時更新
-    # if successfully_refreshed_ids or failed_refresh_ids:
-    #     print("\n開始批次更新 Google Sheet 刷新結果...")
-    #     update_requests = []
-    #     # ... (原本準備 update_requests 的程式碼) ...
-    #
-    #     if update_requests:
-    #         try:
-    #             batch_update_cells(update_requests)
-    #             print(f"  成功更新 {len(successfully_refreshed_ids)} 個成功，{len(failed_refresh_ids)} 個失敗的狀態。")
-    #         except Exception as e:
-    #             print(f"  批次更新 Google Sheet 失敗: {str(e)}")
-    #     else:
-    #         print("  沒有需要更新的狀態。")
-
-    # 8. 整理最終結果
+                        # 捕獲呼叫 update_sheet_cell 本身可能發生的意外錯誤
+                        print(f"    - !! 呼叫 update_sheet_cell 時發生未預期錯誤: {str(update_e)}")
+                        # 即使更新 Sheet 失敗，也要記錄下來
+                else:
+                     print(f"    - 無法取得工作表名稱，跳過更新 Sheet 第 {row_number} 列狀態。")
+
+    # 7. 整理最終結果 (移除舊的批次更新邏輯)
+    print("\n=== 批次處理完成 ===")
     final_success_ids = [item[0] for item in ids_to_set_oo_phase1] + [item[0] for item in successfully_refreshed_ids]
     final_failed_ids = [item[0] for item in failed_refresh_ids]
     # 將在工作表中找不到的 ID 也視為失敗
     initial_not_found = [vid for vid in video_ids_list if vid not in video_id_to_row_map]
     final_failed_ids.extend(initial_not_found)
 
-    print("\n=== 批次處理完成 ===")
     print(f"成功處理 (或 GCS 已存在): {len(final_success_ids)} 個")
     print(f"處理失敗 (或未找到/刷新失敗): {len(final_failed_ids)} 個")