current_user.update_lat() def make_celery(app=None): app = app or create_app()[0] celery.conf.update(app.config) ext = FlaskCeleryExt(app) return ext.celery # Health-check health = HealthCheck(current_app, "/health-check") health.add_check(health_check_celery) health.add_check(health_check_db) with current_app.app_context(): current_app.config['MIGRATION_STATUS'] = check_migrations() health.add_check(health_check_migrations) # http://stackoverflow.com/questions/9824172/find-out-whether-celery-task-exists @after_task_publish.connect def update_sent_state(sender=None, headers=None, **kwargs): # the task may not exist if sent using `send_task` which # sends tasks by name, so fall back to the default result backend # if that is the case. task = celery.tasks.get(sender) backend = task.backend if task else celery.backend backend.store_result(headers['id'], None, 'WAITING') # register celery tasks. removing them will cause the tasks to not function. so don't remove them
def test_migrations(self): """Method to test the database migrations""" with self.app.test_request_context(): result = check_migrations().split(',') assert result[0] == 'success'
with app.app_context(): return task_base.__call__(self, *args, **kwargs) celery.Task = ContextTask return celery celery = make_celery(current_app) # Health-check health = HealthCheck(current_app, "/health-check") envdump = EnvironmentDump(current_app, "/environment", include_config=False) health.add_check(health_check_celery) health.add_check(health_check_db) with current_app.app_context(): current_app.config['MIGRATION_STATUS'] = check_migrations() health.add_check(health_check_migrations) # http://stackoverflow.com/questions/9824172/find-out-whether-celery-task-exists @after_task_publish.connect def update_sent_state(sender=None, body=None, **kwargs): # the task may not exist if sent using `send_task` which # sends tasks by name, so fall back to the default result backend # if that is the case. task = celery.tasks.get(sender) backend = task.backend if task else celery.backend backend.store_result(body['id'], None, 'WAITING') # register celery tasks. removing them will cause the tasks to not function. so don't remove them
def test_migrations(self): """Method to test the database migrations""" with app.test_request_context(): result = check_migrations().split(',') self.assertEqual(result[0], 'success')
def test_migrations(self): with app.test_request_context(): result = check_migrations().split(',') self.assertEqual(result[0], 'success')