Python中如何覆盖重载函数？Squish框架installEventHandler覆盖疑问

I get it, dealing with overloaded methods in Python—especially from a framework like Squish—can be tricky since Python doesn’t natively support method overloading like statically typed languages. The issue with using *args is that Squish’s underlying overload resolution looks for specific parameter counts and types—not a tuple of arguments. Here’s a straightforward way to build your MyinstallEventHandler wrapper that handles all three overloads:

Solution: Manual Parameter Matching #

Since Python lacks built-in overload resolution, we can manually check the number and type of arguments passed to route calls to the correct Squish installEventHandler overload. This also lets you add custom logic (like logging or validation) before/after calling the original method.

def MyinstallEventHandler(*args):
    # Handle overload 1: installEventHandler(eventName, handlerFunctionName)
    if len(args) == 2:
        event_name, handler_func = args
        if isinstance(event_name, str) and callable(handler_func):
            # Add your custom logic here (logging, validation, etc.)
            print(f"Registering handler for event: {event_name}")
            return installEventHandler(event_name, handler_func)
    
    # Handle overload 2: installEventHandler(className, eventName, handlerFunctionName)
    elif len(args) == 3:
        arg1, arg2, arg3 = args
        # Check if first two args are strings, third is a handler
        if isinstance(arg1, str) and isinstance(arg2, str) and callable(arg3):
            class_name, event_name, handler_func = args
            print(f"Registering handler for class {class_name}, event {event_name}")
            return installEventHandler(class_name, event_name, handler_func)
        # Handle overload 3: installEventHandler(object, eventName, handlerFunctionName)
        elif not isinstance(arg1, str) and isinstance(arg2, str) and callable(arg3):
            target_obj, event_name, handler_func = args
            print(f"Registering handler for object {target_obj}, event {event_name}")
            return installEventHandler(target_obj, event_name, handler_func)
    
    # If no overload matches, raise a clear error
    raise TypeError(
        f"Invalid arguments for MyinstallEventHandler. Expected:\n"
        "- (eventName: str, handler: callable)\n"
        "- (className: str, eventName: str, handler: callable)\n"
        "- (object: Any, eventName: str, handler: callable)\n"
        f"Got: {args}"
    )

How It Works #

1. Check argument count first: The three overloads have 2 or 3 arguments, so we split logic based on len(args).
2. Validate types: For each argument count, we check if the types match what Squish expects for the corresponding overload:
   • 2 arguments: First is a string (event name), second is a callable handler.
   • 3 arguments: Either two strings + handler (class + event) or an object + string + handler (target object + event).
3. Custom logic: You can insert any pre/post processing (like logging input or modifying arguments) before calling the original installEventHandler method.

Example Usage #

# Overload 1: Register handler for a global event
def on_button_clicked():
    print("Button clicked!")
MyinstallEventHandler("buttonClicked", on_button_clicked)

# Overload 2: Register handler for all objects of a class
def on_pushbutton_clicked():
    print("QPushButton clicked!")
MyinstallEventHandler("QPushButton", "clicked", on_pushbutton_clicked)

# Overload 3: Register handler for a specific object
target_btn = waitForObject(names.my_window_push_button)
def on_specific_btn_clicked():
    print("Specific button clicked!")
MyinstallEventHandler(target_btn, "clicked", on_specific_btn_clicked)

Why *args Alone Doesn’t Work #

Squish’s overload resolution relies on receiving distinct parameters, not a single tuple. When you pass *args, Squish sees a single tuple argument instead of the separate parameters it expects for each overload, so it can’t match any of the defined methods. Manual type/count checking avoids this by unpacking the arguments and passing them in the format Squish requires.

内容的提问来源于stack exchange，提问作者Amrita