将对象方法作为对象参数值传递

Question

我创建了一个BetterMarker 类，它继承自folium.Marker，这是一个地理点对象，可以添加到生成Leaflet HTML 地图的folium.Map 对象。

import folium, datetime

class BetterMarker(folium.Marker):

    def __init__(self, latitude, longitude, **kwargs): 
        super().__init__([latitude, longitude], **kwargs)
        self.latitude = latitude

    def north_or_south(self):
        if self.latitude > 0:
            return "North"
        else:
            return "South"

我现在可以初始化BetterMarker 并将其实例添加到folium.Map 对象：

mymap = folium.Map() # Create Map object
bm = BetterMarker(30, 40, popup = "Hi there")
bm.add_to(mymap) # Add BetterMarker object to Map
mymap.save("mymap.html") # Save the map with the marker into HTML

上面的代码可以很好地生成以下地图：

但是，我不想在bm = BetterMarker(30, 40, popup = "Hi there") 中使用字符串Hi there，而是使用bm.north_or_south()：

bm = BetterMarker(30, 40, popup = bm.north_or_south())

当然，以上行不通。下面的代码可以工作，但这意味着初始化类两次：

bm = BetterMarker(30, 40)
bm = BetterMarker(30, 40, popup = bm.north_or_south())

这样做的明智方法是什么？我做错了什么？

Answer 1

根据评论更新

我的第一次尝试并没有完全理解要求。我认为其他一些答案是正确的。这是一种可行的方法：

class BetterMarker(folium.Marker):

    def __init__(self, latitude, longitude, **kwargs):
        if kwargs['popup_method']:
            kwargs['popup'] = kwargs['popup_method'](latitude, longitude, **kwargs)
            del kwargs['popup_method']
        super().__init__([latitude, longitude], **kwargs)

    @staticmethod
    def quadrant(latitude, longitude, **kwargs):
        if latitude > 0 and longitude > 0:
            return "North East"
        elif latitude > 0 and longitude <= 0:
            return "North West"
        elif latitude <= 0 and longitude > 0:
            return "South East"
        else:
            return "South West"

至于调用，重要的是只传递方法，不要通过括号来调用它：

bm = BetterMarker(-30, 40, popup_method=BetterMarker.quadrant)

请注意，我创建了一个名为 popup_method 的新关键字参数，因为 popup 表示文本。此外，我在弹出方法的签名中添加了经度和 kwargs，因此这些方法可以访问调用者传递的所有内容。

Answer 2

与有问题的 cmets 有关。像这样：

class Foo:
    def __init__(self, *args, **kwargs):
        self.alert = kwargs['alert']

    def print_alert(self):
        print(self.alert())

class Bar(Foo):
    def __init__(self, *args, **kwargs):
        if "alert" not in kwargs:
            kwargs['alert'] = self.custom_alert() # Notice here.

        super().__init__(*args, **kwargs)

    def custom_alert(self):
        return "Custom Alert"

Bar().print_alert() # >>> Custom Alert

---

编辑

根据下面的评论，您可以简单地将方法定义为@staticmethod：

class Foo:
    def __init__(self, *args, **kwargs):
        self.alert = kwargs["alert"]

    def print_alert(self):
        print(self.alert()) # Notice we DO call the variable.


class Bar(Foo):
    def __init__(self, *args, **kwargs):
        if "alert" not in kwargs:
            kwargs["alert"] = self.default_alert

        super().__init__(*args, **kwargs)

    @staticmethod
    def default_alert():
        return "Custom Alert!"

    @staticmethod
    def random_alert():
        return "Random Alert!"


a = Bar()
a.print_alert()

b = Bar(alert=Bar.random_alert)  # Notice we do not CALL the method.
b.print_alert()

输出：

default
rand

Answer 3

您可以尝试将 north_or_south 设为静态方法

@staticmethod
def north_or_south(latitude):
    if latitude > 0:
        return "North"
    else:
        return "South"

这使得它可以从一个对象或一个对象类运行，但函数本身不知道它正在运行的对象或类，所以你不能再拥有

self.north_or_south()

因为函数对对象一无所知。相反，我们将纬度作为参数传递

self.north_or_south(self.latitude)

瞧，它有效。代码有点长，但函数不再需要从哪里运行的对象。因此，在您的问题中，您只需

bm = BetterMarker(30, 40, popup = BetterMarker.north_or_south(40))

Answer 4

[编辑] 也许只是将弹出生成函数作为关键字参数传递？

import folium, datetime

class CustomPopup:
       @staticmethod
       def north_or_south(latitude, longtitude, **kwargs):
        if latitude > 0:
            return "North"
        else:
            return "South" 

        ...other methods

class BetterMarker(folium.Marker):

    def __init__(self, latitude, longitude, **kwargs): 
        self.latitude = latitude
        if 'msg' in kwargs:
             kwargs['popup'] = kwargs['msg'](latitude, longtitude, **kwargs)
        super().__init__([latitude, longitude], **kwargs)


mymap = folium.Map() # Create Map object
bm = BetterMarker(30, 40, msg=CustomPopup.north_or_south)
bm.add_to(mymap) # Add BetterMarker object to Map
mymap.save("mymap.html")