change excel

excel/excel_translate.py

@@ -1,4 +1,4 @@
-import xlwings as xw
+import openpyxl
 from typing import Dict, List
 from translate.translator import translate_text_dict
 import math
@@ -25,24 +25,23 @@ def translate_xlsx(file_id: str, sheet_name: str = None, from_lang: str = 'en',
         temp_file.write(file_data)
         temp_file_path = temp_file.name
     
-    # Khởi tạo xlwings (chạy ẩn ứng dụng Excel)
-    app = xw.App(visible=False)
-    wb = xw.Book(temp_file_path)  # Mở workbook từ file tạm
+    # Đọc file Excel bằng openpyxl
+    wb = openpyxl.load_workbook(temp_file_path)
     
     # Chọn sheet được chỉ định hoặc tất cả các sheet
-    sheets = [wb.sheets[sheet_name]] if sheet_name else wb.sheets
+    sheets = [wb[sheet_name]] if sheet_name else wb.worksheets
 
-    for sheet in sheets:
-        last_row = sheet.used_range.rows.count
-        last_col = sheet.used_range.columns.count
+    for ws in sheets:
+        max_row = ws.max_row
+        max_col = ws.max_column
         
         # Tạo dictionary lưu trữ nội dung cần dịch và mapping từ key đến cell
         text_dict: Dict[str, List[str]] = {}
         cell_map: Dict[str, any] = {}  # lưu mapping key -> cell object
         
-        for row in range(1, last_row + 1):
-            for col in range(1, last_col + 1):
-                cell = sheet.cells[row, col]
+        for row in range(1, max_row + 1):
+            for col in range(1, max_col + 1):
+                cell = ws.cell(row=row, column=col)
                 if isinstance(cell.value, str):
                     key = f"R{row}C{col}"  # key theo dạng R{row}C{col}
                     text_dict[key] = [cell.value]  # Lưu giá trị dưới dạng danh sách với 1 phần tử
@@ -59,18 +58,16 @@ def translate_xlsx(file_id: str, sheet_name: str = None, from_lang: str = 'en',
                     cell.value = translated_text_list[0]
     
     # Lưu workbook vào file tạm thời
-    wb.save(temp_file_path)
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".xlsx") as output_file:
+        wb.save(output_file.name)
+        output_file.seek(0)
+        translated_file_id = fs_output.put(output_file.read(), filename=f"translated_{file_id}.xlsx")
+
+    # Đóng workbook và xóa file tạm
     wb.close()
-    app.quit()
-    
-    # Đọc lại file tạm để lưu vào MongoDB
-    with open(temp_file_path, "rb") as f:
-        translated_file_id = fs_output.put(f, filename=f"translated_{file_id}.xlsx")
-    
-    # Xóa file tạm
     os.remove(temp_file_path)
-    
-    print(f"Translation complete! Saved to MongoDB with file_id: {translated_file_id}")
+
+    print(f"✅ Dịch thành công! File đã lưu vào MongoDB với file_id: {translated_file_id}")
     return translated_file_id