From e5b5f02bd6a79aab6d3d458a3c2237e45b0c5bc1 Mon Sep 17 00:00:00 2001 From: Dan Urbano Date: Tue, 30 Mar 2021 09:32:32 -0700 Subject: [PATCH] Converts `shared_task` to a factory to mimick `celery.shared_task` API. Only needed when `celery` not installed. --- influxdb_metrics/tasks.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/influxdb_metrics/tasks.py b/influxdb_metrics/tasks.py index 9d3682a..759abcf 100644 --- a/influxdb_metrics/tasks.py +++ b/influxdb_metrics/tasks.py @@ -4,10 +4,12 @@ try: from celery import shared_task except ImportError: - def shared_task(func): - def wrapper(*args, **kwargs): - return func(*args, **kwargs) - return wrapper + def shared_task(*args, **kwargs): + def decorator(func): + def wrapper(*args, **kwargs): + return func(*args, **kwargs) + return wrapper + return decorator from .utils import write_points as write_points_normal