Question

我在Python类中有一个函数，它将接口添加到列表中。

def RegisterAsListener(self, inListener):
    self.__TransitListeners.append(inListener)

这很好，因为一个类只需要从我的接口继承，抓住这个对象，并为所有更新注册自己。

class ITransit():
    def TransitUpdate(self, data):
        raise NotImplementedError("You must define this function.")

（假设我正确地创建了一个接口）

由于我不是这个项目中唯一的一个，我不希望有人用不正确的数据类型调用RegisterAsListener函数。我可以输入代码来检查寄存器函数中的类型，但如果编译器在尝试推送不正确的数据类型时只是对程序员大吼大叫，那么它会更容易。

def RegisterAsListener(self, IListener inListener):
    self.__TransitListeners.append(inListener)

有没有办法做到这一点？

Answer 1

因为我不是这个项目中唯一的人，所以我不想要有人使用不正确的数据类型调用RegisterAsListener函数

彻底记录函数，然后让不正确的参数抛出异常。 RegisterAListener的用户应该能够使用文档来了解函数期望的数据类型，并且 - 如果给出不正确的参数 - 结果异常应该清楚RegisterAListener的用户做错了什么

Answer 2

虽然我强烈建议不要这样做，并且只使用抽象基类（http://docs.python.org/2/library/abc.html）强制执行某些方法，但这是可能的。

以下是有关如何执行此类操作的示例：http://www.artima.com/weblogs/viewpost.jsp?thread=101605

# mm.py
registry = {}                                                                   

class MultiMethod(object):                                                      
    def __init__(self, name):                                                   
        self.name = name                                                        
        self.typemap = {}                                                       
    def __call__(self, *args):                                                  
        types = tuple(arg.__class__ for arg in args) # a generator expression!  
        function = self.typemap.get(types)                                      
        if function is None:                                                    
            raise TypeError("no match")                                         
        return function(*args)                                                  
    def register(self, types, function):                                        
        if types in self.typemap:                                               
            raise TypeError("duplicate registration")                           
        self.typemap[types] = function                                          

def multimethod(*types):                                                        
    def register(function):                                                     
        function = getattr(function, "__lastreg__", function)                   
        name = function.__name__                                                
        mm = registry.get(name)                                                 
        if mm is None:                                                          
            mm = registry[name] = MultiMethod(name)                             
        mm.register(types, function)                                            
        mm.__lastreg__ = function                                               
        return mm                                                               
    return register                                                             

    if hasattr(function, "__lastreg__"):                                        
        function = function.__lastreg__

使用它的代码：

import mm                                                                       

@mm.multimethod(int)                                                            
def spam(a):                                                                    
    print 'Calling the int method'                                              
    print '%s: %r' % (type(a), a)                                               

@mm.multimethod(float)                                                          
def spam(a):                                                                    
    print 'Calling the float method'                                            
    print '%s: %r' % (type(a), a)                                               

spam(5)                                                                         
spam(5.0)

示例输出：

Calling the int method
<type 'int'>: 5
Calling the float method
<type 'float'>: 5.0

在Python中强制使用函数参数类型？

2 个答案: