import os

import redis

from flask_script import Command
from rq import Worker, Queue, Connection

listen = ["default"]

conn = redis.Redis(host=os.environ.get("REDIS_HOST"))


class RQWorker(Command):
    def run(self):
        with Connection(conn):
            worker = Worker(list(map(Queue, listen)))
            worker.work()


class RQWorkerCleaner(Command):
    def run(self):
        with Connection(conn):
            Queue(conn=conn).empty()