第一个自动化运维程序

# coding = utf-8
import os
if os.getuid()== 0:
    pass
else:
    print '当前用户不是root用户，请以root用户执行脚本'
    sys.exit(1)
version = raw_input('请输入你想安装的Python版本（2.7/3.6）')
if version == '2.7':
    url = 'https://www.python.org/ftp/python/2.7.15/python-2.7.15.amd64-pdb.zip'
elif version == '3.6':
    url = 'https://www.python.org/ftp/python/3.6.5/python-3.6.5-embed-amd64.zip'
else:
    print '你输入的版本号有误，请输入2.7或3.6'
    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.15'
else :
    package_name = 'Python-3.6.5'
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 install'
res = os.system(cmd)
if res != 0:
    print '编译Python源码失败，请检查是否缺少依赖库'
    sys.exit(1)