Nhận lớp mà phương thức xác định

Làm cách nào để lấy lớp đã định nghĩa một phương thức trong Python?

Tôi muốn ví dụ sau để in " __main__.FooClass":

class FooClass:
    def foo_method(self):
        print "foo"

class BarClass(FooClass):
    pass

bar = BarClass()
print get_class_that_defined_method(bar.foo_method)

import inspect

def get_class_that_defined_method(meth):
    for cls in inspect.getmro(meth.im_class):
        if meth.__name__ in cls.__dict__: 
            return cls
    return None

Cảm ơn Sr2222 đã chỉ ra rằng tôi đã thiếu điểm ...

Đây là cách tiếp cận đã sửa chữa giống như của Alex nhưng không yêu cầu nhập bất kỳ thứ gì. Tôi không nghĩ rằng đó là một cải tiến, trừ khi có một hệ thống phân cấp lớn của các lớp kế thừa vì cách tiếp cận này dừng ngay khi tìm thấy lớp xác định, thay vì trả lại toàn bộ kế thừa như getmrohiện tại. Như đã nói, đây là một kịch bản rất khó xảy ra.

def get_class_that_defined_method(method):
    method_name = method.__name__
    if method.__self__:    
        classes = [method.__self__.__class__]
    else:
        #unbound method
        classes = [method.im_class]
    while classes:
        c = classes.pop()
        if method_name in c.__dict__:
            return c
        else:
            classes = list(c.__bases__) + classes
    return None

Và ví dụ:

>>> class A(object):
...     def test(self): pass
>>> class B(A): pass
>>> class C(B): pass
>>> class D(A):
...     def test(self): print 1
>>> class E(D,C): pass

>>> get_class_that_defined_method(A().test)
<class '__main__.A'>
>>> get_class_that_defined_method(A.test)
<class '__main__.A'>
>>> get_class_that_defined_method(B.test)
<class '__main__.A'>
>>> get_class_that_defined_method(C.test)
<class '__main__.A'>
>>> get_class_that_defined_method(D.test)
<class '__main__.D'>
>>> get_class_that_defined_method(E().test)
<class '__main__.D'>
>>> get_class_that_defined_method(E.test)
<class '__main__.D'>
>>> E().test()
1

Giải pháp Alex trả về kết quả tương tự. Miễn là phương pháp tiếp cận Alex có thể được sử dụng, tôi sẽ sử dụng nó thay vì phương pháp này.

Tôi không biết tại sao không ai đưa ra điều này hoặc tại sao câu trả lời hàng đầu có 50 lượt ủng hộ khi nó chậm như địa ngục, nhưng bạn cũng có thể làm như sau:

def get_class_that_defined_method(meth):
    return meth.im_class.__name__

Đối với python 3, tôi tin rằng điều này đã thay đổi và bạn sẽ cần phải xem xét .__qualname__.

Trong Python 3, nếu bạn cần đối tượng lớp thực tế, bạn có thể làm:

import sys
f = Foo.my_function
vars(sys.modules[f.__module__])[f.__qualname__.split('.')[0]]  # Gets Foo object

Nếu hàm có thể thuộc về một lớp lồng nhau, bạn sẽ cần phải lặp lại như sau:

f = Foo.Bar.my_function
vals = vars(sys.modules[f.__module__])
for attr in f.__qualname__.split('.')[:-1]:
    vals = vals[attr]
# vals is now the class Foo.Bar

Tôi bắt đầu làm điều gì đó tương tự, về cơ bản, ý tưởng đang kiểm tra bất cứ khi nào một phương thức trong lớp cơ sở được triển khai hay không trong lớp con. Hóa ra theo cách tôi đã làm ban đầu, tôi không thể phát hiện ra khi nào một lớp trung gian đang thực sự triển khai phương thức này.

Cách giải quyết của tôi cho nó thực sự khá đơn giản; thiết lập một thuộc tính phương thức và kiểm tra sự hiện diện của nó sau đó. Đây là đơn giản hóa toàn bộ:

class A():
    def method(self):
        pass
    method._orig = None # This attribute will be gone once the method is implemented

    def run_method(self, *args, **kwargs):
        if hasattr(self.method, '_orig'):
            raise Exception('method not implemented')
        self.method(*args, **kwargs)

class B(A):
    pass

class C(B):
    def method(self):
        pass

class D(C):
    pass

B().run_method() # ==> Raises Exception: method not implemented
C().run_method() # OK
D().run_method() # OK

CẬP NHẬT: Thực tế cuộc gọi method()từrun_method() (đó không phải là linh hồn sao?) Và nó truyền tất cả các đối số chưa được sửa đổi vào phương thức.

Tái bút: Câu trả lời này không trực tiếp trả lời câu hỏi. IMHO có hai lý do mà người ta muốn biết lớp nào đã định nghĩa một phương thức; đầu tiên là trỏ ngón tay vào một lớp trong mã gỡ lỗi (chẳng hạn như trong xử lý ngoại lệ), và thứ hai là xác định xem phương thức đã được triển khai lại hay chưa (trong đó phương thức là một sơ khai được lập trình viên thực hiện). Câu trả lời này giải quyết trường hợp thứ hai đó theo một cách khác.

Python 3

Giải quyết nó theo một cách rất đơn giản:

str(bar.foo_method).split(" ", 3)[-2]

Điều này cho

'FooClass.foo_method'

Tách trên dấu chấm để lấy lớp và tên hàm riêng biệt