切换网络和代理

#coding=GB2312

import os
import sys 
import subprocess
import _winreg
#import psutil

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)