UI自动化web端框架config.py代码

import json
from lib.core.path import CONF_PATH


class ConfigType(object):
    MYSQL = 'mysql'
    REDIS = 'redis'
    LOG = 'log'


class Config(object):
    def __init__(self):
        self.config = {}

    def make(self):
        # mysql配置信息
        self.config[ConfigType.MYSQL] = {}
        self.config[ConfigType.MYSQL]['host'] = '116.196.122.78'
        self.config[ConfigType.MYSQL]['port'] = 3306
        self.config[ConfigType.MYSQL]['user'] = 'zbox'
        self.config[ConfigType.MYSQL]['passwd'] = 'Houyafan123'
        self.config[ConfigType.MYSQL]['db'] = 'zbox'
        self.config[ConfigType.MYSQL]['charset'] = 'utf8'

        # redis配置信息
        self.config[ConfigType.REDIS] = {}
        self.config[ConfigType.REDIS]['host'] = '116.196.122.78'
        self.config[ConfigType.REDIS]['port'] = 6379
        self.config[ConfigType.REDIS]['passwd'] = 'Houyafan123'

        # log配置文件信息
        # self.config[ConfigType.LOG] = {}
        # self.config[ConfigType.LOG]['console'] = logging.DEBUG
        # self.config[ConfigType.LOG]['main'] = logging.DEBUG
        f = open(CONF_PATH, 'w')
        f.write(json.dumps(self.config))
        f.close()

    def read(self, config):
        f = open(CONF_PATH)
        data = json.load(f)  # 二维字典
        if config == ConfigType.MYSQL:
            return data[ConfigType.MYSQL]  # 通过二维字典里的mysql，打印出的是一个一维字典
        elif config == ConfigType.REDIS:
            return data[ConfigType.REDIS]


if __name__ == '__main__':
    Config().make()
    print(Config().read(ConfigType.MYSQL))
    # print(Config().read(ConfigType.REDIS))