#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
  @Time  : 2018/6/22
  @Author : LiuXueWen
  @Site  : 
  @File  : Util_Ini_Operation.py
  @Software: PyCharm
  @Description: ini配置文件操作工具類
    1.讀取.ini配置文件
    2.修改.ini配置文件
    [section]
    option:value
"""
import ConfigParser

'''
  基礎讀取配置文件
    -read(filename)     直接讀取文件內容
    -sections()       得到所有的section，并以列表的形式返回
    -options(section)    得到該section的所有option
    -items(section)     得到該section的所有鍵值對
    -get(section,option)  得到section中option的值，返回為string類型
    -getint(section,option) 得到section中option的值，返回為int類型，還有相應的getboolean()和getfloat() 函數。
'''
class get_ini():

  # 初始化配置文件對象
  def __init__(self,path):
    # 實例化
    self.cf = ConfigParser.ConfigParser()
    # 讀取配置文件
    self.cf.read(path)

  # 獲取所有的sections
  def get_sections(self):
    sections = self.cf.sections()
    return sections

  # 獲取section下的所有key
  def get_options(self,section):
    opts = self.cf.options(section=section)
    return opts

  # 獲取section下的所有鍵值對
  def get_kvs(self,section):
    kvs = self.cf.items(section=section)
    return kvs

  # 根據section和option獲取指定的value
  def get_key_value(self,section,option):
    opt_val = self.cf.get(section=section,option=option)
    return opt_val

  # 更新指定section的option下的value
  # def update_section_option_val(self,section,option,value,path,module):
  #   self.cf.set(section=section,option=option,value=value)
  #   with open(path,module) as f:
  #     self.cf.write(f)

'''
  基礎寫入配置文件
    -write(fp)             將config對象寫入至某個 .init 格式的文件 Write an .ini-format representation of the configuration state.
    -add_section(section)       添加一個新的section
    -set(section, option, value)    對section中的option進行設置，需要調用write將內容寫入配置文件 ConfigParser2
    -remove_section(section)      刪除某個 section
    -remove_option(section, option)  刪除某個 section 下的 option
'''
class write_ini():

  def __init__(self,path,module):
    # 實例化配置對象
    self.cf = ConfigParser.ConfigParser()
    # 獲取寫入文件路徑，若采用w+方式則該文件可以不存在
    self.path = path
    # 配置寫入方式，寫入方式"w+"清空寫
    self.module = module

  # 寫入配置文件
  def write_ini_file(self):
    with open(self.path,self.module) as f:
      self.cf.write(f)

  # 新增section
  def add_section(self,section):
    self.cf.add_section(section=section)
    self.write_ini_file()

  # 刪除某個 section
  def remove_section(self,section):
    self.cf.remove_section(section=section)
    self.write_ini_file()

  # 刪除某個 section 下的 option
  def remove_option(self,section,option):
    self.cf.remove_option(section=section,option=option)
    self.write_ini_file()

if __name__ == '__main__':
  pass