Update src/firestore_db.py

src/firestore_db.py

@@ -1,3 +1,4 @@
+# File: src/firestore_db.py (đã chỉnh sửa)
 from firebase_admin import credentials, firestore, initialize_app
 from config.settings import Settings
 from src.logger import logger
@@ -26,10 +27,26 @@ class FirestoreDB:
         return [(doc.to_dict()["text"], doc.to_dict()["vector"]) for doc in self.documents.stream()]
 
     def save_lottery(self, data):
-        # Tạo key duy nhất với timestamp để tránh ghi đè
-        key = f"{data['ngay']}_{data['dai']}_{datetime.now().strftime('%Y%m%d%H%M%S%f')}"
-        self.lottery.document(key).set(data)
-        logger.info(f"Đã lưu lottery: {key}")
+        # Kiểm tra trùng lặp dựa trên ngay và dai
+        ngay = data.get("ngay")
+        dai = data.get("dai")
+        if not all([ngay, dai]):
+            logger.error("Dữ liệu thiếu ngay hoặc dai")
+            return False
+
+        # Tạo query để kiểm tra document có cùng ngay và dai
+        query = self.lottery.where("ngay", "==", ngay).where("dai", "==", dai).limit(1).get()
+        exists = any(doc.exists for doc in query)
+
+        if not exists:
+            # Tạo key duy nhất nếu chưa tồn tại
+            key = f"{ngay}_{dai}_{datetime.now().strftime('%Y%m%d%H%M%S%f')}"
+            self.lottery.document(key).set(data)
+            logger.info(f"Đã lưu lottery: {key}")
+            return True
+        else:
+            logger.warning(f"Document với ngay={ngay} và dai={dai} đã tồn tại, bỏ qua lưu.")
+            return False
 
     def lottery_exists(self, key):
         return bool(self.lottery.document(key).get().exists)
@@ -42,7 +59,6 @@ class FirestoreDB:
                 .stream()]
 
     def get_lottery_by_date(self, dai, date):
-        # Tìm document gần đúng với ngày và đài
         docs = [doc.to_dict() for doc in self.lottery
                 .where("dai", "==", dai)
                 .where("ngay", "==", date)