尝试让 ref_computed 能同时作用到普通函数和类定义的对象方法上

[CrystalWindSnake]

下面代码示例的所有需要导入的模块:

from typing import Any, Callable, TypeVar, cast, Generic
from nicegui import ui

from ex4nicegui import ref_computed
from weakref import WeakKeyDictionary
from functools import partial
from ex4nicegui.utils.signals import ReadonlyRef
import types

当前实现，已经做到 ref_computed 作用到普通函数上

from ex4nicegui import ref_computed

@ref_computed
def _():
     ...

但是不能作用到类中定义的对象方法上

class Model:
    @ref_computed
    def test(self):
          pass

下面是实现能作用到类定义对象方法上:

_T_var_get_return = TypeVar("_T_var_get_return")


class desc_ref_computed(Generic[_T_var_get_return]):
    __isabstractmethod__: bool

    def __init__(
        self,
        fget: Callable[[Any], _T_var_get_return],
    ) -> None:
        self._fget = fget
        self.__instance_map: WeakKeyDictionary[
            object, ReadonlyRef
        ] = WeakKeyDictionary()

    def __get_computed(self, instance):
        if instance not in self.__instance_map:
            cp = ref_computed(partial(self._fget, instance))
            self.__instance_map[instance] = cp

        return self.__instance_map[instance]

    def __get__(self, __instance: Any, __owner: type | None = None) :
        return cast(ReadonlyRef[_T_var_get_return], self.__get_computed(__instance))

但如果把两者融合，无法做到很好的 typing hit，如下是简单融合两者的实现:

_T = TypeVar("_T")


def sim_ref_computed(fn: Callable[[], _T]):
    is_class_define_method = "." in fn.__qualname__ and (
        isinstance(fn, types.FunctionType)
    )

    if not is_class_define_method:
        return ref_computed(fn)

    return cast(desc_ref_computed[_T], desc_ref_computed(fn))

[CrystalWindSnake]

#95