如何为可调用类创建静态参数属性-Python教程-PHP中文网

如何为可调用类创建静态参数属性

本文介绍了如何利用闭包和 functools.partial 为可调用类创建类似静态参数属性的功能，使得在类初始化时预先绑定部分参数，并在调用时方便地使用这些预设参数。通过示例代码详细展示了实现方法，并解释了其背后的原理，帮助读者理解和应用该技术。

在Python中，有时我们需要为一个可调用类预先设置一些参数，这些参数在类的实例被调用时会默认使用，类似于静态参数。虽然Python本身没有直接提供静态参数属性的概念，但我们可以利用闭包和 functools.partial 来实现类似的效果。

使用 functools.partial 捕获参数

functools.partial 是一个非常有用的工具，它可以创建一个新的可调用对象，该对象在调用时会携带预先绑定的一些参数。这允许我们延迟参数的传递，直到实际调用发生。

以下是一个示例，展示了如何使用 functools.partial 来创建一个携带预设参数的函数：

import functools

def capture_args(func, *args, **kwargs):
    """
    创建一个新的函数，该函数在调用时会携带预先绑定的参数。
    """
    return functools.partial(func, *args, **kwargs)

def test_function(x, y, *, kw1):
    """
    一个简单的测试函数，接受位置参数和关键字参数。
    """
    print(x, y, kw1)

# 使用 capture_args 创建一个新的函数，预先绑定 x=100, y=50, kw1=20
prepared_func = capture_args(test_function, 100, 50, kw1=20)

# 调用 prepared_func，它会自动使用预先绑定的参数
prepared_func()  # 输出: 100 50 20

在这个例子中，capture_args 函数接收一个函数和任意数量的位置参数和关键字参数。它使用 functools.partial 创建一个新的函数 prepared_func，该函数在调用时会自动将 x=100、y=50 和 kw1=20 传递给 test_function。

macOS UI 自动化技能包 —— 用 Python 控制你的 Mac

让 macOS 自动化像说话一样简单 —— 用代码控制你的 Mac，无需学习 AppleScript 系统要求 macOS 14+ (Sonoma) Python 3.9+ 需要开启：辅助功能 + 屏幕录制权限

下载

应用于可调用类

现在，我们可以将这个技术应用于可调用类。假设我们有一个可调用类 Test，我们希望在创建类的实例时预先绑定一些参数，并在调用实例时使用这些参数。

import functools
from typing import Callable

class Test:
    def __init__(self, aux=None):
        self.aux = aux
        self.static_args = {} # 用于存储静态参数

    def func(self):
        print("Hello there")

    def __call__(self, fn: Callable):
        print(f"Aux value: {self.aux}")
        fn()
        print("This is a test class.")

def capture_args(func, *args, **kwargs):
    """
    创建一个新的函数，该函数在调用时会携带预先绑定的参数。
    """
    return functools.partial(func, *args, **kwargs)

# 另一个可调用对象
def g():
    print("This is function g.")

# 创建 Test 类的实例，并预先绑定 aux=1
test = Test(aux=1)

# 直接调用 test 实例
test(g)

在这个例子中，Test 类在初始化时接受一个 aux 参数，并将它存储在实例的属性中。__call__ 方法在被调用时，会打印 aux 的值，然后调用传入的函数 fn。

代码解释:

Test 类: 定义了一个可调用类，它在 __init__ 方法中接受一个可选的 aux 参数。__call__ 方法使其类的实例可以像函数一样被调用。
capture_args 函数: 类似于之前的例子，用于创建一个新的函数，该函数在调用时会携带预先绑定的参数。但在这个例子中没有被直接使用，因为参数在类的初始化阶段已经传入。
实例化和调用: 创建 Test 类的实例 test，并传入 aux=1。然后，直接调用 test(g)，这会触发 __call__ 方法的执行，从而打印 aux 的值并调用函数 g。