add sentry support to celery tasks too

This commit is contained in:
codl 2017-08-07 13:46:05 +02:00
parent a1b9cb4311
commit a7f68fde9a
No known key found for this signature in database
GPG Key ID: 6CD7C8891ED1233A
1 changed files with 7 additions and 0 deletions

View File

@ -23,6 +23,13 @@ app.conf.task_default_queue = 'default'
app.conf.task_default_exchange = 'celery'
app.conf.task_default_exchange_type = 'direct'
if 'SENTRY_DSN' in flaskapp.config:
from raven import Client
from raven.contrib.celery import register_signal, register_logger_signal
sentry = Client(flaskapp.config['SENTRY_DSN'])
register_logger_signal(sentry)
register_signal(sentry)
@app.task(autoretry_for=(TwitterError, URLError))
def fetch_acc(id, cursor=None):
acc = Account.query.get(id)