0df7dab345
* OV new package structure * Fixes * More fixes * Fixed code style in ngraph tests * Fixes * Paths to setupvars inside demo scripts * Fixed demo_security_barrier_camera.sh * Added setupvars.sh to old location as well * Fixed path * Fixed MO install path in .co * Fixed install of public headers * Fixed frontends installation * Updated DM config files * Keep opencv in the root * Improvements * Fixes for demo scripts * Added path to TBB * Fix for MO unit-tests * Fixed tests on Windows * Reverted arch * Removed arch * Reverted arch back: second attemp * System type * Fix for Windows * Resolve merge conflicts * Fixed path * Path for Windows * Added debug for Windows * Added requirements_dev.txt to install * Fixed wheel's setup.py * Fixed lin build * Fixes after merge * Fix 2 * Fixes * Frontends path * Fixed deployment manager * Fixed Windows * Added cldnn unit tests installation * Install samples * Fix samples * Fix path for samples * Proper path * Try to fix MO hardcodes * samples binary location * MO print * Added install for libopencv_c_wrapper.so * Added library destination * Fixed install rule for samples * Updated demo scripts readme.md * Samples * Keep source permissions for Python samples * Fixed python * Updated path to fast run scripts * Fixed C samples tests * Removed debug output * Small fixes * Try to unify prefix
37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
#!/usr/bin/env python3
|
|
|
|
# Copyright (C) 2018-2021 Intel Corporation
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
""" Common utilities for OpenVINO install package.
|
|
"""
|
|
import errno
|
|
import os
|
|
from pathlib import Path
|
|
import subprocess
|
|
import sys
|
|
|
|
|
|
def get_openvino_environment(install_prefix: Path):
|
|
""" Get OpenVINO environment variables
|
|
"""
|
|
if sys.platform == "win32":
|
|
script = install_prefix / "setupvars.bat"
|
|
cmd = f"{script} && set"
|
|
else:
|
|
script = install_prefix / "setupvars.sh"
|
|
# setupvars.sh is not compatible with /bin/sh. Using bash.
|
|
cmd = f'bash -c ". {script} && env"'
|
|
|
|
if not os.path.exists(str(script)):
|
|
raise FileNotFoundError(errno.ENOENT, os.strerror(errno.ENOENT), str(script))
|
|
|
|
env = {}
|
|
dump = subprocess.check_output(cmd, shell=True, universal_newlines=True).strip()
|
|
for line in dump.split("\n"):
|
|
# split by first '='
|
|
pair = [str(val).strip() for val in line.split("=", 1)]
|
|
if len(pair) > 1 and pair[0]: # ignore invalid entries
|
|
env[pair[0]] = pair[1]
|
|
return env
|