def _render_GET_thread(self, request: Request) -> bytes:
""" GET request /graphviz/full.{format}
Returns the rendered graph file
"""
set_cors(request, 'GET')
tx_storage = self.manager.tx_storage
graphviz = GraphvizVisualizer(tx_storage)
if b'weight' in request.args:
graphviz.show_weight = self.parse_bool_arg(
request.args[b'weight'][0].decode('utf-8'))
if b'acc_weight' in request.args:
graphviz.show_acc_weight = self.parse_bool_arg(
request.args[b'acc_weight'][0].decode('utf-8'))
if b'verifications' in request.args:
graphviz.include_verifications = self.parse_bool_arg(
request.args[b'verifications'][0].decode('utf-8'))
if b'funds' in request.args:
graphviz.include_funds = self.parse_bool_arg(
request.args[b'funds'][0].decode('utf-8'))
if b'only_blocks' in request.args:
graphviz.only_blocks = self.parse_bool_arg(
request.args[b'only_blocks'][0].decode('utf-8'))
dot = graphviz.dot(format=self.format.dot)
request.setHeader(b'content-type', self.format.content_type)
if self.format is FileFormat.DOT:
return str(dot).encode('utf-8')
return dot.pipe()
def on_new_tx(self, key: HathorEvents, args: EventArguments) -> None:
""" This method is called every change in the DAG. It saves a new snapshot in disk.
"""
if not self.is_running:
return
n = self.sequence
tx_storage = self.manager.tx_storage
graphviz = GraphvizVisualizer(tx_storage)
graphviz.include_verifications = True
graphviz.include_funds = True
dot = graphviz.dot(format=self.format)
dot.render(os.path.join(self.dirname, 'seq_{:010d}'.format(n)))
self.sequence += 1
def _render_GET_thread(self, request: Request) -> bytes:
""" GET request /graphviz/
Expects 'format' parameter in request to set the content-type of the graph
Format options are 'pdf', 'png' and 'jpg'. Default format is 'pdf'
Returns the file
"""
set_cors(request, 'GET')
contenttype = {
'pdf': b'application/pdf',
'png': b'image/png',
'jpg': b'image/jpeg',
'dot': b'application/dot',
}
dotformat = 'pdf'
if b'format' in request.args:
dotformat = request.args[b'format'][0].decode('utf-8')
tx_storage = self.manager.tx_storage
if b'tx' in request.args:
# Getting tx neightborhood
tx_hex = request.args[b'tx'][0].decode('utf-8')
success, message = validate_tx_hash(tx_hex, tx_storage)
if not success:
return json.dumps({
'success': False,
'message': message
},
indent=4).encode('utf-8')
else:
graph_type = request.args[b'graph_type'][0].decode('utf-8')
max_level = int(request.args[b'max_level'][0])
if max_level > settings.MAX_GRAPH_LEVEL:
return json.dumps(
{
'success':
False,
'message':
'Graph max level is {}'.format(
settings.MAX_GRAPH_LEVEL)
},
indent=4).encode('utf-8')
tx = tx_storage.get_transaction(bytes.fromhex(tx_hex))
graphviz = GraphvizVisualizer(tx_storage)
dot = graphviz.tx_neighborhood(tx,
format=dotformat,
max_level=max_level,
graph_type=graph_type)
else:
weight = False
if b'weight' in request.args:
weight = self.parseBoolArg(
request.args[b'weight'][0].decode('utf-8'))
acc_weight = False
if b'acc_weight' in request.args:
acc_weight = self.parseBoolArg(
request.args[b'acc_weight'][0].decode('utf-8'))
include_verifications = True
if b'verifications' in request.args:
include_verifications = self.parseBoolArg(
request.args[b'verifications'][0].decode('utf-8'))
include_funds = False
if b'funds' in request.args:
include_funds = self.parseBoolArg(
request.args[b'funds'][0].decode('utf-8'))
only_blocks = False
if b'only_blocks' in request.args:
only_blocks = self.parseBoolArg(
request.args[b'only_blocks'][0].decode('utf-8'))
graphviz = GraphvizVisualizer(tx_storage)
graphviz.include_verifications = include_verifications
graphviz.include_funds = include_funds
graphviz.only_blocks = only_blocks
graphviz.show_weight = weight
graphviz.show_acc_weight = acc_weight
dot = graphviz.dot(format=dotformat)
if dotformat == 'dot':
request.setHeader(b'content-type', contenttype[dotformat])
return str(dot).encode('utf-8')
request.setHeader(b'content-type', contenttype[dotformat])
return dot.pipe()