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')