Python 2.7x and 3.xx compatibility fixes
Change-Id: I84eb3bf7691fa867acadf9dae8c4f56a9781bf73
Signed-off-by: George Keishing <gkeishin@in.ibm.com>
diff --git a/lib/code_update_utils.py b/lib/code_update_utils.py
index cdb2ac4..16487db 100644
--- a/lib/code_update_utils.py
+++ b/lib/code_update_utils.py
@@ -155,11 +155,13 @@
for member in tar.getmembers():
f = tar.extractfile(member)
content = f.read()
- if "version=" in content:
- content = content.split("\n")
- content = [x for x in content if "version=" in x]
- version = content[0].split("=")[-1]
- break
+ if content.find(b"version=") == -1:
+ # This tar member does not contain the version.
+ continue
+ content = content.decode("utf-8").split("\n")
+ content = [x for x in content if "version=" in x]
+ version = content[0].split("=")[-1]
+ break
tar.close()
return version