def get_running_services():
running_services = TaskModel.select().where(
TaskModel.status == constants.STATUS_RUNNING)
l, m, h = 0, 0, 0
for service in running_services:
if service.queue_name == 'LQueue':
l += 1
elif service.queue_name == 'MQueue':
m += 1
elif service.queue_name == 'HQueue':
h += 1
return {'LQueue': l, 'MQueue': m, 'HQueue': h}
def __init__(self, queue_name, default_limit):
super().__init__()
self.running_tasks = {}
self.done_tasks = []
self.default_limit = default_limit
self.queue = PGQ(queue_name)
self.queue_name = queue_name
self.pool = QThreadPool(self)
self.pool.setMaxThreadCount(5000)
signal.signal(signal.SIGUSR1, self.__signal_handler)
self.items = []
try:
__items = TaskModel.select().where((TaskModel.status == constants.STATUS_RUNNING) & (TaskModel.queue_name == self.queue_name))
for __item in __items:
self.items.append(__item)
except OperationalError:
raise NetworkError('Network is unreachable')
