ortools-clone/tools/check_python_deps.py

import logging, sys, inspect
from optparse import OptionParser

#try to import setuptools
try:
    from setuptools import setup, Extension
    from setuptools.command import easy_install
except ImportError:
    raise ImportError("""setuptools is not installed for \"""" + sys.executable + """\"
Please follow this link for installing instructions :
https://pypi.python.org/pypi/setuptools
make sure you use \"""" + sys.executable + """\" during the installation""")
    raise SystemExit

from pkg_resources import parse_version

required_ortools_version = "5.0.3919"
required_protobuf_version = "3.0.0"

def notinstalled(modulename):
	return modulename + """ is not installed for \"""" + sys.executable + """\"
Please run \"""" + str(sys.executable) + """ setup.py install --user\""""

def wrong_version(module, modulename, required_version, installed_version):
	return """You are using """ + modulename + """-""" + installed_version + """ : """ + inspect.getfile(module) + """, while the required version is : """ + required_version + """
Run \"""" + str(sys.executable) + """ setup.py install --user\" to upgrade.
If the problem persists, then \"""" + inspect.getfile(module) + """\" is binding the newely installed version of """ + modulename + """. Remove it manually or by using pip."""

def log_error_and_exit(error_message):
	logging.error(error_message)
	raise SystemExit

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()
 
	#Create the logger
	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)

	#Display Python Version and path
	logging.info("Python path : " + sys.executable)
	logging.info("Python version : " + sys.version)

	#Choose the pypi package
	if sys.version_info[0] >= 3:
		ortools_name = "py3-ortools"
	else:
		ortools_name = "ortools"

	#try to import ortools
	try:
		import ortools
	except ImportError:
	    log_error_and_exit(notinstalled(ortools_name))

	#try to import protobuf
	try:
		import google.protobuf
	except ImportError:
	    log_error_and_exit(notinstalled("protobuf"))

	#check ortools version
	try:
		if required_ortools_version != ortools.__version__:
			raise Exception
		logging.info("or-tools version : " + ortools.__version__ + "\n" + inspect.getfile(ortools))
	except (AttributeError, Exception):
		log_error_and_exit(wrong_version(ortools, ortools_name, required_ortools_version, ortools.__version__))

	#check protobuf version
	try:
		if required_protobuf_version != google.protobuf.__version__:
			raise Exception
		logging.info("protobuf version : " + google.protobuf.__version__+ "\n" + inspect.getfile(google.protobuf) )
	except (AttributeError, Exception):
		log_error_and_exit(wrong_version(google.protobuf, "protobuf", required_protobuf_version, google.protobuf.__version__))

	# 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