98 lines
3.6 KiB
Python
98 lines
3.6 KiB
Python
import logging, sys, inspect
|
|
from os.path import dirname, abspath
|
|
from optparse import OptionParser
|
|
|
|
def log_error_and_exit(error_message):
|
|
logging.error(error_message)
|
|
raise SystemExit
|
|
|
|
#try to import setuptools
|
|
try:
|
|
from setuptools import setup, Extension
|
|
from setuptools.command import easy_install
|
|
except ImportError:
|
|
log_error_and_exit("""setuptools is not installed for \"""" + sys.executable + """\"
|
|
Follow this link for installing instructions :
|
|
https://pypi.python.org/pypi/setuptools
|
|
make sure you use \"""" + sys.executable + """\" during the installation""")
|
|
|
|
from pkg_resources import parse_version
|
|
|
|
def notinstalled(modulename):
|
|
return modulename + """ could not be imported for \"""" + sys.executable + """\"
|
|
Set PYTHONPATH to the output of this command \"make print-OR_TOOLS_PYTHONPATH\" before running the examples"""
|
|
|
|
def wrong_module(module_file, modulename):
|
|
return """
|
|
The python examples are not importing the """ + modulename + """ module from the sources.
|
|
Remove the site-package that contains \"""" + module_file + """\", either manually or by using pip, and rerun this script again."""
|
|
|
|
# Returns the n_th parent of file
|
|
def n_dirname(n, file):
|
|
directory = file
|
|
for x in range(0, n):
|
|
directory = dirname(directory)
|
|
return directory
|
|
|
|
if __name__ == '__main__':
|
|
parser = OptionParser('Log level')
|
|
parser.add_option('-l','--log',type='string',help='Available levels are CRITICAL (3), ERROR (2), WARNING (1), INFO (0), DEBUG (-1)',default='INFO')
|
|
options,args = parser.parse_args()
|
|
|
|
try:
|
|
loglevel = getattr(logging,options.log.upper())
|
|
except AttributeError:
|
|
loglevel = {3:logging.CRITICAL,
|
|
2:logging.ERROR,
|
|
1:logging.WARNING,
|
|
0:logging.INFO,
|
|
-1:logging.DEBUG,
|
|
}[int(options.log)]
|
|
|
|
logging.basicConfig(format='[%(levelname)s] %(message)s',stream=sys.stdout, level=loglevel)
|
|
|
|
logging.info("Python path : " + sys.executable)
|
|
logging.info("Python version : " + sys.version)
|
|
logging.info("sys.path : " + str(sys.path))
|
|
ortools_project_path = n_dirname(3, abspath(inspect.getfile(inspect.currentframe())))
|
|
|
|
#try to import ortools
|
|
try:
|
|
import ortools
|
|
except ImportError:
|
|
logging.error (notinstalled("ortools"))
|
|
raise SystemExit
|
|
|
|
#check if we're using ortools from the sources or it's binded by pypi's module
|
|
ortools_module_file = inspect.getfile(ortools)
|
|
ortools_module_path = n_dirname(3, ortools_module_file)
|
|
if(ortools_module_path == ortools_project_path):
|
|
logging.info("Or-tools is imported from : " + ortools_module_file)
|
|
else:
|
|
log_error_and_exit(wrong_module(ortools_module_file, "ortools"))
|
|
|
|
# Check if python can load the libraries' modules
|
|
# this is useful when the library architecture is not compatbile with the python executable,
|
|
# or when the library's dependencies are not available or not compatible.
|
|
from ortools.constraint_solver import _pywrapcp
|
|
from ortools.linear_solver import _pywraplp
|
|
from ortools.algorithms import _pywrapknapsack_solver
|
|
from ortools.graph import _pywrapgraph
|
|
|
|
#try to import protobuf
|
|
try:
|
|
import google.protobuf
|
|
except ImportError:
|
|
log_error_and_exit(notinstalled("protobuf"))
|
|
|
|
#check if we're using protobuf from the sources or it's binded by pypi's module
|
|
protobuf_module_file = inspect.getfile(google.protobuf)
|
|
protobuf_module_path = n_dirname(7, protobuf_module_file)
|
|
if(protobuf_module_path == ortools_project_path):
|
|
logging.info("Protobuf is imported from : " + protobuf_module_file)
|
|
else:
|
|
log_error_and_exit(wrong_module(protobuf_module_file, "protobuf"))
|
|
|
|
#Check if the protobuf modules were successfully generated
|
|
from google.protobuf import descriptor as _descriptor
|
|
from google.protobuf import descriptor_pb2 |