Fixed celery tasks

app/__init__.py

@@ -5,12 +5,11 @@ from flask_migrate import Migrate
 from flask_caching import Cache
 from celery import Celery
 
-from app.flask_celery import make_celery
-
 bootstrap = Bootstrap()
 db = SQLAlchemy()
 migrate = Migrate()
 cache = Cache()
+celery = Celery(__name__, broker='redis://localhost:6379/0')
 
 
 def create_app(config_name='development'):
@@ -23,6 +22,7 @@ def create_app(config_name='development'):
     else:
         app.config.from_object('app.config.default')
     app.config.from_envvar("OGN_CONFIG_MODULE", silent=True)
+    celery.config_from_object(app.config)
 
     # Initialize other things
     bootstrap.init_app(app)
@@ -30,8 +30,6 @@ def create_app(config_name='development'):
     migrate.init_app(app, db)
     cache.init_app(app)
 
-    # celery = make_celery(app)
-
     from app.main import bp as bp_main
     app.register_blueprint(bp_main)
 

app/flask_celery.py

@@ -1,14 +0,0 @@
-from celery import Celery
-
-
-def make_celery(app):
-    celery = Celery(app.import_name, backend=app.config["CELERY_RESULT_BACKEND"], broker=app.config["CELERY_BROKER_URL"])
-    celery.conf.update(app.config)
-
-    class ContextTask(celery.Task):
-        def __call__(self, *args, **kwargs):
-            with app.app_context():
-                return self.run(*args, **kwargs)
-
-    celery.Task = ContextTask
-    return celery

app/model/device_info_origin.py

@@ -1,5 +1,6 @@
 import enum
 
+
 class DeviceInfoOrigin(enum.Enum):
     UNKNOWN = 0
     OGN_DDB = 1

celery_worker.py

@@ -0,0 +1,7 @@
+#!/usr/bin/env python
+import os
+from app import celery, create_app
+from app.collect.celery import *
+
+app = create_app(os.getenv('FLASK_CONFIG') or 'default')
+app.app_context().push()

deployment/supervisor/celerybeatd.conf

@@ -1,6 +1,7 @@
 [program:celerybeat]
-command=/home/pi/ogn-python/venv/bin/celery -A app.collect beat -l info
+command=/home/pi/ogn-python/venv/bin/celery -A celery_worker.celery beat -l info
 directory=/home/pi/ogn-python
+environment=FLASK_APP=ogn_python.py
 
 user=pi
 numprocs=1

deployment/supervisor/celeryd.conf

@@ -1,6 +1,7 @@
 [program:celery]
-command=/home/pi/ogn-python/venv/bin/celery -A app.collect worker -l info
+command=/home/pi/ogn-python/venv/bin/celery -A celery_worker.celery worker -l info
 directory=/home/pi/ogn-python
+environment=FLASK_APP=ogn_python.py
 
 user=pi
 numprocs=1

deployment/supervisor/flower.conf

@@ -1,4 +1,5 @@
 [program:flower]
+environment=OGN_CONFIG_MODULE='config/default.py'
 command=/home/pi/ogn-python/venv/bin/celery flower -A app.celery --port=5555 -l info
 directory=/home/pi/ogn-python
 

deployment/supervisor/gunicorn.conf

@@ -1,4 +1,5 @@
 [program:gunicorn]
+environment=OGN_CONFIG_MODULE='config/default.py'
 command=/home/pi/ogn-python/venv/bin/gunicorn --bind :5000 ogn_python:app
 directory=/home/pi/ogn-python
 

deployment/supervisor/ogn-feeder.conf

@@ -1,6 +1,7 @@
 [program:ogn-feeder]
-command=/home/pi/ogn-python/venv/bin/python -m flask gateway run
+command=/home/pi/ogn-python/venv/bin/flask gateway run
 directory=/home/pi/ogn-python
+environment=FLASK_APP=ogn_python.py
 
 user=pi
 stderr_logfile=/var/log/supervisor/ogn-feeder.log