import time
from redis import Redis
from rq import Queue, Connection
from rq.worker import HerokuWorker as Worker
listen = ['high', 'default', 'low']
conn = Redis(host='redis', port=6379, db=0)
if __name__ == '__main__':
with Connection(conn):
worker = Worker(map(Queue, listen))
worker.work()
def connect_worker():
with Connection(conn):
worker = Worker(map(Q, listen))
worker.work()
def start_worker():
with Connection(CONN):
log.info("Starting update worker")
worker = Worker("updates")
worker.work()
port=config.REDIS_PORT,
db=0,
password=config.REDIS_PASSWORD)
logger.info('attempting to connect to database at {}'.format(
config.SQLALCHEMY_DATABASE_URI))
db = create_engine(config.SQLALCHEMY_DATABASE_URI)
Session = sessionmaker(bind=db)
# Note: no additional config required due to webhook.py
# sending required values to telegram
bot = Bot(config.TELEGRAM_TOKEN)
# Bind worker to db and redis
worker.bind(bot, Session, redis, State)
with Connection(redis):
rq_worker = Worker(map(Queue, listen))
# Set final logging things
logging.getLogger('rq.worker').setLevel(logging.WARNING)
logging.getLogger('telegram*').setLevel(logging.WARNING)
# if not config.DEVELOPMENT:
# logging.getLogger('transitions').setLevel(logging.WARNING)
logger.info('worker started')
rq_worker.work()
except redis_exceptions.ConnectionError:
logger.error('Unable to connect to redis')
sleep(10)
def start(self):
log('Starting worker')
with Connection(self._redis_client):
worker = Worker(map(Queue, listen))
worker.work()
