Merge "[VVP] Tox support for installing Win pre-built libraries"

1 files changed, 96 insertions, 0 deletions

diff --git a/install_win_deps.py b/install_win_deps.py
new file mode 100644
index 0000000..01e9503
--- /dev/null
+++ b/install_win_deps.py
@@ -0,0 +1,96 @@
+# ============LICENSE_START====================================================
+# org.onap.vvp/validation-scripts
+# ===================================================================
+# Copyright © 2019 AT&T Intellectual Property. All rights reserved.
+# ===================================================================
+#
+# Unless otherwise specified, all software contained herein is licensed
+# under the Apache License, Version 2.0 (the "License");
+# you may not use this software except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#             http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+#
+#
+# Unless otherwise specified, all documentation contained herein is licensed
+# under the Creative Commons License, Attribution 4.0 Intl. (the "License");
+# you may not use this documentation except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#             https://creativecommons.org/licenses/by/4.0/
+#
+# Unless required by applicable law or agreed to in writing, documentation
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+# ============LICENSE_END============================================
+#
+import os
+import platform
+import subprocess #nosec
+import sys
+import tempfile
+from urllib import request
+
+PREBUILT_DOWNLOAD_SITE = "https://download.lfd.uci.edu/pythonlibs/n5jyqt7p/"
+PREBUILT_WIN_LIBS = [
+    "yappi-1.0-cp{python_version}-cp{python_version}m-{arch}.whl",
+    "setproctitle-1.1.10-cp{python_version}-cp{python_version}m-{arch}.whl"
+]
+
+
+def is_windows():
+    return os.name == 'nt'
+
+
+def python_version():
+    return sys.version[:3].replace(".", "")
+
+
+def system_architecture():
+    arch = platform.architecture()
+    return "win32" if arch[0] != "64bit" else "win_amd64"
+
+
+def download_url(url):
+    resp = request.urlopen(url) #nosec
+    return resp.read()
+
+
+def read_file(path):
+    with open(path, "r") as f:
+        return f.read()
+
+
+def write_file(data, path, mode="w"):
+    with open(path, mode) as f:
+        f.write(data)
+
+
+def install_prebuilt_binaries_on_windows():
+    if not is_windows():
+        return
+    temp_dir = tempfile.mkdtemp()
+    for lib in PREBUILT_WIN_LIBS:
+        filename = lib.format(python_version=python_version(),
+                              arch=system_architecture())
+        url = PREBUILT_DOWNLOAD_SITE + filename
+        print(f"Downloading {url}")
+        contents = download_url(url)
+        file_path = os.path.join(temp_dir, filename)
+        write_file(contents, file_path, mode="wb")
+        print("Download complete. Installing dependency.")
+        subprocess.call(["pip", "install", file_path]) #nosec
+
+
+if __name__ == "__main__":
+    install_prebuilt_binaries_on_windows()