145 lines
5.8 KiB
Python
145 lines
5.8 KiB
Python
from fpdf import FPDF
|
||
|
||
import os
|
||
from datetime import datetime
|
||
from asgiref.sync import sync_to_async
|
||
from django.utils.timezone import make_aware, is_naive, is_aware
|
||
import os
|
||
|
||
# Определение абсолютного пути к папке "reports"
|
||
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
||
REPORTS_DIR = os.path.join(BASE_DIR, "reports")
|
||
|
||
# Убедитесь, что директория существует
|
||
os.makedirs(REPORTS_DIR, exist_ok=True)
|
||
|
||
|
||
|
||
# Асинхронная функция для извлечения данных о бронировании
|
||
def ensure_datetime(value):
|
||
"""Преобразует строку или naive datetime в timezone-aware datetime."""
|
||
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
|
||
|
||
@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
|
||
|
||
|
||
|
||
class CustomPDF(FPDF):
|
||
def __init__(self, hotel_name, start_date, end_date, *args, **kwargs):
|
||
super().__init__(*args, **kwargs)
|
||
self.font_folder = "bot/fonts/"
|
||
self.add_font("DejaVuSans-Bold", "", os.path.join(self.font_folder, "DejaVuSans-Bold.ttf"), uni=True)
|
||
self.add_font("DejaVuSans", "", os.path.join(self.font_folder, "DejaVuSans.ttf"), uni=True)
|
||
self.creation_date = datetime.now().strftime("%d.%m.%Y %H:%M:%S")
|
||
|
||
# Переданные параметры
|
||
self.hotel_name = hotel_name
|
||
self.start_date = start_date
|
||
self.end_date = end_date
|
||
|
||
def header(self):
|
||
"""Добавление заголовка и заголовков таблицы на каждой странице."""
|
||
# Заголовок отчёта
|
||
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.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)
|
||
self.set_font("DejaVuSans", size=8)
|
||
self.cell(60, 10, f"Copyright (C) 2024 by Touchh", align="L")
|
||
self.cell(0, 10, f"Лист {self.page_no()} из {{nb}} / Дата генерации отчета: {self.creation_date}", align="C")
|
||
|
||
|
||
|
||
def trim_text_right(self, text, max_width):
|
||
"""Обрезка текста справа."""
|
||
while self.get_string_width(text) > max_width:
|
||
text = text[:-1]
|
||
return text + "..." if len(text) > 3 else text
|
||
|
||
async def generate_pdf_report(hotel_name, reservations, start_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
|
||
|
||
|