From 3fc7dcb3452b615bdb9b5753b6db3cbe030bdbc9 Mon Sep 17 00:00:00 2001
From: Stefano Alberto Russo <stefano.russo@gmail.com>
Date: Wed, 1 Dec 2021 17:48:46 +0100
Subject: [PATCH] Fixed potential bug with the agent API.
---
services/webapp/code/rosetta/core_app/api.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/services/webapp/code/rosetta/core_app/api.py b/services/webapp/code/rosetta/core_app/api.py
index 5380295..0164a4c 100644
--- a/services/webapp/code/rosetta/core_app/api.py
+++ b/services/webapp/code/rosetta/core_app/api.py
@@ -292,9 +292,10 @@ logger.info(' - ports: "{},{},{}"'.format(port, port+1, port+2))
response = urlopen("'''+webapp_conn_string+'''/api/v1/base/agent/?task_uuid={}&action=set_ip_port&ip={}&port={}".format(task_uuid, ip, port))
response_content = response.read()
-if response_content != 'OK':
+if response_content not in ['OK', b'OK']:
logger.error(response_content)
logger.info('Not everything OK, exiting with status code =1')
+ import sys
sys.exit(1)
else:
logger.info('Everything OK')
--
GitLab