通过 __init__ 传递 configparser.ConfigParser() object？

Question

我目前正在开发一个 dns-enumeration 项目，该项目向各种 API 发送请求。 其中一些 API 需要一个 API 密钥，我在 config.ini 文件中提供了它。 在我当前的设置中，我使用 configparser 将不同的值读入 object，因此我可以在需要时访问 object。 现在，当我尝试实现类似 class 结构的东西时，我想在父 class 的init中读入一次配置文件，这样我就可以从 ZA2F2ED4F8EBC2CBB4C21A29DC4 继承每个需要 API 密钥的工具。

现在设置看起来像这样：

class Source:
    def __init__(self):
        config = configparser.ConfigParser()
        config.read('./config.ini')
        self.config = config

class BinaryEdge(Source):
    def __init__(self):
        super().__init__()

    def query(self, domain, dnsprobe):
        api_key = self.config['BINARYEDGE']['API-KEY']
        url = 'https://api.binaryedge.io/v2/query/domains/subdomain/' + domain
        fqdns = []
        ...

据我了解，如果我启动一个新的 BinaryEdge-Instance，例如这样：

if __name__ == "__main__":
    BinaryEdge = BinaryEdge()
    print(BinaryEdge.query("heise.de", False))

从技术上讲，它应该将配置文件读入 object 并将其传递给新创建的 object，因此我可以通过 self.config 访问它，如下所示：

 def query(self, domain, dnsprobe):
        api_key = self.config['BINARYEDGE']['API-KEY']
        url = 'https://api.binaryedge.io/v2/query/domains/subdomain/' + domain
        fqdns = []
        ...

但是当我调试这个设置时，配置 object 保持默认（并且因此为空），这显然会直接导致一个关键错误：

File "/usr/lib64/python3.9/configparser.py", line 960, in __getitem__
    raise KeyError(key)
KeyError: 'BINARYEDGE'

由于我在 python 编程方面不如我想的那样好，我正在努力自己解决这个错误，并感谢任何先进的输入。

Answer 1

在收到@Jakub Szlaur 的意见后，我自己想通了：

我的文件路径指向了错误的文件夹，因此从未到达 config.ini 文件。

更改后：

config.read('./config.ini')

至

config.read('$HOME/$PROJECT_PATH/config.ini')

它按预期工作。

我还根据“更好的代码样式”的评论更改了“源”类：

class Source:
    def __init__(self):
        self.config = self.readconfig('../config.ini')
        
    def readconfig(self, filename):
        config = configparser.ConfigParser()
        config.read(filename)
        return config

谢谢您的帮助; ;-)

Answer 2

代码看起来应该可以工作（我找不到任何错误）。 尝试检查您的配置文件，看看是否真的有这样的关键。

但是关于代码本身，我建议更改几件事。

首先，虽然读取配置是Source初始化的一部分，但最好将其设为 function，然后调用 function：

class Source:
    def __init__(self):
        self.config = self.readconfig("./config.ini")

    def readconfig(self, filename):
        config = configparser.ConfigParser()
        config.read(filename)
        return config

永远不要这样做： BinaryEdge = BinaryEdge() 。 如果您想创建BinaryEdge的另一个实例，它将改为调用 BinaryEdge 变量。 给它取个不同的名字。