python 实现 2个文件替换更新

业务需求： ansible同步中，hosts需要用变化的zk-hosts文件来更新。并且在指定位置去添加和删除，本例中添加和删除为[prod]这个项目。

思路： 将 每个[prod] 方括号开头项目作为一个字典的 key ，其下的 IP 作为一个 value 。 将其制作为一个字典 。 更新时，先去找到 [ ] 项目名称，后去更新其的值， 包括删除，增加，判断是否存在。 

[root@hello test]$ cat hosts

[test]

192.168.3.3

[pre]

192.168.1.137

[prod]

192.168.1.61

192.168.1.62

192.168.1.63

[prod_cli]

192.168.1.158

192.168.1.159

[prod_admin]

192.168.1.8

[root@hello test]$ cat zk-hosts  变化的文件

[all]

-192.168.1.61

-192.168.1.62

10.0.0.1

10.0.0.2

#  源代码

#!/usr/bin/env python

import os,sys,re

#os.system('cat /root/hosts/test/hosts | tr '\n' ' ' | xargs echo  | tr '[' '\n' | tr ']' ' ' &>hosts.txt')

def update_ip(*project):

project=raw_input("pleas input project ")

os.system('/bin/bash a.sh')  #

d={}

ip=[]

f=open('hosts.txt','a+')

a=f.readlines()

f.close()

for line in a:

env=(line.split()[0])

iplist=line.split()[1:]

# if env==project:

d[env]=iplist

#print d

#print project

ciplist=d[project]

#print ciplist

#

zk=[]

z=open('zk-hosts')

for hosts in z.readlines():

hosts=hosts.strip('\n')

zk.append(hosts)

zk.pop(0)

#print zk

#

for i in zk:

# print i

if i in ciplist:

print "%s is exists" % i

elif re.match(r'-(.*)',i):

i=i.lstrip('-')

if i in ciplist:

ciplist.remove(i)

else:

ciplist.append(i)

#print ciplist

#print d.items()

b=open('hosts','a+')

b.truncate()

for key in d:

b.write('['+key+']'+'\n')

for m in d[key]:

print m

b.write(m+'\n')

b.close()

update_ip()

#python脚本中调用的 shell 脚本。

[root@hello test]$ cat a.sh

#!/bin/bash

cat /root/hosts/test/hosts | tr '\n' ' ' | xargs echo  | tr '[' '\n' | tr ']' ' ' &>hosts.txt

sed -i '1d' hosts.txt

#cat /root/hosts/test/zk-hosts | tr '\n' ' ' | xargs echo | tr '[' '\n' | tr ']' ' ' &>zk-hosts.txt

#sed -i '1d' zk-hosts.txt

执行后结果

[root@hello test]$ cat hosts

[test]

192.168.3.3

[pre]

192.168.1.137

[prod]

192.168.1.63

10.0.0.1

10.0.0.2

[prod_admin]

192.168.1.8

[prod_cli]

192.168.1.158

192.168.1.159