import yaml
import psutil

def locate_celery_process():
    for proc in psutil.process_iter():
        if "celery" in proc.name():
            return "ok"

def generate_alert():
    celery = locate_celery_process()
    if celery == "ok":
        return [0, "OK - Celery is running"]
    else:
        return [2, "CRITICAL - Celery is not running"]

def __run__(params):
    t = generate_alert()
    return t