From 2e3f80a3f437687f73b41ecc1bb0d15bd0c9b6c0 Mon Sep 17 00:00:00 2001 From: nolash Date: Sun, 25 Apr 2021 14:04:32 +0200 Subject: [PATCH] Prune useless lines from graph script --- apps/contract-migration/scripts/cic_eth/paths.py | 13 +++++-------- 1 file changed, 5 insertions(+), 8 deletions(-) diff --git a/apps/contract-migration/scripts/cic_eth/paths.py b/apps/contract-migration/scripts/cic_eth/paths.py index c0e1a183..7de81f7c 100644 --- a/apps/contract-migration/scripts/cic_eth/paths.py +++ b/apps/contract-migration/scripts/cic_eth/paths.py @@ -28,8 +28,9 @@ class Fmtr(celery.utils.graph.GraphFormatter): def label(self, obj): super(Fmtr, self).label(obj) if obj != None: - logg.debug('obj {} attrs'.format(obj.id)) - return obj.queue + if obj.name == None: + raise RuntimeError('task name is not defined. Did you run celery with result_extended=True?') + return obj.name def main(): @@ -41,12 +42,8 @@ def main(): #callback_queue=args.q, ) t = api.create_account(register=False) - for e in t.collect(): - print(e[0].backend.get('celery-task-meta-{}'.format(e[0].id))) - print(e[0].name) - #t.build_graph(intermediate=True, formatter=Fmtr()).to_dot(sys.stdout)ka - #v = celery_app.control.inspect().query_task(t.id).objgraph() - #v = celery_app.control.inspect().stats() + t.get_leaf() + t.build_graph(intermediate=True, formatter=Fmtr()).to_dot(sys.stdout) if __name__ == '__main__':