def __init__(self, name): self.redis = StrictRedis( port=Config.redis_port(), host=Config.redis_host(), ) self._name = name
node_id_attr, run_id, x_consumer_id, ) self.runs.update_one( filter={ 'id': run_id, 'X-Consumer-ID': x_consumer_id, }, update={ '$set': { 'done': True, 'subgraph_ids': subgraph_ids }, }, ) REDIS_URI = 'redis://{}:{}/1'.format(Config.redis_host(), Config.redis_port()) celery = Celery('main', broker=REDIS_URI) @celery.task(name='find-subgraph') def find_subgraphs(run_id): runner = Runner() runner.run(run_id) if __name__ == '__main__': celery.start()