#coding=utf-8

import os

import sys

if os.getuid() == 0:

    pass

else:

    print("当前用户不是Root用户，请以Root用户执行脚本")

    sys.exit(1)

#version = rat_input('请输入您想安装的Python版本号（2.7/3.5）:')

version = '3.5'

if version == '2.7':

    url = 'https://www.python.org/ftp/python/2.7/Python-2.7.tgz'

elif version == '3.5':

    url = 'https://www.python.org/ftp/python/3.5.0/Python-3.5.0.tgz'

else:

    print("您输入的版本号有误，请输入2.7或者3.5")

    sys.exit(1)

cmd = 'wget '+url

res = os.system(cmd)

if res != 0:

    print('下载源码包失败，请检查当前网络')

    sys.exit(1)

if version == '2.7':

    package_name = 'Python-2.7'

else:

    package_name = 'Python-3.5.0'

cmd = 'tar xf '+ package_name+ '.tgz'

res = os.system(cmd)

if res != 0:

    os.system('rm '+package_name+'.tgz')

    print('解压源码包失败，请重新运行这个脚本下载源码包')

    sys.exit(1)

cmd = 'cd ' + package_name+ '&& ./configure --prefix=/usr/local/python && make && make install'

res = os.system(cmd)

if res != 0:

    print '编译Python失败'

    sys.exit(1)

cmd = 'mv /usr/bin/python /usr/bin/python_old'

res = os.system(cmd)

if res !=0:

    print '移动失败，请手动移动(mv /usr/bin/python /usr/bin/python_old)'

    sys.exit(1)

cmd = 'ln -s /usr/local/python/bin/python3.5 /usr/bin/python'

res = os.system(cmd)

if res != 0:

    print '软连接失败，请手动软连(ln -s /usr/local/python/bin/python3.5 /usr/bin/python)'

    sys.exit(1)

cmd = "sed -i 's/python/python_old/g' /usr/bin/yum"

res = os.system(cmd)

if res != 0:

    print '修改失败，请手动修改文件内容：/usr/bin/yum - > #!/usr/bin/python 替换成 #!/usr/bin/python_old'

    sys.exit(1)

cmd = "sed -i 's/python/python_old/g' /usr/libexec/urlgrabber-ext-down"

res = os.system(cmd)

if res != 0:

    print '修改失败，请手动修改文件内容：/usr/libexec/urlgrabber-ext-down - > #!/usr/bin/python 替换成 #!/usr/bin/python_old'

    sys.exit(1)