Added support for file download

This commit is contained in:
Stanislav Hejny
2025-04-09 20:19:50 +01:00
parent 113dd38106
commit 63db3c0758
2 changed files with 6 additions and 3 deletions
+4 -1
View File
@@ -20,7 +20,6 @@ from fastapi import HTTPException
from starlette.responses import FileResponse
from amqp.adapter.data_response_factory import DataResponseFactory
from amqp.adapter.pydantic_serializer import JSONConverter
from amqp.config.amq_configuration import AMQConfiguration
from amqp.model.model import DataMessage, DataResponse
from amqp.adapter.multipart_parser import CleverMultiPartParser, process_form_data
@@ -29,6 +28,10 @@ from amqp.adapter.file_uploader import publish_file_to_rabbitmq
@dataclass
class AMQMessage(DataMessage):
"""
Expand the core DataMessage type to include the channel, which is required to create a dedicated queue
in case the response should initiate file download.
"""
def __init__(self, data_message: DataMessage, channel: AbstractRobustChannel):
# Copy all fields from DataMessage
self.magic = data_message.magic