根据实例变量给类实例不同的方法

Question

我想创建一个类，其中实例根据其变量具有不同的方法。

例如：

class Spam:
    def __init__(self, eggs):
        self.eggs = eggs

那么，如果 Spam 的 self.eggs 为 5，则它应该将 foo 定义如下：

def foo(self):
    print(self.eggs)

否则

def foo(self, excited):
    if excited:
       print("{}!!!!!".format(self.eggs))
    else:
       print("{}...".format(self.eggs))

目前，我通过在类外部声明两个foo 函数（具有不同名称）并在__init__ 中设置实例作为方法的方法来做到这一点。但是，这会从类定义中删除函数定义，我真的不想这样做。还有其他方法吗？

编辑：实际上下文。

我正在制作一些类来表示图中的节点，其中每个图都可以加权或不加权。 我希望节点有一个link 方法，将另一个节点作为参数，将它们连接在一起。加权图中的节点（即那些self.graph.weighted 是True 的节点）应该有一个link 方法和另一个指定连接边权重的参数。有没有更好的方法来做到这一点？例如，将权重参数设为可选，但存在于所有节点的 link 方法中。

Answer 1

扩展一下 jme 提到的内容：

默认参数方式：

class Node:
    def __init__(self, name):
        self.name = name
        self.links = []

    def link(self, other, weight = None):
        self.links.append((other, weight))

继承方式：

class Node:
    def __init__(self, name):
        self.name = name
        self.links = []

    def link(self, other):
        #one of the following, depending if other methods of this base class will be used in the derived class
        self.links.append((other, None))
        #self.links.append(other)

class WeightedNode(Node):
    def link(self, other, weight):
        self.links.append((other, weight))

Answer 2

你可以使foo只有Spam的一种方法，默认参数excited：

In [49]: class Spam:
    ...:    def __init__(self, eggs):
    ...:        self.eggs = eggs
    ...: 
    ...:    def foo(self, excited=False):
    ...:        if self.eggs == 5:
    ...:            print(self.eggs)
    ...:            return
    ...: 
    ...:        if excited:
    ...:           print("{}!!!!!".format(self.eggs))
    ...:        else:
    ...:           print("{}...".format(self.eggs))

In [50]: Spam(5).foo()
5

In [51]: Spam(6).foo()
6...

In [52]: Spam(6).foo(True)
6!!!!!

Answer 3

这是您需要利用“工厂模式”的地方。也就是说，您需要有 N 个类（并非所有与继承相关，而是与鸭类型相关），这样：

 def create_my_class(arg1, arg2, arg3=None, arg4=None):
   #optional arguments? Ah ha!
   if arg3 is not None and arg4 is not None:
     return MyClass(arg1, arg2, arg3, arg4)
   elif arg3 is not None:
     return AnotherClass(arg1, arg3)
   else:
     return YetAnotherClass(arg1, arg2)

所以你解耦了每个类的实现，而不是它的构造。然后，您可以使用（和替代）此方法来帮助创建不同的类。

   node1 = create_my_class(1, 1)
   node2 = create_my_class(1, 2, node1)
   node3 = create_my_class(1, 3, node1, 15)
   #etc...