mirror of
https://github.com/kennethreitz-archive/python-build.git
synced 2026-06-05 15:20:16 +00:00
utahta
110 lines
3.6 KiB
Python
110 lines
3.6 KiB
Python
import os
|
|
import sys
|
|
import glob
|
|
import shutil
|
|
from pythonbrew.util import makedirs, rm_r
|
|
from pythonbrew.define import PATH_BUILD, PATH_BIN, PATH_DISTS, PATH_PYTHONS,\
|
|
PATH_ETC, PATH_SCRIPTS, PATH_SCRIPTS_PYTHONBREW,\
|
|
PATH_SCRIPTS_PYTHONBREW_COMMANDS, PATH_BIN_PYTHONBREW,\
|
|
PATH_LOG, PATH_PATCHES, PATH_ETC_CONFIG,\
|
|
PATH_SCRIPTS_PYTHONBREW_INSTALLER, PATH_VENVS, PATH_HOME_ETC, ROOT
|
|
import stat
|
|
import time
|
|
|
|
class PythonbrewInstaller(object):
|
|
"""pythonbrew installer:
|
|
"""
|
|
|
|
@staticmethod
|
|
def install(installer_root):
|
|
# create directories
|
|
makedirs(PATH_PYTHONS)
|
|
makedirs(PATH_BUILD)
|
|
makedirs(PATH_DISTS)
|
|
makedirs(PATH_ETC)
|
|
makedirs(PATH_BIN)
|
|
makedirs(PATH_LOG)
|
|
makedirs(PATH_VENVS)
|
|
makedirs(PATH_HOME_ETC)
|
|
|
|
# create script directories
|
|
rm_r(PATH_SCRIPTS)
|
|
makedirs(PATH_SCRIPTS)
|
|
makedirs(PATH_SCRIPTS_PYTHONBREW)
|
|
makedirs(PATH_SCRIPTS_PYTHONBREW_COMMANDS)
|
|
makedirs(PATH_SCRIPTS_PYTHONBREW_INSTALLER)
|
|
|
|
# copy all .py files
|
|
for path in glob.glob(os.path.join(installer_root,"*.py")):
|
|
shutil.copy(path, PATH_SCRIPTS_PYTHONBREW)
|
|
for path in glob.glob(os.path.join(installer_root,"commands","*.py")):
|
|
shutil.copy(path, PATH_SCRIPTS_PYTHONBREW_COMMANDS)
|
|
for path in glob.glob(os.path.join(installer_root,"installer","*.py")):
|
|
shutil.copy(path, PATH_SCRIPTS_PYTHONBREW_INSTALLER)
|
|
|
|
# create patches direcotry
|
|
rm_r(PATH_PATCHES)
|
|
shutil.copytree(os.path.join(installer_root,"patches"), PATH_PATCHES)
|
|
|
|
# create a main file
|
|
fp = open("%s/pythonbrew_main.py" % PATH_SCRIPTS, "w")
|
|
fp.write("""import pythonbrew
|
|
if __name__ == "__main__":
|
|
pythonbrew.main()
|
|
""")
|
|
fp.close()
|
|
|
|
# create entry point file
|
|
fp = open(PATH_BIN_PYTHONBREW, "w")
|
|
fp.write("""#!/usr/bin/env bash
|
|
%s %s/pythonbrew_main.py "$@"
|
|
""" % (sys.executable, PATH_SCRIPTS))
|
|
fp.close()
|
|
# mode 0755
|
|
os.chmod(PATH_BIN_PYTHONBREW, stat.S_IRUSR|stat.S_IWUSR|stat.S_IXUSR|stat.S_IRGRP|stat.S_IXGRP|stat.S_IROTH|stat.S_IXOTH)
|
|
|
|
# create a bashrc for pythonbrew
|
|
shutil.copy(os.path.join(installer_root,'etc','bashrc'), os.path.join(PATH_ETC,'bashrc'))
|
|
|
|
# copy config.cfg
|
|
shutil.copy(os.path.join(installer_root,'etc','config.cfg'), PATH_ETC_CONFIG)
|
|
|
|
@staticmethod
|
|
def systemwide_install():
|
|
profile = """\
|
|
#begin-pythonbrew
|
|
if [ -n "${BASH_VERSION:-}" -o -n "${ZSH_VERSION:-}" ] ; then
|
|
export PYTHONBREW_ROOT=%(root)s
|
|
source "${PYTHONBREW_ROOT}/etc/bashrc"
|
|
fi
|
|
#end-pythonbrew
|
|
""" % {'root': ROOT}
|
|
|
|
if os.path.isdir('/etc/profile.d'):
|
|
fp = open('/etc/profile.d/pythonbrew.sh', 'w')
|
|
fp.write(profile)
|
|
fp.close()
|
|
elif os.path.isfile('/etc/profile'):
|
|
# create backup
|
|
shutil.copy('/etc/profile', '/tmp/profile.pythonbrew.%s' % int(time.time()))
|
|
|
|
output = []
|
|
is_copy = True
|
|
fp = open('/etc/profile', 'r')
|
|
for line in fp:
|
|
if line.startswith('#begin-pythonbrew'):
|
|
is_copy = False
|
|
continue
|
|
elif line.startswith('#end-pythonbrew'):
|
|
is_copy = True
|
|
continue
|
|
if is_copy:
|
|
output.append(line)
|
|
fp.close()
|
|
output.append(profile)
|
|
|
|
fp = open('/etc/profile', 'w')
|
|
fp.write(''.join(output))
|
|
fp.close()
|
|
|