Question

我想限制一组选项中的参数。如果调用该函数，则必须将参数限制为几个选项。这就是我现在所拥有的

class GetFileMethod:
     URL = 'url'
     ATTACHMENT = 'attachment'

class MailClient
     def GetFile(self,method)

MailClient.GetFile(GetFileMethod.URL) #works ok, but
MailClient.GetFile("lalala") #should raise an error

有什么建议吗？

Answer 1

def GetFile(self, method):
    if  method not in {'url','attachment'}:
        raise ValueError

我会使GetFileMethod成为MailClient类的一种方法，它会让生活更容易控制输入。

Answer 2

将您的class MainClient更改为： - 您需要在method的命名空间中检查您提供的class GetFileMethod的值，以便： -

GetFileMethod.__dict__.values()


class MailClient:
    def GetFile(self, method):
        if method in GetFileMethod.__dict__.values():
            return 'Yes'
        else:
            return 'No'

Python方法参数限制类

2 个答案: