#!/usr/bin/python
#coding:utf-8
'''
date:9/2/17 18:03 PM
author:lockey
email:lockey@123.com
desc:python自動化安裝用戶指定版本的python環(huán)境
'''
#導入Python的系統(tǒng)編程操作模塊
import os

#導入用來處理Python運行時配置以及資源，與前當程序之外的系統(tǒng)環(huán)境交互的模塊
import sys

#判斷當前用戶是否是root用戶
if os.getuid() == 0:
  pass
else:
  print 'Not under root mode, please switch user!'
  sys.exit(1)

#獲取用戶輸入的python安裝版本
version = raw_input('Please input wanted python version(2.7/3.6)')

#如果嫌官網(wǎng)下載速度太慢可以指定python3.6.2的鏈接地址為本人上傳資源地址
#根據(jù)用戶輸入的python版本選擇源碼包下載地址
if version == '2.7':
  url = 'https://www.python.org/ftp/python/2.7.13/Python-2.7.13.tgz'
elif version == '3.6':
  url = 'https://www.python.org/ftp/python/3.6.2/Python-3.6.2.tgz'
else:
  print 'Please input given version number(2.7/3.5)'
  sys.exit(1)

#拼接源碼包下載地址并執(zhí)行下載命令
cmd = 'wget ' + url
res = os.system(cmd)
if res != 0 :
  print 'Failed to download python source package, please inspect your network!'
  sys.exit(1)

if version == '2.7':
  package_version = 'Python-2.7.13'
else:
  package_version = 'Python-3.6.2'

#解壓下載的源碼包
cmd = 'tar xf ' + package_version + '.tgz'
res = os.system(cmd)

#如果解壓失敗則刪除下載的源碼包并且提示用戶重新執(zhí)行腳本
if res != 0:
  os.system('rm ' + package_version + '.tgz')
  print 'Please reexcute the script to install python'
  sys.exit(1)

#解壓成功則進入解壓后的源碼目錄中依次執(zhí)行配置、編譯、安裝過程
cmd = 'cd ' + package_version + ' && ./configure --prefix=/usr/local/python && make && make install'

res = os.system(cmd)

#安裝失敗則提示用戶安裝失敗了，讓用戶檢查環(huán)境依賴
if res != 0:
  print 'Failed to install python, please inspect dependencies for python install!'
  sys.exit(1)