CSIT Trending: Input files

[csit.git] / resources / tools / presentation / input_data_files.py

diff --git a/resources/tools/presentation/input_data_files.py b/resources/tools/presentation/input_data_files.py
index 439c0d8..b45d21a 100644 (file)
--- a/resources/tools/presentation/input_data_files.py
+++ b/resources/tools/presentation/input_data_files.py
@@ -81,6 +81,7 @@ def download_data_files(spec):
             try:
                 response = get(url, stream=True)
                 code = response.status_code
+
                 # temporary workaround, remove when output.log.xml is not needed
                 if code != codes["OK"] and \
                         spec.input["file-name"].endswith(".gz"):
@@ -98,7 +99,11 @@ def download_data_files(spec):
                         nexus_file_name = "{job}{sep}{build}{sep}{name}".\
                             format(job=job, sep=SEPARATOR, build=build["build"],
                                    name=file_name)
-                        url = "{url}/rls{release}/{dir}/{file}".\
+                        try:
+                            release = "rls".format(int(release))
+                        except ValueError:
+                            pass
+                        url = "{url}/{release}/{dir}/{file}".\
                             format(url=spec.environment["urls"]["URL[NEXUS]"],
                                    release=release,
                                    dir=spec.environment["urls"]["DIR[NEXUS]"],
@@ -148,7 +153,7 @@ def download_data_files(spec):
                 elif spec.input["file-name"].endswith(".gz"):
                     rename(new_name, new_name[:-3])
                     with open(new_name[:-3], 'r') as xml_file:
-                        with gzip.open(new_name, 'wb') as gz_file:
+                        with gzip.open(new_name, 'w') as gz_file:
                             gz_file.write(xml_file.read())
                     new_name = new_name[:-3]
                     status = "downloaded"