账号密码等敏感信息:利用ConfigParser保存提取

先上一个直接能用的代码

# 输入sections返回相应的options、items字典
import ConfigParser
def get_config(user):
    try:
        config_path = r'此处修改为本机存放配置文件的地址'
        conf_dict = {}
        conf = ConfigParser.ConfigParser() # 初始化
        conf.read(config_path) # 读取文件
        sections = conf.sections() # 获取sections名
        for section in sections:
            item = dict(conf.items(section))
            conf_dict[section] = item

        return conf_dict.get(user.lower())

    except Exception, e:
        print e
        print u'获取配置出错!!!'
        exit() # 直接终止所有程序

1. 函数介绍

1.1 读取配置文件

-read(filename) 直接读取ini文件内容
-sections() 得到所有的section,并以列表的形式返回
-options(section) 得到该section的所有option
-items(section) 得到该section的所有键值对
-get(section,option) 得到section中option的值,返回为string类型
-getint(section,option) 得到section中option的值,返回为int类型

1.2 写入配置文件

-add_section(section) 添加一个新的section
-set(section, option, value) 对section中的option进行设置
-write(open(filename,'w')) 写入到配置文件中

2. 实例测试

配置文件test.cfg

[sec_a]  
a_key1 = 20  
a_key2 = 10  

[sec_b]  
b_key1 = 121  
b_key2 = b_value2  
b_key3 = $r  
b_key4 = 127.0.0.1  

测试文件test.py

# -* - coding: UTF-8 -* -  
import ConfigParser  
#生成config对象  
conf = ConfigParser.ConfigParser()  
#用config对象读取配置文件  
conf.read("test.cfg")  
#以列表形式返回所有的section  
sections = conf.sections()  
print 'sections:', sections         #sections: ['sec_b', 'sec_a']  
#得到指定section的所有option  
options = conf.options("sec_a")  
print 'options:', options           #options: ['a_key1', 'a_key2']  
#得到指定section的所有键值对  
kvs = conf.items("sec_a")  
print 'sec_a:', kvs                 #sec_a: [('a_key1', '20'), ('a_key2', '10')]  
#指定section,option读取值  
str_val = conf.get("sec_a", "a_key1")  
int_val = conf.getint("sec_a", "a_key2")  

print "value for sec_a's a_key1:", str_val   #value for sec_a's a_key1: 20  
print "value for sec_a's a_key2:", int_val   #value for sec_a's a_key2: 10  

#写配置文件  
#更新指定section,option的值  
conf.set("sec_b", "b_key3", "new-$r")  
#写入指定section增加新option和值  
conf.set("sec_b", "b_newkey", "new-value")  
#增加新的section  
conf.add_section('a_new_section')  
conf.set('a_new_section', 'new_key', 'new_value')  
#写回配置文件  
conf.write(open("test.cfg", "w"))  

3. RawConfigParser、ConfigParser、SafeConfigParser

RawCnfigParser是最基础的INI文件读取类,ConfigParser、SafeConfigParser支持对%(value)s变量的解析。

[portal]
url = http://%(host)s:%(port)s/Portal
host = localhost
port = 8080

get("url") 得到  http://localhost:8080/Portal

你可能感兴趣的:(账号密码等敏感信息:利用ConfigParser保存提取)