python脚本修改hosts文件

2023-01-31 05:01:58 修改 文件 脚本

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

#!/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()


相关文章