mirror of
https://github.com/alicevision/Meshroom.git
synced 2025-04-29 02:08:08 +02:00
51a9b0e316
* store Nodes and Attributes in Models * expose key attributes/methods as Properties/Slots * update command line scripts
46 lines
1.5 KiB
Python
Executable file
46 lines
1.5 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
import argparse
|
|
import os
|
|
from pprint import pprint
|
|
|
|
from meshroom.processGraph import graph as pg
|
|
|
|
parser = argparse.ArgumentParser(description='Query the status of nodes in a Graph of processes.')
|
|
parser.add_argument('graphFile', metavar='GRAPHFILE.mg', type=str,
|
|
help='Filepath to a graph file.')
|
|
parser.add_argument('--node', metavar='NODE_NAME', type=str,
|
|
help='Process the node alone.')
|
|
parser.add_argument('--graph', metavar='NODE_NAME', type=str,
|
|
help='Process the node and all previous nodes needed.')
|
|
parser.add_argument("--verbose", help="Print full status information",
|
|
action="store_true")
|
|
|
|
args = parser.parse_args()
|
|
|
|
if not os.path.exists(args.graphFile):
|
|
print('ERROR: No graph file "{}".'.format(args.node, args.graphFile))
|
|
exit(-1)
|
|
|
|
graph = pg.loadGraph(args.graphFile)
|
|
|
|
graph.update()
|
|
|
|
if args.node:
|
|
node = graph.node(args.node)
|
|
if node is None:
|
|
print('ERROR: node "{}" does not exist in file "{}".'.format(args.node, args.graphFile))
|
|
exit(-1)
|
|
print('{}: {}'.format(node.name, node.status.status.name))
|
|
if args.verbose:
|
|
print('statusFile: ', node.statusFile())
|
|
pprint(node.status.toDict())
|
|
else:
|
|
startNodes = None
|
|
if args.graph:
|
|
startNodes = [graph.nodes(args.graph)]
|
|
nodes = graph.dfsNodesOnFinish(startNodes=startNodes)
|
|
for node in nodes:
|
|
print('{}: {}'.format(node.name, node.status.status.name))
|
|
if args.verbose:
|
|
pprint([n.status.toDict() for n in nodes])
|
|
|