From 413d8755cc4cb626afd9b58dbe3f4f9dfeb1a7a7 Mon Sep 17 00:00:00 2001
From: AlexSserb <aserbin190@gmail.com>
Date: Sat, 28 Sep 2024 20:16:53 +0400
Subject: [PATCH] fix: response schema for deal barcodes pdf end-point

---
 routers/deal.py    | 8 ++++----
 schemas/barcode.py | 7 +++++++
 2 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/routers/deal.py b/routers/deal.py
index 5c80633..9de0c0d 100644
--- a/routers/deal.py
+++ b/routers/deal.py
@@ -8,7 +8,7 @@ from sqlalchemy.ext.asyncio import AsyncSession
 from backend.dependecies import SessionDependency, CurrentUserDependency
 from backend.session import get_session
 from models import User
-from schemas.barcode import GetProductBarcodePdfResponse, GetDealProductsBarcodesPdfRequest
+from schemas.barcode import GetDealProductsBarcodesPdfRequest, GetDealProductsBarcodesPdfResponse
 from schemas.deal import *
 from services.auth import get_current_user, authorized_user, guest_user
 from services.barcode import BarcodeService
@@ -398,8 +398,8 @@ async def add_kit_to_deal_product(
 
 @deal_router.post(
     '/barcodes/get-pdf',
-    operation_id='get_deal_barcodes_pdf',
-    response_model=GetProductBarcodePdfResponse
+    operation_id='get_deal_products_barcodes_pdf',
+    response_model=GetDealProductsBarcodesPdfResponse
 )
 async def get_deal_products_barcodes_pdf(
         request: GetDealProductsBarcodesPdfRequest,
@@ -408,7 +408,7 @@ async def get_deal_products_barcodes_pdf(
     filename, pdf_buffer = await BarcodeService(session).get_deal_barcodes_pdf(request)
     pdf_buffer: BytesIO
     base64_string = base64.b64encode(pdf_buffer.read()).decode('utf-8')
-    return GetProductBarcodePdfResponse(
+    return GetDealProductsBarcodesPdfResponse(
         base64_string=base64_string,
         filename=filename,
         mime_type='application/pdf'
diff --git a/schemas/barcode.py b/schemas/barcode.py
index 4f904ae..b886abb 100644
--- a/schemas/barcode.py
+++ b/schemas/barcode.py
@@ -135,4 +135,11 @@ class GetProductBarcodePdfResponse(BaseSchema):
     base64_string: str
     filename: str
     mime_type: str
+
+
+class GetDealProductsBarcodesPdfResponse(BaseSchema):
+    base64_string: str
+    filename: str
+    mime_type: str
+
 # endregion