Python class_or_instance_method decorator, passes cls and self (if available)

class_or_instance_method.py

from typing import Callable, Type
import functools


class class_or_instance_method:
    func: Callable

    def __init__(self, func: Callable):
        self.func = func

    def __get__(self, instance: object, owner: Type = None):
        if owner is None and instance:
            owner = instance.__class__

        @functools.wraps(self.func)
        def _func(*args, **kwargs):
            return self.func(
                owner,
                instance,
                *args,
                **kwargs,
            )

        return _func
        
# Usage:
# class Foo:
#     @class_or_instance_method
#     def test(cls, self, *args, **kwargs):
#         print(cls)
#         print(self)
#         print(args)
#         print(kwargs)
# Foo.test()
# Foo().test()