Skip to content

queue_service

django_spire.celery.services.queue_service

CeleryTaskQueueService

Bases: BaseDjangoModelService['CeleryTask']

Source code in django_spire/contrib/constructor/constructor.py
def __init__(self, obj: Any = None):
    self._obj_type_name: str = str(next(iter(self.__class__.__annotations__.values()))).split(
        '.'
    )[-1]

    if obj is None:
        return

    self._obj_mro_type_names = [cls.__name__ for cls in obj.__class__.__mro__]

    if self._obj_type_name not in self._obj_mro_type_names:
        message = f'{self.__class__.__name__} was instantiated with obj type "{obj.__class__.__name__}" and failed as it was expecting "{self._obj_type_name}".'
        raise ConstructorError(message)

    self._obj_type: type[TypeAny] = obj.__class__

    if self._obj_type is None or self._obj_type is ...:
        message = f'{self.__class__.__name__} top class attribute must have an annotated type.'
        raise ConstructorError(message)

    self.obj: TypeAny = obj

    if ABC not in self.__class__.__bases__:
        if not self._obj_is_valid:
            message = f'{self._obj_type_name} failed to validate on {self.__class__.__name__}'
            raise ConstructorError(message)

    self.__post_init__()

obj instance-attribute

get_estimated_queue_size

Source code in django_spire/celery/services/queue_service.py
def get_estimated_queue_size(self) -> int:
    return self.obj_class.objects.by_unready().count()

get_estimated_queue_time_seconds

Source code in django_spire/celery/services/queue_service.py
def get_estimated_queue_time_seconds(self) -> int:
    celery_tasks = self.obj_class.objects.by_completed()[:100]
    queue_times_seconds = [celery_task.queue_time_seconds for celery_task in celery_tasks]

    return int(sum(queue_times_seconds) / len(queue_times_seconds))