fix: uploading barcode pdf files instead of images
This commit is contained in:
		@@ -1,7 +1,11 @@
 | 
			
		||||
from io import BytesIO
 | 
			
		||||
from typing import BinaryIO
 | 
			
		||||
 | 
			
		||||
from fpdf import FPDF
 | 
			
		||||
import pdfrw
 | 
			
		||||
from pdfrw import PdfReader
 | 
			
		||||
from pdfrw.objects.pdfdict import PdfDict
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class PdfMaker:
 | 
			
		||||
    def __init__(self, size: tuple):
 | 
			
		||||
@@ -36,3 +40,29 @@ class PdfMaker:
 | 
			
		||||
        self.writer.write(result_io)
 | 
			
		||||
        result_io.seek(0)
 | 
			
		||||
        return result_io
 | 
			
		||||
 | 
			
		||||
    @staticmethod
 | 
			
		||||
    def check_is_correct_aspects_ratio(file: BinaryIO):
 | 
			
		||||
        pdf = PdfReader(file)
 | 
			
		||||
        allowed_aspects_ratio = 1.45
 | 
			
		||||
        page: PdfDict
 | 
			
		||||
 | 
			
		||||
        try:
 | 
			
		||||
            page: PdfDict = pdf.getPage(0)
 | 
			
		||||
        except IndexError:
 | 
			
		||||
            raise Exception("Ошибка. В документе нет страниц.")
 | 
			
		||||
 | 
			
		||||
        try:
 | 
			
		||||
            pdf.getPage(1)
 | 
			
		||||
            raise Exception("Ошибка. В документе должна быть только одна страница.")
 | 
			
		||||
        except IndexError:
 | 
			
		||||
            pass
 | 
			
		||||
 | 
			
		||||
        media_box = page.MediaBox
 | 
			
		||||
        width = float(media_box[2]) - float(media_box[0])
 | 
			
		||||
        height = float(media_box[3]) - float(media_box[1])
 | 
			
		||||
 | 
			
		||||
        aspect_ratio = width / height
 | 
			
		||||
        if abs(aspect_ratio - allowed_aspects_ratio) > 0.01:
 | 
			
		||||
            raise Exception("Ошибка. Страница документа должна быть размером 58х40.")
 | 
			
		||||
        file.seek(0)
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user