import os import pickle import json import hashlib from datetime import datetime from typing import Any, Dict, List, Optional import re # Import re để phân tích range import time # Import để sử dụng sleep import random # Import để tạo độ trễ ngẫu nhiên from google.oauth2.service_account import Credentials from google.auth.transport.requests import Request from google_auth_oauthlib.flow import InstalledAppFlow from googleapiclient.discovery import build from loguru import logger from .utils import timing_decorator_sync from .constants import SHEET_RANGE SCOPES = ['https://www.googleapis.com/auth/spreadsheets'] def generate_conversation_id(user_id: str, page_id: str, timestamp: str) -> str: """Tạo ID hội thoại duy nhất.""" hash_input = f"{user_id}:{page_id}:{timestamp}" return hashlib.sha256(hash_input.encode()).hexdigest()[:32] def _flatten_and_unique_timestamps(items: Any) -> List[Any]: """ Hàm tiện ích để làm phẳng danh sách timestamp (xử lý list lồng nhau) và loại bỏ các giá trị trùng lặp, giữ nguyên thứ tự. """ if not isinstance(items, list): return [items] flat_list = [] for item in items: if isinstance(item, list): flat_list.extend(_flatten_and_unique_timestamps(item)) else: flat_list.append(item) return list(dict.fromkeys(flat_list)) def _get_start_row_from_range(range_string: str) -> int: """ Phân tích một chuỗi range (ví dụ: 'Sheet1!A2:Z') để lấy ra số của dòng bắt đầu. """ match = re.search(r"[A-Z]+([0-9]+)", range_string) if match: try: return int(match.group(1)) except (ValueError, IndexError): pass logger.warning(f"Không thể xác định dòng bắt đầu từ range '{range_string}'. Mặc định là 2.") return 2 class SheetsClient: def __init__(self, credentials_file: str, token_file: str, sheet_id: str): self.credentials_file = credentials_file self.token_file = token_file self.sheet_id = sheet_id self.creds = None self.service = None @timing_decorator_sync def authenticate(self) -> None: """Xác thực với Google Sheets API.""" credentials_json = os.getenv("GOOGLE_SHEETS_CREDENTIALS_JSON") if credentials_json: info = json.loads(credentials_json) self.creds = Credentials.from_service_account_info(info, scopes=SCOPES) else: if os.path.exists(self.token_file): with open(self.token_file, 'rb') as token: self.creds = pickle.load(token) if not self.creds or not self.creds.valid: if self.creds and self.creds.expired and self.creds.refresh_token: self.creds.refresh(Request()) else: flow = InstalledAppFlow.from_client_secrets_file(self.credentials_file, SCOPES) self.creds = flow.run_local_server(port=0) with open(self.token_file, 'wb') as token: pickle.dump(self.creds, token) self.service = build('sheets', 'v4', credentials=self.creds) @timing_decorator_sync def get_conversation_history(self, user_id: str, page_id: str) -> List[Dict[str, Any]]: """Lấy lịch sử hội thoại từ sheet bằng vị trí cột cố định để đảm bảo độ tin cậy.""" logger.debug(f"[get_conversation_history] Bắt đầu lấy lịch sử cho user_id: '{user_id}' và page_id: '{page_id}'") try: if not self.service: self.authenticate() range_name = SHEET_RANGE result = self.service.spreadsheets().values().get( spreadsheetId=self.sheet_id, range=range_name ).execute() values = result.get('values', []) history = [] if not values: logger.warning(f"[get_conversation_history] Không tìm thấy dữ liệu trong sheet. Range: {range_name}") return [] logger.debug(f"[get_conversation_history] Đã lấy được {len(values)} dòng từ sheet. Bắt đầu xử lý...") start_row = _get_start_row_from_range(range_name) for i, row in enumerate(values, start=start_row): if len(row) < 14: row.extend([""] * (14 - len(row))) sheet_recipient_id = row[4] # Cột E sheet_page_id = row[5] # Cột F if str(sheet_recipient_id) == str(user_id) and str(sheet_page_id) == str(page_id): try: timestamps_raw = json.loads(row[12]) # Cột M timestamps = _flatten_and_unique_timestamps(timestamps_raw) except (json.JSONDecodeError, TypeError): timestamps = [] history.append({ 'conversation_id': row[0], 'originalcommand': row[1], 'originalcontent': row[2], 'originalattachments': json.loads(row[3]) if row[3] else [], 'recipient_id': row[4], 'page_id': row[5], 'originaltext': row[6], 'originalvehicle': row[7], 'originalaction': row[8], 'originalpurpose': row[9], 'originalquestion': row[10], 'systemresponse': row[11], 'timestamp': timestamps, 'isdone': str(row[13]).lower() == 'true' }) logger.debug(f"[get_conversation_history] Hoàn tất xử lý. Tìm thấy {len(history)} bản ghi lịch sử.") return history except Exception as e: logger.error(f"Error getting conversation history: {e}", exc_info=True) return [] @timing_decorator_sync def log_conversation( self, **kwargs: Any ) -> Optional[Dict[str, Any]]: """ Thực hiện "UPSERT" (Update hoặc Insert) một hội thoại với logic chống trùng lặp mạnh mẽ. """ try: if not self.service: self.authenticate() # --- 1. Thiết lập & Lấy Header --- sheet_name_match = re.match(r"([^!]+)!", SHEET_RANGE) sheet_name = sheet_name_match.group(1) if sheet_name_match else "Sheet1" header_range = f"{sheet_name}!A1:Z1" header_result = self.service.spreadsheets().values().get(spreadsheetId=self.sheet_id, range=header_range).execute() header = header_result.get('values', [[]])[0] if not header: logger.error(f"Không thể lấy được header từ range '{header_range}'.") return None # --- 2. Đọc dữ liệu và xác định các định danh --- data_result = self.service.spreadsheets().values().get(spreadsheetId=self.sheet_id, range=SHEET_RANGE).execute() values = data_result.get('values', []) # Định danh của sự kiện đang xử lý recipient_id = str(kwargs.get('recipient_id')).strip() page_id = str(kwargs.get('page_id')).strip() ts_list = _flatten_and_unique_timestamps(kwargs.get('timestamp', [])) event_timestamp = str(ts_list[-1]).strip() if ts_list else '' logger.debug(f"UPSERT: Bắt đầu tìm kiếm với recipient_id='{recipient_id}', page_id='{page_id}', event_timestamp='{event_timestamp}'") # --- 3. Tìm kiếm bản ghi đã tồn tại --- found_row_index = -1 found_row_data = {} start_row = _get_start_row_from_range(SHEET_RANGE) try: id_col_idx = header.index('conversation_id') recipient_col_idx = header.index('recipient_id') page_col_idx = header.index('page_id') timestamp_col_idx = header.index('timestamp') except ValueError as e: logger.error(f"Thiếu cột bắt buộc trong header: {e}") return None target_conv_id = str(kwargs.get('conversation_id') or '').strip() if target_conv_id: logger.debug(f"UPSERT: Ưu tiên tìm bằng conversation_id='{target_conv_id}'") for i, row in enumerate(values, start=start_row): if len(row) > id_col_idx: sheet_conv_id = str(row[id_col_idx]).strip() is_match = sheet_conv_id == target_conv_id # logger.trace(f"Dòng {i}: So sánh ID: '{sheet_conv_id}' == '{target_conv_id}' -> {is_match}") if is_match: found_row_index = i found_row_data = dict(zip(header, row)) logger.success(f"Tìm thấy bằng conversation_id tại dòng {i}.") break if found_row_index == -1: logger.debug(f"UPSERT: Không tìm thấy bằng ID, chuyển sang tìm bằng (user, page, timestamp).") for i, row in enumerate(values, start=start_row): if len(row) <= max(recipient_col_idx, page_col_idx, timestamp_col_idx): continue sheet_recipient_id = str(row[recipient_col_idx]).strip() sheet_page_id = str(row[page_col_idx]).strip() id_match = (sheet_recipient_id == recipient_id) and (sheet_page_id == page_id) # logger.trace(f"Dòng {i}: So sánh (user, page): ('{sheet_recipient_id}' == '{recipient_id}') AND ('{sheet_page_id}' == '{page_id}') -> {id_match}") if id_match: try: sheet_timestamps = [str(ts).strip() for ts in _flatten_and_unique_timestamps(json.loads(row[timestamp_col_idx]))] ts_match = event_timestamp and event_timestamp in sheet_timestamps # logger.trace(f"Dòng {i}: So sánh timestamp: '{event_timestamp}' in {sheet_timestamps} -> {ts_match}") if ts_match: found_row_index = i found_row_data = dict(zip(header, row)) logger.success(f"Tìm thấy bằng (user, page, timestamp) tại dòng {i}.") break except (json.JSONDecodeError, TypeError): continue # --- 4. Thực hiện UPDATE hoặc INSERT --- if found_row_index != -1: # --- LOGIC CẬP NHẬT (UPDATE) --- logger.info(f"Đang cập nhật hội thoại tại dòng {found_row_index}") updated_data = found_row_data.copy() for key, value in kwargs.items(): if value is not None and value != '' or isinstance(value, bool): updated_data[key] = value existing_ts = _flatten_and_unique_timestamps(json.loads(found_row_data.get('timestamp', '[]'))) new_ts = _flatten_and_unique_timestamps(kwargs.get('timestamp', [])) updated_data['timestamp'] = _flatten_and_unique_timestamps(existing_ts + new_ts) row_data_to_write = [] for col_name in header: value = updated_data.get(col_name, '') if col_name in ['originalattachments', 'timestamp']: row_data_to_write.append(json.dumps(value or [])) elif col_name == 'isdone': row_data_to_write.append(str(value).lower()) else: row_data_to_write.append(str(value)) range_to_update = f"{sheet_name}!A{found_row_index}" body = {'values': [row_data_to_write]} self.service.spreadsheets().values().update(spreadsheetId=self.sheet_id, range=range_to_update, valueInputOption='RAW', body=body).execute() kwargs.update(updated_data) return kwargs else: # --- LOGIC TẠO MỚI (INSERT) --- logger.info(f"Không tìm thấy dòng khớp. Tiến hành tạo bản ghi mới.") kwargs['conversation_id'] = kwargs.get('conversation_id') or generate_conversation_id(recipient_id, page_id, event_timestamp) kwargs['timestamp'] = _flatten_and_unique_timestamps(kwargs.get('timestamp', [])) row_data_to_write = [] for col_name in header: value = kwargs.get(col_name, '') if col_name in ['originalattachments', 'timestamp']: row_data_to_write.append(json.dumps(value or [])) elif col_name == 'isdone': row_data_to_write.append(str(value).lower()) else: row_data_to_write.append(str(value)) body = {'values': [row_data_to_write]} self.service.spreadsheets().values().append(spreadsheetId=self.sheet_id, range=SHEET_RANGE, valueInputOption='RAW', insertDataOption='INSERT_ROWS', body=body).execute() return kwargs except Exception as e: logger.error(f"Lỗi khi ghi/cập nhật conversation: {e}", exc_info=True) return None