quick fix timestamp

app/sheets.py

@@ -16,20 +16,22 @@ from .constants import SHEET_RANGE
 
 SCOPES = ['https://www.googleapis.com/auth/spreadsheets']
 
-def generate_conversation_id(user_id: str, page_id: str, timestamp: Any) -> str:
+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: List[Any]) -> List[Any]:
+# --- HÀM TIỆN ÍCH ĐƯỢC THÊM VÀO ĐỂ FIX LỖI ---
+def _flatten_and_unique_timestamps(items: Any) -> List[Any]:
     """
-    Hàm tiện ích để làm phẳng danh sách timestamp và loại bỏ các giá trị trùng lặp.
-    Xử lý được cả list lồng nhau.
+    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ự.
     """
-    flat_list = []
     if not isinstance(items, list):
+        # Nếu đầu vào không phải list, gói nó vào một list
         return [items]
         
+    flat_list = []
     for item in items:
         if isinstance(item, list):
             flat_list.extend(_flatten_and_unique_timestamps(item))
@@ -70,118 +72,164 @@ class SheetsClient:
 
     @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, xử lý timestamp một cách an toàn."""
+        """Lấy lịch sử hội thoại từ sheet."""
         try:
             if not self.service:
                 self.authenticate()
-            
+            range_name = SHEET_RANGE
             result = self.service.spreadsheets().values().get(
                 spreadsheetId=self.sheet_id,
-                range=SHEET_RANGE
+                range=range_name
             ).execute()
             values = result.get('values', [])
-            if not values:
+            history = []
+            if not values or len(values) < 1:
                 return []
-
+            
             header = values[0]
-            history = []
-            for row in values[1:]: # Bỏ qua dòng header
-                # Đảm bảo row có đủ cột
+            for row in values[1:]:
                 row_data = dict(zip(header, row + [""] * (len(header) - len(row))))
                 
                 if row_data.get('recipient_id') == user_id and row_data.get('page_id') == page_id:
-                    # Cải tiến: Xử lý an toàn việc đọc timestamp
                     try:
-                        timestamps_raw = json.loads(row_data.get('timestamp', '[]'))
-                        timestamps = _flatten_and_unique_timestamps(timestamps_raw)
+                        # Dùng hàm tiện ích để đảm bảo timestamp đọc ra luôn phẳng
+                        timestamps = _flatten_and_unique_timestamps(json.loads(row_data.get('timestamp', '[]')))
                     except (json.JSONDecodeError, TypeError):
                         timestamps = []
-
+                    
                     row_data['timestamp'] = timestamps
                     row_data['originalattachments'] = json.loads(row_data.get('originalattachments', '[]'))
                     row_data['isdone'] = str(row_data.get('isdone', 'false')).lower() == 'true'
                     history.append(row_data)
             return history
         except Exception as e:
-            logger.error(f"Lỗi khi lấy lịch sử hội thoại: {e}")
+            logger.error(f"Error getting conversation history: {e}")
             return []
 
     @timing_decorator_sync
-    def log_conversation(self, **kwargs: Any) -> Optional[Dict[str, Any]]:
-        """
-        Ghi lại hoặc cập nhật một hội thoại.
-        - Nếu có 'conversation_id', sẽ tìm và CẬP NHẬT dòng đó.
-        - Nếu không, sẽ THÊM MỚI một dòng.
-        """
+    def log_conversation(
+        self,
+        conversation_id: str,
+        recipient_id: str,
+        page_id: str,
+        originaltext: str = "",
+        originalcommand: str = "",
+        originalcontent: str = "",
+        originalattachments: Optional[List[str]] = None,
+        originalvehicle: str = "",
+        originalaction: str = "",
+        originalpurpose: str = "",
+        originalquestion: str = "",
+        systemresponse: str = "",
+        timestamp: Optional[Any] = None, # Cho phép timestamp có thể là list
+        isdone: bool = False
+    ) -> Optional[Dict[str, Any]]:
+        """Ghi lại một hội thoại, giữ nguyên logic gốc của người dùng."""
         try:
             if not self.service:
                 self.authenticate()
 
-            # Lấy toàn bộ dữ liệu từ sheet một lần
-            sheet = self.service.spreadsheets().values().get(spreadsheetId=self.sheet_id, range=SHEET_RANGE).execute()
-            values = sheet.get('values', [])
+            result = self.service.spreadsheets().values().get(
+                spreadsheetId=self.sheet_id,
+                range=SHEET_RANGE
+            ).execute()
+            values = result.get('values', [])
             header = values[0] if values else []
-            if not header:
-                logger.error("Sheet rỗng hoặc không có header.")
-                return None
-
-            conversation_id = kwargs.get('conversation_id')
-            row_index_to_update = -1
-
-            # Tìm dòng cần cập nhật nếu có conversation_id
-            if conversation_id:
-                for i, row in enumerate(values[1:], start=2): # start=2 vì sheet index bắt đầu từ 1 và bỏ qua header
-                    if row and row[0] == conversation_id:
-                        row_index_to_update = i
-                        break
-
-            # Làm phẳng và đảm bảo timestamp là một list duy nhất
-            timestamps_list = _flatten_and_unique_timestamps(kwargs.get('timestamp', []))
-            kwargs['timestamp'] = timestamps_list # Cập nhật lại kwargs
-
-            # Chuẩn bị dữ liệu cho dòng mới/cập nhật
-            row_data = [str(kwargs.get(h, '')) for h in header]
-            # Chuyển đổi các trường đặc biệt sang JSON string
-            row_data[header.index('originalattachments')] = json.dumps(kwargs.get('originalattachments', []))
-            row_data[header.index('timestamp')] = json.dumps(timestamps_list)
-            row_data[header.index('isdone')] = str(kwargs.get('isdone', False)).lower()
-
-            if row_index_to_update != -1:
-                # --- CẬP NHẬT DÒNG HIỆN CÓ ---
-                logger.info(f"Đang cập nhật conversation: {conversation_id}")
-                range_to_update = f"A{row_index_to_update}"
-                body = {'values': [row_data]}
-                self.service.spreadsheets().values().update(
-                    spreadsheetId=self.sheet_id,
-                    range=range_to_update,
-                    valueInputOption='RAW',
-                    body=body
-                ).execute()
+
+            # --- SỬA LỖI LỒNG TIMESTAMP TẠI ĐÂY ---
+            # 1. Luôn đảm bảo `timestamps_list` là một danh sách phẳng, không lồng nhau.
+            timestamps_list = _flatten_and_unique_timestamps(timestamp or [])
+
+            # 2. Xác định timestamp cho sự kiện hiện tại (dùng để kiểm tra trùng lặp).
+            #    Lấy timestamp mới nhất (cuối cùng trong list). Nếu list rỗng thì tạo mới.
+            if not timestamps_list:
+                current_ts = datetime.now().isoformat()
+                timestamps_list.append(current_ts)
             else:
-                # --- THÊM DÒNG MỚI ---
-                if not conversation_id:
-                    # Tạo ID mới nếu chưa có
-                    new_id = generate_conversation_id(
-                        kwargs.get('recipient_id', ''),
-                        kwargs.get('page_id', ''),
-                        timestamps_list[0] if timestamps_list else datetime.now().isoformat()
-                    )
-                    kwargs['conversation_id'] = new_id
-                    row_data[header.index('conversation_id')] = new_id
-                
-                logger.info(f"Đang thêm mới conversation: {kwargs['conversation_id']}")
-                body = {'values': [row_data]}
-                self.service.spreadsheets().values().append(
-                    spreadsheetId=self.sheet_id,
-                    range=SHEET_RANGE,
-                    valueInputOption='RAW',
-                    insertDataOption='INSERT_ROWS',
-                    body=body
-                ).execute()
+                current_ts = timestamps_list[-1]
+            # --- KẾT THÚC SỬA LỖI ---
+
+            # Logic kiểm tra trùng lặp của bạn được giữ nguyên
+            if header:
+                for row in values[1:]:
+                    row_data = dict(zip(header, row + [""] * (len(header) - len(row))))
+                    try:
+                        row_timestamps = json.loads(row_data.get('timestamp', '[]'))
+                        if not isinstance(row_timestamps, list):
+                            row_timestamps = [row_timestamps]
+                    except Exception:
+                        row_timestamps = []
+
+                    if str(current_ts) in [str(ts) for ts in row_timestamps] and \
+                       row_data.get('recipient_id') == str(recipient_id) and \
+                       row_data.get('page_id') == str(page_id):
+                        logger.info(f"Found duplicate conversation for user {recipient_id}, page {page_id}, timestamp {current_ts}")
+                        # Trả về dữ liệu tìm thấy, logic không đổi
+                        return {
+                            'conversation_id': row_data.get('conversation_id'),
+                            'originalcommand': row_data.get('originalcommand'),
+                            'originalcontent': row_data.get('originalcontent'),
+                            'originalattachments': json.loads(row_data.get('originalattachments', '[]')),
+                            'recipient_id': row_data.get('recipient_id'),
+                            'page_id': row_data.get('page_id'),
+                            'originaltext': row_data.get('originaltext'),
+                            'originalvehicle': row_data.get('originalvehicle'),
+                            'originalaction': row_data.get('originalaction'),
+                            'originalpurpose': row_data.get('originalpurpose'),
+                            'originalquestion': row_data.get('originalquestion'),
+                            'systemresponse': row_data.get('systemresponse'),
+                            'timestamp': row_timestamps,
+                            'isdone': str(row_data.get('isdone', 'false')).lower() == 'true'
+                        }
+
+            # Logic tạo dòng mới và append vào sheet được giữ nguyên
+            if not conversation_id:
+                conversation_id = generate_conversation_id(recipient_id, page_id, current_ts)
             
-            # Trả về dữ liệu đã được xử lý
-            return kwargs
+            new_row = [
+                conversation_id,
+                originalcommand,
+                originalcontent,
+                json.dumps(originalattachments or []),
+                recipient_id,
+                page_id,
+                originaltext,
+                originalvehicle,
+                originalaction,
+                originalpurpose,
+                originalquestion,
+                systemresponse,
+                json.dumps(timestamps_list), # Lưu danh sách đã được làm phẳng
+                str(isdone).lower()
+            ]
+
+            body = {'values': [new_row]}
+            self.service.spreadsheets().values().append(
+                spreadsheetId=self.sheet_id,
+                range=SHEET_RANGE,
+                valueInputOption='RAW',
+                body=body
+            ).execute()
 
+            logger.info(f"Thêm mới conversation: {conversation_id}")
+
+            # Trả về dữ liệu đã được ghi, logic không đổi
+            return {
+                'conversation_id': conversation_id,
+                'originalcommand': originalcommand,
+                'originalcontent': originalcontent,
+                'originalattachments': originalattachments or [],
+                'recipient_id': recipient_id,
+                'page_id': page_id,
+                'originaltext': originaltext,
+                'originalvehicle': originalvehicle,
+                'originalaction': originalaction,
+                'originalpurpose': originalpurpose,
+                'originalquestion': originalquestion,
+                'systemresponse': systemresponse,
+                'timestamp': timestamps_list, # Trả về danh sách đã được làm phẳng
+                'isdone': isdone
+            }
         except Exception as e:
-            logger.error(f"Lỗi khi ghi/cập nhật conversation: {e}", exc_info=True)
+            logger.error(f"Error logging conversation: {e}", exc_info=True)
             return None