在测试过程中,需要频繁的切换host,有些麻烦,就写个脚本自动切换吧。
1 #coding:utf-8 2 __author__ = 'Administrator' 3 4 import os 5 6 def host_change(ip_address): 7 with open('C:\\WINDOWS\\system32\\drivers\\etc\\HOSTS','r') as h,\ 8 open('C:\\WINDOWS\\system32\\drivers\\etc\\HOSTS_bak','w') as f: 9 for line in h: 10 if ip_address in line: 11 if line.startswith('#'): 12 print u'\033[031;1m当前为正式环境\033[0m' 13 order = raw_input('\033[031;1m是否要切换为预发布环境:\033[0m').lower() 14 if order == 'y': 15 line = line.replace('#','') 16 f.write(line) 17 else: 18 f.write(line) 19 else: 20 print u'\033[031;1m当前为预发布环境\033[0m' 21 order = raw_input('\033[031;1m是否要切换为正式环境:\033[0m').lower() 22 if order == 'y': 23 line = '#' + line 24 f.write(line) 25 else: 26 f.write(line) 27 28 else: 29 f.write(line) 30 31 os.remove('C:\\WINDOWS\\system32\\drivers\\etc\\HOSTS') 32 os.rename('C:\\WINDOWS\\system32\\drivers\\etc\\HOSTS_bak',\
'C:\\WINDOWS\\system32\\drivers\\etc\\HOSTS') 33 34 host_change('10.251.92.25')