Python使用toml库读写配置文件示例

Python使用toml库读写配置文件示例

# -*- coding: utf-8 -*-
# @Time : 2019-11-18 09:31
# @Author : cxa
# @File : toml_demo.py
# @Software: PyCharm
import toml
import os

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


class FileOperation:
  def __init__(self):
    self.dic = dict()
    self.toml_file_path = os.path.join(BASE_DIR, "config.toml")

  def __add__(self, other):
    self.dic.update(self.other)
    return self.dic

  def write(self):
    mysql_dic = {"user": "root", "password": "Aa1234"}
    mysql2_dic = {"user1": "root", "password2": "Aa1234"}
    mysql_dic.update(mysql2_dic)
    with open(self.toml_file_path, "w", encoding="utf-8") as fs:
      toml.dump(mysql_dic, fs)

  def read(self):
    with open(self.toml_file_path, "r", encoding="utf-8") as fs:
      t_data = toml.load(fs)
    return t_data


if __name__ == '__main__':
  f = FileOperation()
  data = f.read()
  print(data)