记一次使用python脚本来修改hosts文件，由于公司服务器需要换ip地址，服务器里都是Linux系统，而hosts文件中有些解析的ip，手动一台一台的解析太麻烦，就写了这个脚本。本来以为感觉很简单，但是写的过程真不好写，也试过shell脚本，不过也不好写。然后就直接用python写，也找了一些网上的教程，不过也没有很好的解决问题。下面把脚本贴出来记录下

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
#!/usr/bin/python
#coding:utf8
import  os
import  sys
import  re
 
hostsfile = "/etc/hosts"   #hosts文件绝对路径
 
ip_dict  =  { "1.1.1.1" : "2.2.2.1" , "1.1.1.2" : "2.2.2.2" , "1.1.1.3" : "2.2.2.3" }   #这是自定义的字典形式为：old_ip:new_ip
 
ip  =  []  #定一个空列表，来存储old_ip的列表
Line = []  # 定一个空列表，来存储修改后的hosts文件内容的列表
 
fd  =  open (hostsfile).readlines()   #打开文件
 
#用for循环得到old_ip的列表
for  old_ip  in  ip_dict.keys():
     ip.append(old_ip)
 
#用for循环列出每一行，并进行匹配old_ip 如果匹配到就使用re.sub()进行替换。并存储在Line这个列表中
for  line  in  fd:
     if  line.strip()  = =  '':
         #continue
         Line.append(line)  ##如果是空行也加入列表中，保证文件内容与原内容形式一致
     else :
         h_ip  =  line.strip().split()[ 0 ]  ##取得hosts文件中的ip地址
         if  h_ip  in  ip:
             lin  =  re.sub(h_ip,ip_dict[h_ip],line)  #如果匹配到就进行替换
             print  "文件修改的内容如下："
             print  "%s --> %s"  % (line.strip( "\n" ),lin)
             Line.append(lin)
         else :
             Line.append(line)
 
#最后得到Line列表
#重新把列表的内容写入到/etc/hosts文件中
fc  =  open (hostsfile, 'w' )
fc.writelines(Line)
fc.close()