Let's say that I have a class defined in moduleA.py which I want to add a method to, using some sort of loader method that takes a the name of a second module and the method defined there that should be bound
class ClassA(object):
def __init__(self,config):
super(ClassA, self).__init__()
self.a = 1
self.b = 2
self.meth1 = self. bind_method(config)
def bind_method(self,config):
# load method
<return method defined in config as a str 'moduleB.meth2'>
def calling_method():
return self.meth1()
where the method defined in moduleB.py looks something like:
def meth2(self):
return self.a + self.b
The point being that I want to be able to write meth2 to be able to access class variables of ClassA once it is bound. This way, when you would have something like:
from moduleA import ClassA
A = ClassA()
aout = A.calling_method()
Calling A.calling_method() properly calls the method defined in moduleB.py.
I've seen this sort of binding done in answers on SO after ClassA is instantiated using types.MethodType, but I haven't been able to dig up how to bind inside the class definition so that it is done internally when the class is instantiated.
Any suggestions on what should go in the bind_method method would be much appreciated.
4 Answers 4
import sys
import types
def getobj(astr):
"""
getobj('scipy.stats.stats') returns the associated module
getobj('scipy.stats.stats.chisquare') returns the associated function
"""
try:
return globals()[astr]
except KeyError:
try:
return __import__(astr, fromlist=[''])
except ImportError:
modname, _, basename = astr.rpartition('.')
if modname:
mod = getobj(modname)
return getattr(mod, basename)
else:
raise
class ClassA(object):
def __init__(self, methpath):
super(ClassA, self).__init__()
self.a = 1
self.b = 2
self.meth1 = types.MethodType(getobj(methpath), self)
a = ClassA('moduleB.meth2')
print(a.meth1())
# 3
5 Comments
meth1.im_class. I was working on a similar solution that involved creating a classmethod like so: def bind_method(cls, instance, func), that returns a MethodType. That seemed like the most elegant way to ensure that the class gets bound correctly as well as the instance....im_class but didn't got it to work correctly. I would be interested in seeing your solution as welltypes.Method(<obj>,self)), but I was botching my version of str_to_obj. I need to work through your implementation of that helper method, since there are some things in there that I don't immediately understand.meth1.im_class by supplying a third argument to types.MethodType: types.MethodType(str_to_obj(methpath), self, type(self)). However, Python3 removes the third argument from types.MethodType, and does not check that the instance is of the right class. So for the sake of Python3 compatibility, I thought it best to drop the third argument.Skipping the config stuff which wasn't clear to me, the binding itself would look like this:
from moduleB import meth2
ClassA.meth1 = meth2
The important part is that you're binding to the class, not to an instance. This way if you call meth1 on an instance, it will automatically receive the instance as the first argument.
2 Comments
Since meth2() is a function, it is a descriptor and you can bind it by calling the __get__() method.
def meth2(self):
return self.a + self.b
class ClassA(object):
def __init__(self,config):
super(ClassA, self).__init__()
self.a = 1
self.b = 2
self.meth1 = config.__get__(self, ClassA)
c = ClassA(meth2)
print c.meth1() #correctly prints 3
Comments
There's actually a much simpler way to do this:
class ClassA(object):
def __init__(self,config):
super(ClassA, self).__init__()
self.a = 1
self.b = 2
from moduleB import meth2 as meth1
def calling_method():
return self.meth1()
configplay here? AsClassAis currently defined, the third line inmoduleB.pywould cause an error...configis maybe a dictionary that has an item defining where the method to be bound is located, or something like that. All that matters is that is specifies the module and method therein that should be added toClassA