pymolscriptrepo.py

'''
Support module for Pymol-script-repo

Adds repo binary directories to PATH if PYMOL_GIT_MOD is defined.
'''

import os

if 'PYMOL_GIT_MOD' in os.environ:
	import sys
	import platform

	repopath = [os.environ.get('PATH', '.')]


	##
	## this is ugly, we need proper paths for each platform, but not for
	## every app an extra path!
	##
	if sys.platform.startswith('linux'):
		if platform.architecture()[0] == '32bit':
			repopath.append(os.path.join(os.environ['PYMOL_GIT_MOD'], 'autodock_423', 'i86Linux2'))
		else:
			repopath.append(os.path.join(os.environ['PYMOL_GIT_MOD'], 'autodock_423', 'ia64Linux2'))
		repopath.append(os.path.join(os.environ['PYMOL_GIT_MOD'], 'autodock_vina', 'autodock_vina_1_1_2_linux_x86'))
	elif sys.platform.startswith('darwin'):
		version = platform.release().split('.')[0]
		if version in ('10', '9', '8'):
			repopath.append(os.path.join(os.environ['PYMOL_GIT_MOD'], 'autodock_423', 'universalDarwin' + version))
		repopath.append(os.path.join(os.environ['PYMOL_GIT_MOD'], 'autodock_vina', 'autodock_vina_1_1_2_mac'))
	elif sys.platform.startswith('win'):
		repopath.append(os.path.join(os.environ['PYMOL_GIT_MOD'], 'autodock_423', 'win32'))
		repopath.append(os.path.join(os.environ['PYMOL_GIT_MOD'], 'autodock_vina', 'autodock_vina_1_1_2_win32'))
	
	if len(repopath) > 1:
		os.environ['PATH'] = os.pathsep.join(repopath)

def which(*names):
	'''
	Return full path to executable or empty string if not found in PATH.
	'''
	if 'PATHEXT' in os.environ:
		pathext = [''] + os.environ['PATHEXT'].split(';')
		names = [n + ext for n in names for ext in pathext]
	path = os.environ['PATH'].split(os.pathsep)
	for n in names:
		for p in path:
			full = os.path.join(p, n)
			if os.path.isfile(full):
				return full
	return ''