如何为 Python 原生对象添加函数 - V2EX
V2EX = way to explore
V2EX 是一个关于分享和探索的地方
现在注册
已注册用户请  登录
推荐学习书目
Learn Python the Hard Way
Python Sites
PyPI - Python Package Index
http://diveintopython.org/toc/index.html
Pocoo
值得关注的项目
PyPy
Celery
Jinja2
Read the Docs
gevent
pyenv
virtualenv
Stackless Python
Beautiful Soup
结巴中文分词
Green Unicorn
Sentry
Shovel
Pyflakes
pytest
Python 编程
pep8 Checker
Styles
PEP 8
Google Python Style Guide
Code Style from The Hitchhiker's Guide
lingeo
V2EX    Python

如何为 Python 原生对象添加函数

  •  
  •   lingeo 2024-01-19 09:34:30 +08:00 3032 次点击
    这是一个创建于 632 天前的主题,其中的信息可能已经有所发展或是发生改变。

    举个例子,我实现了一个过滤出字符串中所有数字的功能,函数名为filter_number,我想要附加到 str 对象上,这样调用函数可以写成"abc123abc".filter_number()
    我目前知道的做法就是新建对象继承 str ,但是调用前还是要手动转一遍s = CustomStr("abc123abc"),这就有点脱裤子放屁了。
    不知道这个需求能不能实现,最好还要能被 ide 感知,编写时能够提示出来。

    第 1 条附言    2024-01-19 13:41:16 +08:00
    看样子应该是不行,初衷也只是为了调用函数时候能方便点,大部分函数功能是对字节和字符串做处理。
    既然 Python 没有相关支持,那也没必要纠结下去,毕竟只是个锦上添花的操作。
    #11 发的项目,名字就叫禁果,python 刚好还是条蛇。
    26 条回复    2024-01-22 09:59:08 +08:00
    abersheeran
        1
    abersheeran  
       2024-01-19 09:45:50 +08:00   1
    省流:做不了,IDE 感知不到。
    chaleaochexist
        2
    chaleaochexist  
       2024-01-19 09:48:16 +08:00
    str 不知道
    我知道 dict,
    有一个 Userdict 是官方标准库提供的, 专门给用户干这事儿的.
    capbone
        3
    capbone  
       2024-01-19 10:02:10 +08:00
    ```python
    import builtins


    class _CustomStr(str):
    def filter_number(self):
    pass


    builtins.str = _CustomStr
    ```

    对于显示声明为 str 的对象有效,但是对字面值无效:

    ```
    str("abc123abc").filter_number() # OK
    ```

    ```
    "abc123abc".filter_number()

    AttributeError: 'str' object has no attribute 'filter_number'
    ```
    capbone
        4
    capbone  
       2024-01-19 10:03:33 +08:00
    不知道 v2 code block 的语法,抱歉……
    McZoden
        5
    McZoden  
       2024-01-19 10:03:37 +08:00
    同关注
    C# 有 extension method 的语法糖,虽然实现方法也是类似于,新定义一个 Static Class ,包含一个 Static Method ,但调用方法就直接是用原生类型的实例去调用了
    Python 可能针对 str ,这种内置的对象,不行,没有这样的语法糖
    mightybruce
        6
    mightybruce  
       2024-01-19 10:05:46 +08:00
    在 Python 中,不能直接在 str 类型上添加自定义方法。但是,可以通过创建一个新的类来继承 str 类,并在新类中添加自定义方法。以下是一个示例:


    class MyStr(str):
    def my_custom_method(self):
    # 在这里实现你的自定义方法
    pass

    # 使用自定义类
    my_str = MyStr("Hello, world!")
    my_str.my_custom_method()

    或者修改 str 的 built-in 方法

    一些常见对象的魔术方法
    __len__, __getitem__, __contains__, __add__, __mul__, __mod__, __format__, __eq__, __ne__, __lt__, __le__, __gt__, __ge__, __hash__, __repr__, __str__, __bytes__, __bool__, __int__, __float__, __complex__, __index__, __trunc__, __floor__, __ceil__, __round__, __enter__, __exit__, __new__, __init__, __del__, __call__, __getattr__, __setattr__, __delattr__, __dir__, __getattribute__, __setitem__, __delitem__, __iter__, __next__, __reversed__, __len__, __contains__, __add__, __mul__, __rmul__, __mod__, __format__, __eq__, __ne__, __lt__, __le__, __gt__, __ge__, __hash__, __repr__, __str__, __bytes__, __bool__, __int__, __float__, __complex__, __index__, __trunc__, __floor__, __ceil__, __round__, __enter__, __exit__, __new__, __init__, __del__, __call__, __getattr__, __setattr__, __delattr__, __dir__, __getattribute__, __setitem__, __delitem__, __iter__, __next__, __reversed__, __contains__, __add__, __mul__, __rmul__, __mod__, __format__, __eq__, __ne__, __lt__, __le__, __gt__, __ge__, __hash__, __repr__, __str__, __bytes__, __bool__, __int__, __float__, __complex__, __index__, __trunc__, __floor__, __ceil__, __round__, __enter__, __exit__, __new__, __init__, __del__, __call__, __getattr__, __setattr__, __delattr__, __dir__, __getattribute__, __setitem__, __delitem__, __iter__, __next__, __reversed__, __len__,
    CaptainD
        7
    CaptainD  
       2024-01-19 10:08:39 +08:00
    我自己的理解是做不到,不知道有没有什么奇淫巧技

    态添加函数可以有两种方式做到
    1. 继承
    2. 修改对象的内置属性

    而字符串是不可变的,2 做不到,只能是 1 ,也就是你说的“新建对象继承 str”

    字典是可以使用 2 的,因为字典是可变对象,但 UserDict 其实也是用继承
    kmyq
        8
    kmyq  
       2024-01-19 10:11:24 +08:00
    昨天刚问 Chatgpt 这个问题,已声明的倒是可以
    --
    在 Python 中,您可以使用 types.MethodType 将一个函数绑定到一个类的实例上。这允许您只为具体的实例添加函数,而不会更改整个类的定义。以下是一个示例:

    python
    import types

    class MyClass:
    def __init__(self, value):
    self.value = value

    # 定义一个将要绑定的函数
    def print_value(self):
    print(self.value)

    # 创建一个 MyClass 的实例
    obj = MyClass(10)

    # 将函数绑定到这个实例
    obj.print_value = types.MethodType(print_value, obj)

    # 调用新绑定的函数
    obj.print_value()
    在上面的示例中,我们在 MyClass 的实例 obj 上绑定了 print_value 函数。具体细节如下:

    首先,我们定义了一个 MyClass 。该类包含了一个__init__构造函数,以及我们希望将其绑定的函数。
    我们定义了 print_value 函数,该函数接受一个参数 self ,它将打印 self.value 。
    然后,我们创建了 MyClass 的实例 obj ,并将要绑定的函数 print_value 绑定到了这个具体的实例上。
    最后,我们调用新绑定的函数:obj.print_value()。
    这样,我们就通过函数式编程功能将一个函数绑定到了 MyClass 的实例上。
    zzhaolei
        9
    zzhaolei  
       2024-01-19 10:15:22 +08:00
    没有语言能无副作用的直接为内置类型添加方法。如果你为 str 添加了方法 a ,lib b 也为 str 添加了方法 a 和 b ,那么调用 str.a() 的时候是调用你的 a 还是 lib b 的 a ?而且你直接调用 str.b() 的时候是给你提示 lib b 定义的 b 还是给你报错?这复杂度直接爆炸。

    现在的通用做法是,如果你要给 str 添加方法,那么你就定义一个新的类型,然后在自己定义的类型上定义方法。

    python 可以 class MyStr(str) 继承 str 。
    mightybruce
        10
    mightybruce  
       2024-01-19 10:18:19 +08:00
    给内置类型添加是不行的,但是搞一个类模拟函数,给函数添加是可以的

    class NamedFunction:
    def __init__(self, name, f):
    self.f = f
    self.name = name

    def __call__(self, *args, **kwargs):
    return self.f(*args, **kwargs)

    def __str__(self):
    return self.name


    f = NamedFunction("lambda: 'blah'", lambda: 'blah')
    print(f())
    print(f)
    vok2aDe12AsWDirE
        11
    vok2aDe12AsWDirE  
       2024-01-19 10:25:53 +08:00
    c2const
        12
    c2const  
       2024-01-19 10:52:54 +08:00
    或许可以换成 C++语言 :)
    Alias4ck
        13
    Alias4ck  
       2024-01-19 11:19:05 +08:00
    xy 问题
    lolizeppelin
        14
    lolizeppelin  
       2024-01-19 11:59:35 +08:00   2
    我敢打赌 楼主原来是写 js 的
    msg7086
        15
    msg7086  
       2024-01-19 12:57:41 +08:00
    你这 Python 怎么写得一股 Ruby 味啊(
    bybyte
        16
    bybyte  
       2024-01-19 13:08:04 +08:00
    修改 py 源码?
    atuocn
        17
    atuocn  
       2024-01-19 13:33:01 +08:00
    参考 8 楼,依稀记得 meta class 可以做到
    MiketsuSmasher
        18
    MiketsuSmasher  
       2024-01-19 13:36:02 +08:00 via Android
    标准库里提供了一个 collections.UserString ,专门方便你干这种事情的
    codehz
        19
    codehz  
       2024-01-19 13:47:03 +08:00
    https://github.com/xpodev/extype
    也不是完全不行,用原生扩展包里改
    bianhui
        20
    bianhui  
    2024-01-19 13:49:45 +08:00
    随便定义一个类型继承 str,写个空方法
    然后再使用的试试
    a:自定义类型 = ""或者
    typing 下 cast
    a = cast(a,自定义类型)
    编辑器就可以自己识别了
    frostming
        21
    frostming  
       2024-01-19 13:51:19 +08:00
    如 #1 所说,没用,IDE 感知不到,但你非要骚操作的话:

    import gc

    d = gc.get_referents(str.__dict__)[0]
    d['show'] = lambda self: 'hello? are you okay?'
    'foo'.show()
    # hello? are you okay?
    listenerri
        22
    listenerri  
       2024-01-19 13:52:25 +08:00
    不要魔改基础库,你改了用着爽了,别人怎么维护

    虽然此例能实现的话目前影响不大,但应该要以小见大,很多深坑都是一点点堆出来的
    forgottencoast
        23
    forgottencoast  
       2024-01-19 22:25:03 +08:00
    @zzhaolei
    看#5, #14
    zzhaolei
        24
    zzhaolei  
       2024-01-20 11:18:51 +08:00
    @forgottencoast #23 学到了。突然想起来这种有 vm 的语言本身是超级动态的
    351994
        25
    351994  
       2024-01-22 09:56:50 +08:00
    可以实现,aHR0cHM6Ly9ibG9nLnplYmVkeS5jb20vcG9zdC82Y2YyZTIwOS5odG1s
    参考这里的,base64
    351994
        26
    351994  
       2024-01-22 09:59:08 +08:00   1
    是可以实现的,
    import ctypes


    class PyType(ctypes.Structure):
    pass


    class PyObject(ctypes.Structure):
    Py_ssize_t = (
    ctypes.c_int64 if ctypes.sizeof(ctypes.c_void_p) == 8 else ctypes.c_int32
    )
    _fields_ = [
    ("ob_refcnt", Py_ssize_t),
    ("ob_type", ctypes.POINTER(PyType)),
    ]


    class PyTypeObject(PyObject):
    _fields_ = [
    ("dict", ctypes.POINTER(PyObject))
    ]


    def inject(class_, function_name):
    def _(function):
    name_, dict_ = class_.__name__, class_.__dict__
    proxy_dict = PyTypeObject.from_address(id(dict_))
    namespace = {}
    ctypes.pythonapi.PyDict_SetItem(
    ctypes.py_object(namespace),
    ctypes.py_object(name_),
    proxy_dict.dict
    )
    namespace[name_][function_name] = function

    return _

    可以参考#25 中的内容,发不了网址,base64 了一下
    关于     帮助文档     自助推广系统     博客     API     FAQ     Solana     2713 人在线   最高记录 6679       Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 27ms UTC 12:10 PVG 20:10 LAX 05:10 JFK 08:10
    Do have faith in what you're doing.
    ubao snddm index pchome yahoo rakuten mypaper meadowduck bidyahoo youbao zxmzxm asda bnvcg cvbfg dfscv mmhjk xxddc yybgb zznbn ccubao uaitu acv GXCV ET GDG YH FG BCVB FJFH CBRE CBC GDG ET54 WRWR RWER WREW WRWER RWER SDG EW SF DSFSF fbbs ubao fhd dfg ewr dg df ewwr ewwr et ruyut utut dfg fgd gdfgt etg dfgt dfgd ert4 gd fgg wr 235 wer3 we vsdf sdf gdf ert xcv sdf rwer hfd dfg cvb rwf afb dfh jgh bmn lgh rty gfds cxv xcv xcs vdas fdf fgd cv sdf tert sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf shasha9178 shasha9178 shasha9178 shasha9178 shasha9178 liflif2 liflif2 liflif2 liflif2 liflif2 liblib3 liblib3 liblib3 liblib3 liblib3 zhazha444 zhazha444 zhazha444 zhazha444 zhazha444 dende5 dende denden denden2 denden21 fenfen9 fenf619 fen619 fenfe9 fe619 sdf sdf sdf sdf sdf zhazh90 zhazh0 zhaa50 zha90 zh590 zho zhoz zhozh zhozho zhozho2 lislis lls95 lili95 lils5 liss9 sdf0ty987 sdft876 sdft9876 sdf09876 sd0t9876 sdf0ty98 sdf0976 sdf0ty986 sdf0ty96 sdf0t76 sdf0876 df0ty98 sf0t876 sd0ty76 sdy76 sdf76 sdf0t76 sdf0ty9 sdf0ty98 sdf0ty987 sdf0ty98 sdf6676 sdf876 sd876 sd876 sdf6 sdf6 sdf9876 sdf0t sdf06 sdf0ty9776 sdf0ty9776 sdf0ty76 sdf8876 sdf0t sd6 sdf06 s688876 sd688 sdf86