Source code for ubelt.util_func

# -*- coding: utf-8 -*-
"""
Helpers for functional programming. Currently this module contains an identify
function that simply returns its own inputs and a function to "inject" another
function into a class instance as a method. This is useful for monkey patching.
"""


[docs]def identity(arg): """ The identity function. Simply returns its inputs. Args: arg (object): some value Returns: object: arg: the same value Example: >>> assert identity(42) == 42 """ return arg
[docs]def inject_method(self, func, name=None): """ Injects a function into an object instance as a bound method The main use case of this function is for monkey patching. While monkey patching is sometimes necessary it should generally be avoided. Thus, we simply remind the developer that there might be a better way. Args: self (object): instance to inject a function into func (func): the function to inject (must contain an arg for self) name (str): name of the method. optional. If not specified the name of the function is used. Example: >>> class Foo(object): >>> def bar(self): >>> return 'bar' >>> def baz(self): >>> return 'baz' >>> self = Foo() >>> assert self.bar() == 'bar' >>> assert not hasattr(self, 'baz') >>> inject_method(self, baz) >>> assert not hasattr(Foo, 'baz'), 'should only change one instance' >>> assert self.baz() == 'baz' >>> inject_method(self, baz, 'bar') >>> assert self.bar() == 'baz' """ # TODO: if func is a bound method we should probably unbind it new_method = func.__get__(self, self.__class__) if name is None: name = func.__name__ setattr(self, name, new_method)