From 0d87c2f74e827425de4fbc52b1dbdf389b962d24 Mon Sep 17 00:00:00 2001
From: Julio Menendez <hello@juliomenendez.com>
Date: Mon, 5 Mar 2018 14:55:58 -0800
Subject: [PATCH] Initializes Celery app in CeleryEmitter.

---
 snowplow_tracker/emitters.py | 29 +++++++++++++++++------------
 1 file changed, 17 insertions(+), 12 deletions(-)

diff --git a/snowplow_tracker/emitters.py b/snowplow_tracker/emitters.py
index 2eccb0fb..2ad33a1e 100644
--- a/snowplow_tracker/emitters.py
+++ b/snowplow_tracker/emitters.py
@@ -52,16 +52,6 @@
 
 new_contract("redis", lambda x: isinstance(x, (redis.Redis, redis.StrictRedis)))
 
-try:
-    # Check whether a custom Celery configuration module named "snowplow_celery_config" exists
-    import snowplow_celery_config
-    app = Celery()
-    app.config_from_object(snowplow_celery_config)
-
-except ImportError:
-    # Otherwise configure Celery with default settings
-    app = Celery("Snowplow", broker="redis://guest@localhost//")
-
 
 class Emitter(object):
     """
@@ -172,7 +162,6 @@ def reached_limit(self):
         else:
             return self.bytes_queued >= self.byte_limit or len(self.buffer) >= self.buffer_size
 
-    @task(name="Flush")
     def flush(self):
         """
             Sends all events in the buffer to the collector.
@@ -393,16 +382,32 @@ class CeleryEmitter(Emitter):
         Works like the base Emitter class,
         but on_success and on_failure callbacks cannot be set.
     """
+    celery_app = None
+
     def __init__(self, endpoint, protocol="http", port=None, method="get", buffer_size=None, byte_limit=None):
         super(CeleryEmitter, self).__init__(endpoint, protocol, port, method, buffer_size, None, None, byte_limit)
 
+        try:
+            # Check whether a custom Celery configuration module named "snowplow_celery_config" exists
+            import snowplow_celery_config
+            self.celery_app = Celery()
+            self.celery_app.config_from_object(snowplow_celery_config)
+        except ImportError:
+            # Otherwise configure Celery with default settings
+            self.celery_app = Celery("Snowplow", broker="redis://guest@localhost//")
+
+        self.async_flush = self.celery_app.task(self.async_flush)
+
     def flush(self):
         """
             Schedules a flush task
         """
-        super(CeleryEmitter, self).flush.delay()
+        self.async_flush.delay()
         logger.info("Scheduled a Celery task to flush the event queue")
 
+    def async_flush(self):
+        super(CeleryEmitter, self).flush()
+
 
 class RedisEmitter(object):
     """