Commit b3768d71 authored by Leo Singer's avatar Leo Singer
Browse files

WIP

parent c66ba43c
Loading
Loading
Loading
Loading
Loading
+9 −0
Original line number Diff line number Diff line
"""Communication with GraceDB."""
import functools
import random

from ligo.gracedb import rest
from celery.utils.log import get_task_logger
@@ -44,6 +45,14 @@ def task(*args, **kwargs):
                    retry_kwargs=dict(max_retries=10))


@task
@catch_retryable_http_errors
def fail_randomly():
    if random.randint(0, 1) == 0:
        raise rest.HTTPError(404, 'Not Found', b'"File p_astro.json not found"')
    return 'foobar'


@task(shared=False)
@catch_retryable_http_errors
def create_event(filecontents, search, pipeline, group):
+6 −0
Original line number Diff line number Diff line
@@ -202,6 +202,12 @@ def handle_cbc_event(alert):
        ).delay()


@app.task(shared=False, autoretry_for=(HTTPError, URLError, TimeoutError),
          default_retry_delay=2.0, retry_kwargs=dict(max_retries=2))
def _fail_randomly():
    return gracedb.fail_randomly._orig_run()


@app.task(autoretry_for=(HTTPError, URLError, TimeoutError),
          default_retry_delay=20.0, retry_backoff=True,
          retry_kwargs=dict(max_retries=500), shared=False)