python小练习-修改host文件

在测试过程中，需要频繁的切换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')

 

posted @ 2017-04-28 15:29  灵主  阅读(1134)  评论(0)    收藏  举报