python - 如何在我的函数中替换“eval”？

Question

eval我很清楚在 Python 中使用这不是一个好习惯。这是我的代码：

from hashlib import * # added for clarification
def get_hash(self):
        if self.hash_type in ['md5', 'sha1', 'sha224', 'sha256', 'sha384', 'sha512']:
            data = eval(self.hash_type) # <--- how can I replace this?

我听说过这个setattr功能，但老实说我不知道​​如何实现它。

是否也可以通过使用该功能来做到这一点map？

请注意，'md5', 'sha1', 'sha224'etc 是局部变量。当我eval在其中一个上使用该功能时，它们就变成了功能。

Answer 1

您可以使用getattr：

import hashlib
if self.hash_type in ['md5', 'sha1', 'sha224', 'sha256', 'sha384', 'sha512']:
     func = getattr(hashlib, self.hash_type)

Answer 2

为什么不将所有信息存储在字典中？例如：

self.hash_types = {
    'md5': md5,
    'sha1': sha1,
    ...
}

然后在get_hash：

def get_hash(self):
    if self.hash_type in self.hash_types:
        data = self.hash_types[self.hash_type]

或者，

def get_hash(self):
    data = self.hash_types.get(self.hash_type, self.DEFAULT)