Idle autoshutdown celery bootstep, from http://pastie.org/8629675

idle-autoshutdown.py

from celery import Celery
from celery import bootsteps
from celery.worker import consumer

class MaxIdleTimeStep(bootsteps.StartStopStep):
    idle_max = 1.0
    last_active = None
    requires = (consumer.Tasks, )

    def start(self, consumer):
        self.tref = consumer.hub.call_repeatedly(
            self.idle_max, self.check_active, consumer,
        )

    def stop(self, consumer):
        if self.tref is not None:
            self.tref.cancel()
            self.tref = None
    shutdown = stop

    def check_active(self, consumer):
        current = consumer.controller.state.all_total_count[0]
        if self.last_active is not None and self.last_active == current:
            consumer.controller.state.should_stop = True
            self.tref.cancel()
        self.last_active = current

app = Celery()
app.steps['consumer'].add(MaxIdleTimeStep)