feat: pdf generation using product barcode images
This commit is contained in:
@@ -10,8 +10,9 @@ from reportlab.pdfbase.ttfonts import TTFont
|
||||
from reportlab.pdfgen import canvas
|
||||
from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, PageBreak
|
||||
|
||||
from barcodes.pdf.pdf_maker import PdfMaker
|
||||
from barcodes.types import PdfBarcodeImageGenData, PdfBarcodeGenData
|
||||
from constants import APP_PATH
|
||||
from schemas.barcode import PdfBarcodeGenData
|
||||
|
||||
|
||||
class PDFGenerator:
|
||||
@@ -102,90 +103,94 @@ class PDFGenerator:
|
||||
bottomMargin=1 * mm
|
||||
)
|
||||
|
||||
def generate(self, barcodes_data: List[PdfBarcodeGenData]) -> BytesIO:
|
||||
def _generate_for_one_product(self, barcode_data: PdfBarcodeGenData) -> BytesIO:
|
||||
buffer = BytesIO()
|
||||
|
||||
# Создаем документ с указанным размером страницы
|
||||
doc = self._create_doc(buffer)
|
||||
|
||||
# Список элементов для добавления в документ
|
||||
# Создаем абзац с новым стилем
|
||||
paragraph = Paragraph(barcode_data['text'], self.small_style)
|
||||
|
||||
# Получаем ширину и высоту абзаца
|
||||
paragraph_width, paragraph_height = paragraph.wrap(self.page_width - 2 * mm, self.page_height)
|
||||
|
||||
# Рассчитываем доступное пространство для штрихкода
|
||||
human_readable_height = 6 * mm # Высота human-readable текста
|
||||
space_between_text_and_barcode = 4 * mm # Отступ между текстом и штрихкодом
|
||||
barcode_height = self.page_height - paragraph_height - human_readable_height - space_between_text_and_barcode - 4 * mm # Учитываем поля и отступы
|
||||
|
||||
# Создаем штрихкод
|
||||
available_width = self.page_width - 2 * mm # Учитываем поля
|
||||
|
||||
# Приблизительное количество элементов в штрихкоде Code 128 для средней длины
|
||||
num_elements = 11 * len(barcode_data['barcode_value']) # Примерная оценка: 11 элементов на символ
|
||||
|
||||
# Рассчитываем ширину штриха
|
||||
bar_width = available_width / num_elements
|
||||
barcode = code128.Code128(
|
||||
barcode_data['barcode_value'],
|
||||
barWidth=bar_width,
|
||||
barHeight=barcode_height,
|
||||
humanReadable=True
|
||||
)
|
||||
|
||||
# Добавление штрихкодов в список элементов документа
|
||||
elements = []
|
||||
product_barcode_canvases = []
|
||||
|
||||
for barcode_data in barcodes_data:
|
||||
# Создаем абзац с новым стилем
|
||||
paragraph = Paragraph(barcode_data.text, self.small_style)
|
||||
|
||||
# Получаем ширину и высоту абзаца
|
||||
paragraph_width, paragraph_height = paragraph.wrap(self.page_width - 2 * mm, self.page_height)
|
||||
|
||||
# Рассчитываем доступное пространство для штрихкода
|
||||
human_readable_height = 6 * mm # Высота human-readable текста
|
||||
space_between_text_and_barcode = 4 * mm # Отступ между текстом и штрихкодом
|
||||
barcode_height = self.page_height - paragraph_height - human_readable_height - space_between_text_and_barcode - 4 * mm # Учитываем поля и отступы
|
||||
|
||||
# Создаем штрихкод
|
||||
available_width = self.page_width - 2 * mm # Учитываем поля
|
||||
|
||||
# Приблизительное количество элементов в штрихкоде Code 128 для средней длины
|
||||
num_elements = 11 * len(barcode_data.barcode_value) # Примерная оценка: 11 элементов на символ
|
||||
|
||||
# Рассчитываем ширину штриха
|
||||
bar_width = available_width / num_elements
|
||||
barcode = code128.Code128(
|
||||
barcode_data.barcode_value,
|
||||
barWidth=bar_width,
|
||||
barHeight=barcode_height,
|
||||
humanReadable=True
|
||||
)
|
||||
product_barcode_canvases.append(barcode)
|
||||
|
||||
# Добавление штрихкодов в список элементов документа
|
||||
for _ in range(barcode_data.num_duplicates):
|
||||
elements.append(paragraph)
|
||||
elements.append(Spacer(1, space_between_text_and_barcode)) # Отступ между текстом и штрихкодом
|
||||
elements.append(PageBreak())
|
||||
|
||||
# Добавление спейсеров
|
||||
for _ in range(self.number_of_spacing_pages):
|
||||
elements.append(PageBreak())
|
||||
|
||||
# Удалить последние спейсеры
|
||||
elements = elements[:-2]
|
||||
|
||||
product_counter, product_barcodes_counter = 0, 0
|
||||
for _ in range(barcode_data['num_duplicates']):
|
||||
elements.append(paragraph)
|
||||
elements.append(Spacer(1, space_between_text_and_barcode)) # Отступ между текстом и штрихкодом
|
||||
elements.append(PageBreak())
|
||||
|
||||
# Функция для отрисовки штрихкода на canvas
|
||||
def add_barcode(canvas, doc):
|
||||
nonlocal product_barcodes_counter, product_counter
|
||||
|
||||
barcode_canvas = product_barcode_canvases[product_counter]
|
||||
|
||||
product_barcodes_counter += 1
|
||||
|
||||
# Если данная страница это спейсер, то оставить пустой
|
||||
num_duplicates = barcodes_data[product_counter].num_duplicates
|
||||
if product_barcodes_counter > num_duplicates:
|
||||
if product_barcodes_counter >= num_duplicates + self.number_of_spacing_pages:
|
||||
product_barcodes_counter = 0
|
||||
product_counter += 1
|
||||
return
|
||||
|
||||
# Отрисовка штрихкода
|
||||
barcode_width = barcode_canvas.width
|
||||
barcode_width = barcode.width
|
||||
barcode_x = (self.page_width - barcode_width) / 2 # Центрируем штрихкод
|
||||
barcode_y = human_readable_height + 2 * mm # Размещаем штрихкод снизу с учетом отступа
|
||||
barcode_canvas.drawOn(canvas, barcode_x, barcode_y)
|
||||
barcode.drawOn(canvas, barcode_x, barcode_y)
|
||||
|
||||
# Создаем документ
|
||||
doc.build(elements[:-1], onFirstPage=add_barcode, onLaterPages=add_barcode) # Убираем последний PageBreak
|
||||
doc.build(elements, onFirstPage=add_barcode, onLaterPages=add_barcode) # Убираем последний PageBreak
|
||||
|
||||
buffer.seek(0)
|
||||
return buffer
|
||||
|
||||
def _generate_for_one_product_using_img(self, barcode_data: PdfBarcodeImageGenData) -> BytesIO:
|
||||
with open(barcode_data["barcode_image_url"], 'rb') as pdf_file:
|
||||
pdf_bytes = pdf_file.read()
|
||||
|
||||
pdf_maker = PdfMaker((self.page_width, self.page_height))
|
||||
for _ in range(barcode_data['num_duplicates']):
|
||||
pdf_maker.add_pdfs(BytesIO(pdf_bytes))
|
||||
|
||||
return pdf_maker.get_bytes()
|
||||
|
||||
def _generate_spacers(self) -> BytesIO:
|
||||
buffer = BytesIO()
|
||||
doc = self._create_doc(buffer)
|
||||
elements = []
|
||||
for _ in range(self.number_of_spacing_pages):
|
||||
elements.append(PageBreak())
|
||||
doc.build(elements)
|
||||
buffer.seek(0)
|
||||
return buffer
|
||||
|
||||
def generate(self, barcodes_data: List[PdfBarcodeGenData | PdfBarcodeImageGenData]) -> BytesIO:
|
||||
pdf_maker = PdfMaker((self.page_width, self.page_height))
|
||||
|
||||
pdf_files: list[BytesIO] = []
|
||||
|
||||
for barcode_data in barcodes_data:
|
||||
if "barcode_value" in barcode_data:
|
||||
pdf_files.append(self._generate_for_one_product(barcode_data))
|
||||
else:
|
||||
pdf_files.append(self._generate_for_one_product_using_img(barcode_data))
|
||||
pdf_files.append(self._generate_spacers())
|
||||
|
||||
for file in pdf_files[:-1]:
|
||||
pdf_maker.add_pdfs(file)
|
||||
|
||||
return pdf_maker.get_bytes()
|
||||
|
||||
def generate_barcode_image(self, barcode_image_url: str, path_to_save_pdf: str):
|
||||
print(type(path_to_save_pdf))
|
||||
print(path_to_save_pdf)
|
||||
c = canvas.Canvas(path_to_save_pdf, pagesize=(self.page_width, self.page_height))
|
||||
c.drawImage(barcode_image_url, 0, 0, width=self.page_width, height=self.page_height)
|
||||
c.save()
|
||||
|
||||
Reference in New Issue
Block a user