def source_bash_script(self, to="/etc/bash.bashrc"):

if path.exists(to) and path.exists("/usr/share/bm/bm.sh"):

f = open(to)

bashrc = f.read()

f.close()

cmd = "source /usr/share/bm/bm.sh"

if cmd in bashrc.splitlines():

return

else:

print ("Sourcing bm.sh")

try:

f = open(to, "a")

except:

print ("Error: Couldn't write to %s" % to)

return

f.write(cmd + os.linesep)

f.close()

else:

print ("Warning: the shell functions could not be sourced")

def update_windows_registry(self):

try:

import _winreg

except:

print ("Couldn't change PATH variable in registry, because the _winreg modules could not be loaded.")

return

path = os.path.join(os.environ["PROGRAMFILES"], "btools", "bin")

regpath = r'SYSTEM\CurrentControlSet\Control\Session Manager\Environment'

reg = _winreg.ConnectRegistry(None, _winreg.HKEY_LOCAL_MACHINE)

try:

key = _winreg.OpenKey(reg, regpath, 0, _winreg.KEY_ALL_ACCESS)

except:

print ("Error getting PATH from registry")

return

value, type_id = _winreg.QueryValueEx(key, "PATH")

current = value.split(";")

if path in current:

return

try:

_winreg.SetValueEx(key, "PATH", 0, type_id, ";".join(current + [path]))

print ("Updated %s\\PATH registry value" % (regpath))

except:

print ("Error updating %s\\PATH registry value" % (regpath))

def run(self):

install_data.run(self)

if not WINDOWS:

self.source_bash_script()

else: