python切換網絡連接配接

#coding=GB2312

import os
import sys 
import subprocess
import _winreg

LOCAL_CONNECT_NAME = '本地連接配接'
WIFI_CONNECT_NAME  = '無線網絡連接配接'
INTERNET_SETTING_REG = r'Software\Microsoft\Windows\CurrentVersion\Internet Settings'
PROXY_ENABLE_NAME = 'ProxyEnable'

def setEnable(connName, enable=True):
	str_enable = ''
	if enable:
		str_enable = 'enable'
	else:
		str_enable = 'disable'

	cmd='netsh int set int name=%s admin=%s' %\
			('"'+connName+'"',\
			str_enable)
	print cmd
	ret =subprocess.call(cmd, shell=True)
	#print 'ret=',ret
	if ret != 0:
		print '網絡連接配接設定失敗'
	else:
		print '網絡連接配接設定成功'
		
def setIEProxyEnable(val= 1):
	try:
		#OpenKey預設以隻讀的權限打開，由于沒有設定_winreg.KEY_ALL_ACCESS
		#導緻後面的setValueEx挂掉，出現WindowsError [5]
		#要特别注意打開權限
		key = _winreg.OpenKey(_winreg.HKEY_CURRENT_USER,INTERNET_SETTING_REG,0, _winreg.KEY_ALL_ACCESS)
		v, t = _winreg.QueryValueEx(key, PROXY_ENABLE_NAME)
		if v != val:
			_winreg.SetValueEx(key, PROXY_ENABLE_NAME, 0, t, val)
	except Exception,e:
		print '設定IE代理失敗：' , e.message
		sys.exit(0)
	print '設定代理成功'
	
if __name__ == '__main__':
	reload(sys)
	sys.setdefaultencoding('GB2312')
	
	setIEProxyEnable(0)
	if len(sys.argv) != 2:
		print 'switch_net.py N\r\n0:local 1:wifi.'
		sys.exit(0)
		
	if int(sys.argv[1]) == 0:
		setEnable(LOCAL_CONNECT_NAME)
		setEnable(WIFI_CONNECT_NAME,False)
		setIEProxyEnable()
	else:
		setEnable(LOCAL_CONNECT_NAME,False)
		setEnable(WIFI_CONNECT_NAME)
		setIEProxyEnable(0)