欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

Python比较配置文件

程序员文章站 2022-06-28 21:07:39
工作中最常见的配置文件有四种:普通key=value的配置文件、Json格式的配置文件、HTML格式的配置文件以及YAML配置文件。 这其中以第一种居多,后三种在成熟的开源产品中较为常见,本文只针对第一种配置文件。 一般来说Linux shell下提供了diff命令来比较普通文本类的配置文件,Pyt ......

工作中最常见的配置文件有四种:普通key=value的配置文件、json格式的配置文件、html格式的配置文件以及yaml配置文件。

这其中以第一种居多,后三种在成熟的开源产品中较为常见,本文只针对第一种配置文件。

一般来说linux shell下提供了diff命令来比较普通文本类的配置文件,python的difflib也提供了str和html的比较接口,但是实际项目中这些工具其实并不好用,主要是因为我们的配置文件并不是标准化统一化的。

为了解决此类问题,最好针对特定的项目写特定的配置文件比较工具,这样在版本发布时会很有用处。

其他话懒的说了,直接贴代码:

#!/usr/bin/python
#coding=utf-8

'''
此脚本适用于比较key=value格式的配置文件
使用方式为:
config_match.py old_file new_file
最终会将new_file中的新增配置项添加入old_file中,old_file中已有的配置项不会做任何改变。
'''

import re
import os,sys
reload(sys)
sys.setdefaultencoding("utf-8")

try:
    old_file = sys.argv[1]
    new_file = sys.argv[2]
except exception,e:
    print("error:" +str(e))
    print("usage: config_match.py old_file new_file")
    sys.exit()

def list2dict(file):
    with open(file,'r') as f:
        list=f.readlines()
    #遍历str list,排除空行和以#开头的str,使用split将str元素转为[k,v]元素
    for e in list[0:]:
        if re.match('^#',e) or re.match('^$',e):
            list.remove(e)
    i=0
    for e in list:
        e_split=e.strip().split('=')
        if len(e_split)==2:
            k,v=e.strip().split('=')
            list[i]=[k,v]
        else:
            pass
        i=i+1
    #至此list变为了[[k,v],[k,v],...]这样的list
    #包含多个=号或者不包含=号的配置项会被忽略,这点要注意
    return dict(list)

old_dict=list2dict(old_file)
new_dict=list2dict(new_file)

'''
遍历新配置文件,将新增配置项加入conf_need_added{}字典中,然后将conf_need_added{}字典以k=v格式append入旧文件中。
如果重复的键值也需要更新那就太简单了,dict类型自带的update()方法就可以很好的完成任务,根本无需这么折腾了。
'''
conf_need_added={}
for k,v in new_dict.items():
    if k not in old_dict.keys():
        conf_need_added[k]=v
    else:
        pass

with open(old_file,'a') as f:
    for k,v in conf_need_added.items():
        f.write('\n#以下为新增配置项:\n')
        f.write(str(k)+'='+str(v)+'\n')