在 Python 中读取 YAML 配置文件

Question

我正在尝试读取 YAML 配置文件并将其显示到终端。现在我想尝试检查 YAML 文件中的数据库（db）是否不是 Sqlite 或 Postgres，然后会引发异常，但我不知道如何。我试过但失败了，我做错了什么？

我的test.yaml文件：

database: 

dbopt:

   host: bvcbvcbvcb.com

   port: 5432

   dbname: db1

   user: username

   password: 1234

   client_encoding: utf-8

   connect_timeout: 60

   sslmode: none

query:

   select * from manufacturing_product

我的代码：

# process_yaml.py file`

import yaml

with open(r'D:\Python\test.yaml') as file:

    # The FullLoader parameter handles the conversion from YAML

    # scalar values to Python the dictionary format

    data = yaml.full_load(file)

    for item, doc in data.items():

        print(item, ":", doc)

    def __init__(self, dbconf):

        self._dbconf = dict(dbconf)

        # checking for database type

        dbtype = self.get_db_type()

        if dbtype != 'sqlite' and dbtype != 'postgres':

            raise exceptions.InvalidConfigError(

                'E01001', 'Invalid database type, should be sqlite or postgres.')

        else:

            self.dbtype = dbtype

我的程序仍然无法捕获异常。我的终端：

database: 

dbopt:

   host: 

   port: 5432

   dbname: db1

   user: username

   password: 1234

   client_encoding: utf-8

   connect_timeout: 60

   sslmode: none

query:

   select * from manufacturing_product

Answer 1

您的代码中缺少几部分，并且您的函数__init__永远不会被调用。您可能从一些带有类的示例中复制了它，该类也有一个方法get_db_type()。class InvalidConfigError(Exception):    passclass DB:    def __init__(self, dbconf):        self._dbconf = dict(dbconf)        # checking for database type        dbtype = self.get_db_type()        if dbtype != 'sqlite' and dbtype != 'postgres':            raise InvalidConfigError(                'E01001', 'Invalid database type, should be sqlite or postgres.')        else:            self.dbtype = dbtype    def get_db_type(self):        return self._dbconf['db']with open('test.yaml') as file:    data = yaml.full_load(file)    for item, doc in data.items():        print(item, ":", doc)    db = DB(data)哪个打印：db : mysqldbopt : {'host': 'bvcbvcbvcb.com', 'port': 5432, 'dbname': 'db1', 'user': 'username', 'password': '1234', 'client_encoding': 'utf-8', 'connect_timeout': 60, 'sslmode': 'none'}query : select * from manufacturing_product然后给出：init raise InvalidConfigError( main .InvalidConfigError: ('E01001', '无效的数据库类型，应该是 sqlite 或 postgres。') 进程错误命令 '['ryd', '--force', 'so-60160957.ryd']'返回非零退出状态 1。评论# The FullLoader parameter handles the conversion from YAML# scalar values to Python the dictionary format是相当的标记。FullLoader 解析 YAML 并尝试将所有节点实例化为 Python 对象：YAML 映射到 dict，YAML 序列到列表，以及作为 Python 类型（字符串、整数、浮点数、布尔值等）的标量的 YAML 节点