快捷導(dǎo)航

Python configparser模塊封裝及構(gòu)造配置文件

更新時(shí)間：2020年08月07日 11:56:03 作者：不放棄自己

這篇文章主要介紹了Python configparser模塊封裝及構(gòu)造配置文件,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下

1.configparser模塊簡(jiǎn)介

使用配置文件來(lái)靈活的配置一些參數(shù)是一件很常見的事情，配置文件的解析并不復(fù)雜，在python里更是如此，在官方發(fā)布的庫(kù)中就包含有做這件事情的庫(kù)，那就是configParser

configParser解析的配置文件的格式比較象ini的配置文件格式，就是文件中由多個(gè)section構(gòu)成，每個(gè)section下又有多個(gè)配置項(xiàng)

2.看一下configparser生成的配置文件的格式

ini配置文件格式如下：

這里是注釋

[log]
log_path = base_dir/OutPut/log/

[image]
img_path = base_dir/OutPut/image/

[report]
report_path = base_dir/OutPut/report/

[test_case]
test_case_path = base_dir/TestData/case.xlsx

3.讀取文件內(nèi)容

import configparser
import os
import sys

BASE_DIR = os.path.dirname(os.path.dirname(os.path.dirname(__file__)))

if sys.platform == "win32":
  ENV_CONF_DIR = os.path.join(BASE_DIR, 'Common/conf/env_config.ini').replace('/', '\\')
else:
  ENV_CONF_DIR = os.path.join(BASE_DIR, 'Common/conf/env_config.ini')

class Config(object):

  def __init__(self, path):
    self.path = path #配置文件名
    self.cf = configparser.ConfigParser() #創(chuàng)建一個(gè)配置文件對(duì)象
    self.cf.read(self.path, encoding='utf-8') # 調(diào)用配置文件對(duì)象的讀取方法，并傳入一個(gè)配置文件名

  def get(self, field, key): # 獲取字符串類型的選項(xiàng)值
    result = ""
    try:
      result = self.cf.get(field, key)
    except:
      result = ""
    return result

  def set(self, field, key, value):
    try:
      self.cf.set(field, key, value)
      self.cf.write(open(self.path, 'w'))#創(chuàng)建一個(gè)配置文件并將獲取到的配置信息使用配置文件對(duì)象的寫入方法進(jìn)行寫入
    except:
      return False
    return True

def r_config(config_file_path, field, key):
  rf = configparser.ConfigParser()
  try:
    rf.read(config_file_path, encoding='utf-8')
    if sys.platform == "win32":
      result = rf.get(field, key).replace('base_dir', str(BASE_DIR)).replace('/', '\\')
    else:
      result = rf.get(field, key).replace('base_dir', str(BASE_DIR))
  except:
    sys.exit(1)
  return result
def w_config(config_file_path, field, key, value):
  wf = configparser.ConfigParser()
  try:
    wf.read(config_file_path)
    wf.set(field, key, value)
    wf.write(open(config_file_path, 'w'))
  except:
    sys.exit(1)
  return True
if __name__ == '__main__':
  print(r_config(ENV_CONF_DIR, 'log', 'log_path'))
  print(r_config(ENV_CONF_DIR, 'DB', 'database'))