14 lines
395 B
Python
14 lines
395 B
Python
from pyinfra.pyampq.queue_manager.queue_manager import QueueManager
|
|
|
|
|
|
class Consumer:
|
|
def __init__(self, callback, queue_manager: QueueManager):
|
|
self.queue_manager = queue_manager
|
|
self.callback = callback
|
|
|
|
def consume_and_publish(self):
|
|
self.queue_manager.consume_and_publish(self.callback)
|
|
|
|
def consume(self):
|
|
yield from self.queue_manager.consume()
|