several FontAwesome icons changed.

2024-12-23 15:11:49 +09:00
parent 31e4a50ad8
commit d850f50e25
7 changed files with 244 additions and 120 deletions
--- a/bot/utils/pdf_report.py
+++ b/bot/utils/pdf_report.py
@@ -17,22 +17,49 @@ os.makedirs(REPORTS_DIR, exist_ok=True)

 # Асинхронная функция для извлечения данных о бронировании
 def ensure_datetime(value):
-    """Преобразует строку или naive datetime в timezone-aware datetime."""
+    """
+    Преобразует строку или naive datetime в timezone-aware datetime.
+    Если значение не удается преобразовать, возвращается None.
+    """
+    if isinstance(value, datetime):
+        return make_aware(value) if is_naive(value) else value
    if isinstance(value, str):
-        value = datetime.strptime(value, '%Y-%m-%d %H:%M:%S')
-    if isinstance(value, datetime) and is_naive(value):
-        value = make_aware(value)
-    return value
+        try:
+            return make_aware(datetime.strptime(value, '%Y-%m-%d %H:%M:%S'))
+        except ValueError:
+            print(f"[WARNING] Невозможно преобразовать строку в datetime: {value}")
+    return None
+
+# @sync_to_async
+# def get_reservation_data(res):
+#     print(f"[DEBUG] Processing reservation {res.id}")
+    
+#     # Убедитесь, что даты являются timezone-aware
+#     check_in = ensure_datetime(res.check_in)
+#     check_out = ensure_datetime(res.check_out)
+    
+#     result = {
+#         "hotel_name": res.hotel.name,
+#         "pms": getattr(res.hotel, 'pms', 'N/A'),
+#         "reservation_id": res.reservation_id,
+#         "room_number": res.room_number if res.room_number else "Не указан",
+#         "room_type": res.room_type,
+#         "check_in": check_in,
+#         "check_out": check_out,
+#         "status": res.status,
+#     }
+#     # print(f"[DEBUG] Reservation data: {result}")
+#     return result

@sync_to_async
 def get_reservation_data(res):
-    print(f"[DEBUG] Processing reservation {res.id}")
-    
-    # Убедитесь, что даты являются timezone-aware
    check_in = ensure_datetime(res.check_in)
    check_out = ensure_datetime(res.check_out)
-    
-    result = {
+
+    if not check_in or not check_out:
+        raise ValueError(f"Некорректные даты бронирования: check_in={res.check_in}, check_out={res.check_out}")
+
+    return {
        "hotel_name": res.hotel.name,
        "pms": getattr(res.hotel, 'pms', 'N/A'),
        "reservation_id": res.reservation_id,
@@ -42,8 +69,6 @@ def get_reservation_data(res):
        "check_out": check_out,
        "status": res.status,
    }
-    # print(f"[DEBUG] Reservation data: {result}")
-    return result



@@ -61,27 +86,21 @@ class CustomPDF(FPDF):
        self.end_date = end_date

    def header(self):
-        """Добавление заголовка и заголовков таблицы на каждой странице."""
-        # Заголовок отчёта
-        if self.page == 1:  # Заголовок отчёта только на первой странице
+        if self.page == 1:
            self.set_font("DejaVuSans-Bold", size=14)
            self.cell(0, 10, f"Отчет о бронированиях отеля {self.hotel_name}", ln=1, align="C")
            self.ln(5)

            self.set_font("DejaVuSans", size=10)
-            self.cell(0, 10, f"за период {self.start_date.strftime('%Y-%m-%d %H:%M:%S')} - {self.end_date.strftime('%Y-%m-%d %H:%M:%S')}", ln=1, align="C")
+            self.cell(
+                0, 
+                10, 
+                f"за период {self.start_date.strftime('%Y-%m-%d %H:%M:%S')} - {self.end_date.strftime('%Y-%m-%d %H:%M:%S')}", 
+                ln=1, 
+                align="C"
+            )
            self.ln(10)

-        # Заголовки таблицы
-        self.set_font("DejaVuSans-Bold", size=8)
-        headers = ["Отель", "№ бронирования", "№ комнаты", "Тип комнаты", "Заезд", "Выезд", "Статус"]
-        col_widths = [30, 30, 30, 60, 35, 35, 30]
-        row_height = 10
-
-        for col_width, header in zip(col_widths, headers):
-            self.cell(col_width, row_height, header, border=1, align="C")
-        self.ln()
-
    def footer(self):
        """Добавление колонтитула внизу страницы."""
        self.set_y(-15)
@@ -97,13 +116,68 @@ class CustomPDF(FPDF):
            text = text[:-1]
        return text + "..." if len(text) > 3 else text

+# async def generate_pdf_report(hotel_name, reservations, start_date, end_date):
+    
+#     # Преобразование дат в timezone-aware datetime
+#     start_date = ensure_datetime(start_date)
+#     end_date = ensure_datetime(end_date)
+
+#     if not start_date or not end_date:
+#         raise ValueError(f"Некорректные периоды: start_date={start_date}, end_date={end_date}")
+
+#     # Создание экземпляра PDF с передачей параметров
+#     pdf = CustomPDF(hotel_name=hotel_name, start_date=start_date, end_date=end_date, orientation="L", unit="mm", format="A4")
+#     pdf.alias_nb_pages()
+#     pdf.add_page()  # Заголовок отчёта и таблица будут добавлены через методы header и footer
+
+#     # Таблица
+#     pdf.set_font("DejaVuSans", size=8)
+#     col_widths = [30, 30, 30, 60, 35, 35, 30]
+#     row_height = 10
+
+#     for res in reservations:
+#         try:
+#             res_data = await get_reservation_data(res)
+
+#             row_data = [
+#                 res_data["hotel_name"],
+#                 str(res_data["reservation_id"]),
+#                 res_data["room_number"],
+#                 res_data["room_type"],
+#                 res_data["check_in"].strftime('%Y-%m-%d %H:%M:%S'),
+#                 res_data["check_out"].strftime('%Y-%m-%d %H:%M:%S'),
+#                 res_data["status"],
+#             ]
+
+#             for col_width, data in zip(col_widths, row_data):
+#                 pdf.cell(col_width, row_height, data, border=1, align="C")
+#             pdf.ln()
+#         except Exception as e:
+#             print(f"pdf_report.py [ERROR] Error processing reservation {res.id}: {e}")
+
+#     # Сохранение PDF
+#     hotel_name_safe = hotel_name.replace(" ", "_").replace("/", "_")
+#     start_date_str = start_date.strftime('%Y-%m-%d')
+#     end_date_str = end_date.strftime('%Y-%m-%d')
+
+#     pdf_output_path = os.path.join(REPORTS_DIR, f"{hotel_name_safe}_report_{start_date_str}-{end_date_str}.pdf")
+#     pdf.output(pdf_output_path)
+
+#     if not os.path.exists(pdf_output_path):
+#         raise RuntimeError(f"PDF file was not created at: {pdf_output_path}")
+
+#     return pdf_output_path
+
+
 async def generate_pdf_report(hotel_name, reservations, start_date, end_date):
-    # Создание экземпляра PDF с передачей параметров
+    # Преобразование дат
+    start_date = ensure_datetime(start_date)
+    end_date = ensure_datetime(end_date)
+
    pdf = CustomPDF(hotel_name=hotel_name, start_date=start_date, end_date=end_date, orientation="L", unit="mm", format="A4")
    pdf.alias_nb_pages()
-    pdf.add_page()  # Заголовок отчёта и таблица будут добавлены через методы header и footer
+    pdf.add_page()

-    # Таблица
    pdf.set_font("DejaVuSans", size=8)
    col_widths = [30, 30, 30, 60, 35, 35, 30]
    row_height = 10
@@ -111,29 +185,36 @@ async def generate_pdf_report(hotel_name, reservations, start_date, end_date):
    for res in reservations:
        try:
            res_data = await get_reservation_data(res)
+            
+            # Отладочный вывод
+            print(f"[DEBUG] Reservation Data: {res_data}")
+            print(f"[DEBUG] check_in type: {type(res_data['check_in'])}, value: {res_data['check_in']}")
+            print(f"[DEBUG] check_out type: {type(res_data['check_out'])}, value: {res_data['check_out']}")
+
+            # Проверка и корректировка данных
+            res_data["check_in"] = ensure_datetime(res_data["check_in"])
+            res_data["check_out"] = ensure_datetime(res_data["check_out"])

            row_data = [
-                res_data["hotel_name"],
+                str(res_data["hotel_name"]),
                str(res_data["reservation_id"]),
-                res_data["room_number"],
-                res_data["room_type"],
+                str(res_data["room_number"]),
+                str(res_data["room_type"]),
                res_data["check_in"].strftime('%Y-%m-%d %H:%M:%S'),
                res_data["check_out"].strftime('%Y-%m-%d %H:%M:%S'),
-                res_data["status"],
+                str(res_data["status"]),
            ]

+
            for col_width, data in zip(col_widths, row_data):
                pdf.cell(col_width, row_height, data, border=1, align="C")
            pdf.ln()
        except Exception as e:
-            print(f"pdf_report.py [ERROR] Error processing reservation {res.id}: {e}")
+            print(f"[ERROR] Error processing reservation {res.id}: {e}")
+    
+    print("[DEBUG] PDF metadata:", vars(pdf))

-    # Сохранение PDF
-    hotel_name_safe = hotel_name.replace(" ", "_").replace("/", "_")
-    start_date_str = start_date.strftime('%Y-%m-%d')
-    end_date_str = end_date.strftime('%Y-%m-%d')
-
-    pdf_output_path = os.path.join(REPORTS_DIR, f"{hotel_name_safe}_report_{start_date_str}-{end_date_str}.pdf")
+    pdf_output_path = os.path.join(REPORTS_DIR, f"{hotel_name.replace(' ', '_')}_report_{start_date.strftime('%Y-%m-%d')}-{end_date.strftime('%Y-%m-%d')}.pdf")
    pdf.output(pdf_output_path)

    if not os.path.exists(pdf_output_path):