14 lines
408 B
Python
14 lines
408 B
Python
from pyinfra.queue.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, **kwargs):
|
|
return self.queue_manager.consume(**kwargs)
|