reports converted to FPDF2 framework. Fully functional
This commit is contained in:
@@ -1,65 +1,144 @@
|
||||
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")
|
||||
|
||||
def generate_pdf_report(hotel_name, reservations, start_date, end_date):
|
||||
"""Генерация PDF отчета."""
|
||||
pdf = FPDF(orientation='L', unit="mm", format="A4")
|
||||
pdf.add_page()
|
||||
# Убедитесь, что директория существует
|
||||
os.makedirs(REPORTS_DIR, exist_ok=True)
|
||||
|
||||
# Укажите путь к шрифту
|
||||
font_path = os.path.join("bot", "fonts", "OpenSans-Regular.ttf")
|
||||
if not os.path.exists(font_path):
|
||||
raise FileNotFoundError(f"Шрифт {font_path} не найден. Убедитесь, что он находится в указанной папке.")
|
||||
|
||||
pdf.add_font("OpenSans", "", font_path, uni=True)
|
||||
pdf.set_font("OpenSans", size=12)
|
||||
|
||||
# Заголовки
|
||||
title = f"Отчет по заселениям: {hotel_name}"
|
||||
period = (
|
||||
f"Период: {start_date.strftime('%d.%m.%Y')} - {end_date.strftime('%d.%m.%Y')}"
|
||||
if start_date and end_date
|
||||
else "Период: Все время"
|
||||
)
|
||||
# Асинхронная функция для извлечения данных о бронировании
|
||||
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
|
||||
|
||||
pdf.cell(0, 10, txt=title, ln=True, align="C")
|
||||
pdf.cell(0, 10, txt=period, ln=True, align="C")
|
||||
pdf.ln(10)
|
||||
@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
|
||||
|
||||
# Ширины колонок
|
||||
page_width = pdf.w - 10
|
||||
col_widths = [page_width * 0.2, page_width * 0.2, page_width * 0.15, page_width * 0.25, page_width * 0.1, page_width * 0.1]
|
||||
|
||||
# Заголовки таблицы
|
||||
pdf.set_font("OpenSans", size=10)
|
||||
headers = ["Дата заезда", "Дата выезда", "Номер", "Тип комнаты", "Цена"]
|
||||
for width, header in zip(col_widths, headers):
|
||||
pdf.cell(width, 15, header, border=1, align="C")
|
||||
pdf.ln()
|
||||
|
||||
total_price = 0
|
||||
total_discount = 0
|
||||
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:
|
||||
price = res.price or 0
|
||||
discount = res.discount or 0
|
||||
total_price += price
|
||||
total_discount += discount
|
||||
try:
|
||||
res_data = await get_reservation_data(res)
|
||||
|
||||
pdf.cell(col_widths[0], 10, res.check_in.strftime('%d.%m.%Y %H:%M'), border=1)
|
||||
pdf.cell(col_widths[1], 10, res.check_out.strftime('%d.%m.%Y %H:%M'), border=1)
|
||||
pdf.cell(col_widths[2], 10, res.room_number, border=1)
|
||||
pdf.cell(col_widths[3], 10, res.room_type, border=1)
|
||||
pdf.cell(col_widths[4], 10, f"{price:.2f} р.", border=1, align="R")
|
||||
pdf.ln()
|
||||
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
|
||||
|
||||
pdf.ln(5)
|
||||
pdf.set_font("OpenSans", size=18)
|
||||
pdf.cell(0, 10, "Итоги:", ln=True)
|
||||
pdf.cell(0, 10, f"Общая сумма цен: {total_price:.2f} руб.", ln=True)
|
||||
|
||||
file_path = os.path.join("reports", f"{hotel_name}_report.pdf")
|
||||
os.makedirs(os.path.dirname(file_path), exist_ok=True)
|
||||
pdf.output(file_path)
|
||||
return file_path
|
||||
|
||||
Reference in New Issue
Block a user