自定配置文件的增删改查

自定配置文件的增删改查

 

# @Time    : 2017/4/18 15:31
# @Author  : Cui X
# @File    : configparser_mod.py


import configparser


class Config(object):
"""置文件操作"""

def __init__(self, sections, prices):
        self.sections = sections
        self.prices = prices

# 生成配置文件模板
    @classmethod
def config_create(cls):
        config = configparser.ConfigParser()
        config["db"] = {'db_host': '192.168.1.1',
'db_port': '3306',
'db_user': 'root',
'db_pass': 'password'}
# 两种写法，上面是一种，下面是一种
        config['concurrent'] = {}
        config['concurrent']['thread'] = '200'
        config['concurrent']['processor'] = '400'
        config.write(open('test.conf', 'w'))
# 把数据读出来
    @classmethod
def config_read(cls):
        conf = configparser.ConfigParser()
        conf.read('test.conf')
return conf

# 判断数据是否在存在文件中
def config_judge(self):
        config = self.config_read()
if self.sections in config.sections():
if self.prices in config[self.sections]:
return True
else:
return False
else:
return False

# 增加
def config_add(self, price):
if self.config_judge() is False:
            config = self.config_read()
if self.sections in config.sections():
                config[self.sections][self.prices] = price
                config.write(open('test.conf', 'w'))
print('数据写入成功')
else:
print('%s 不存在' % self.sections)
else:
print('数据已存在')

def config_delete(self):
if self.config_judge() is True:
            config = self.config_read()
del config[self.sections][self.prices]
            config.write(open('test.conf', 'w'))
print('数据删除成功')
else:
print('删除的值不存在')

# 修改配置
def config_change(self, price):
if self.config_judge() is True:
            config = self.config_read()
            config[self.sections][self.prices] = price
            config.write(open('test.txt', 'w'))
print('%s %s 修改成功' % (self.sections, self.prices))

# 查询配置
def config_select(self):
if self.config_judge() is True:
            config = self.config_read()
            select = config[self.sections][self.prices]
print(select)

    @classmethod
def show_sections_all(cls):
        config = cls.config_read()
for s in config.sections():
print('[%s]' % s)
for pr in config.options(s):
print(pr, ' = ', config[s][pr])


if __name__ == '__main__':
print("""\033[32;1m
    (0) 初始化配置文件
    (1) 增加配置
    (2) 删除配置
    (3) 修改配置
    (4) 查询配置
    (5) 查看所有配置\033[0m
""")
while True:
        choose = input('选择操作项 >> ').strip()
if choose.isdigit():
            choose = int(choose)
if choose < 5 and choose > 0:
                section = input('sections >> ').strip()
                item = input('item >> ').strip()
                conf = Config(section, item)
if choose == 1:
                    price = input('add item_price >> ').strip()
                    conf.config_add(price)
elif choose == 2:
                    conf.config_delete()
elif choose == 3:
                    price = input('change item_price >> ').strip()
                    conf.config_change(price)
else:
                    conf.config_select()
elif choose == 0:
                affirm = input('yes/no ? >> ').strip()
if affirm == 'yes':
                    Config.config_create()
print('\033[32;1m配置文件初始化成功\033[0m')
else:
print('\033[31;1m已取消操作\033[0m')
elif choose == 5:
                Config.show_sections_all()
else:
print('\033[31;1m没有此项目\033[0m')
else:
print('\033[31;1m请正确输入\033[0m')