From 51a459cbf04e9884cf6b7c2c3145206ecf1a0ffb Mon Sep 17 00:00:00 2001 From: Matthias Bisping Date: Thu, 21 Apr 2022 13:03:48 +0200 Subject: [PATCH] set prefetch count to 1; removed obsolete imports --- pyinfra/queue/queue_manager/pika_queue_manager.py | 1 + pyinfra/storage/adapters/azure.py | 4 +--- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/pyinfra/queue/queue_manager/pika_queue_manager.py b/pyinfra/queue/queue_manager/pika_queue_manager.py index 4346658..8bc70d4 100644 --- a/pyinfra/queue/queue_manager/pika_queue_manager.py +++ b/pyinfra/queue/queue_manager/pika_queue_manager.py @@ -72,6 +72,7 @@ class PikaQueueManager(QueueManager): self.connection = pika.BlockingConnection(parameters=connection_params) self.channel = self.connection.channel() + self.channel.basic_qos(prefetch_count=1) if not dead_letter_queue: dead_letter_queue = CONFIG.rabbitmq.queues.dead_letter diff --git a/pyinfra/storage/adapters/azure.py b/pyinfra/storage/adapters/azure.py index aa28279..334dbaf 100644 --- a/pyinfra/storage/adapters/azure.py +++ b/pyinfra/storage/adapters/azure.py @@ -1,10 +1,8 @@ -from itertools import repeat import logging -from functools import wraps +from itertools import repeat from operator import attrgetter from azure.storage.blob import ContainerClient, BlobServiceClient -from retry import retry from pyinfra.storage.adapters.adapter import StorageAdapter