Làm cách nào để có danh sách tất cả các lớp trong mô-đun hiện tại bằng Python?

Question

Tôi đã nhìn thấy rất nhiều ví dụ về những người giải nén tất cả các lớp từ một mô-đun, thường cái gì đó như:

# foo.py 
class Foo: 
    pass 

# test.py 
import inspect 
import foo 

for name, obj in inspect.getmembers(foo): 
    if inspect.isclass(obj): 
     print obj 

ảnh vui nhộn.

Nhưng tôi không thể tìm hiểu cách nhận tất cả các lớp học từ mô-đun hiện tại.

# foo.py 
import inspect 

class Foo: 
    pass 

def print_classes(): 
    for name, obj in inspect.getmembers(???): # what do I do here? 
     if inspect.isclass(obj): 
      print obj 

# test.py 
import foo 

foo.print_classes() 

Đây có thể là điều thực sự rõ ràng, nhưng tôi chưa thể tìm thấy bất kỳ thứ gì. Ai có thể giúp tôi không?

Answer 1

Hãy thử điều này:

import sys 
current_module = sys.modules[__name__] 

Trong bối cảnh của bạn:

import sys, inspect 
def print_classes(): 
    for name, obj in inspect.getmembers(sys.modules[__name__]): 
     if inspect.isclass(obj): 
      print(obj) 

Và thậm chí tốt hơn:

clsmembers = inspect.getmembers(sys.modules[__name__], inspect.isclass) 

Vì inspect.getmembers() mất một vị ngữ.

Answer 2

Tôi không biết nếu có cách 'thích hợp' để làm điều đó, nhưng đoạn mã của bạn đang đi đúng hướng: chỉ cần thêm import foo vào foo.py, làm inspect.getmembers(foo) và nó sẽ hoạt động tốt.

Answer 3

gì về

g = globals().copy() 
for name, obj in g.iteritems(): 

?

Answer 4

import pyclbr 
print(pyclbr.readmodule(__name__).keys()) 

Lưu ý rằng mô-đun trình duyệt lớp Python của stdlib sử dụng phân tích nguồn tĩnh, vì vậy nó chỉ hoạt động cho các mô-đun được hỗ trợ bởi tệp .py thực.

Answer 5

Tôi có thể nhận tất cả những gì tôi cần từ số dir được tích hợp thêm getattr.

# Works on pretty much everything, but be mindful that 
# you get lists of strings back 

print dir(myproject) 
print dir(myproject.mymodule) 
print dir(myproject.mymodule.myfile) 
print dir(myproject.mymodule.myfile.myclass) 

# But, the string names can be resolved with getattr, (as seen below) 

Mặc dù, nó đi ra trông như một Hairball:

def list_supported_platforms(): 
    """ 
     List supported platforms (to match sys.platform) 

     @Retirms: 
      list str: platform names 
    """ 
    return list(itertools.chain(
     *list(
      # Get the class's constant 
      getattr(
       # Get the module's first class, which we wrote 
       getattr(
        # Get the module 
        getattr(platforms, item), 
        dir(
         getattr(platforms, item) 
        )[0] 
       ), 
       'SYS_PLATFORMS' 
      ) 
      # For each include in platforms/__init__.py 
      for item in dir(platforms) 
      # Ignore magic, ourselves (index.py) and a base class. 
      if not item.startswith('__') and item not in ['index', 'base'] 
     ) 
    )) 

Answer 6

Một giải pháp mà các công trình bằng Python 2 và 3:

#foo.py 
import sys 

class Foo(object): 
    pass 

def print_classes(): 
    current_module = sys.modules[__name__] 
    for key in dir(current_module): 
     if isinstance(getattr(current_module, key), type): 
      print(key) 

# test.py 
import foo 
foo.print_classes() 

Answer 7

Nếu bạn muốn có tất cả các lớp , thuộc về mô-đun hiện tại, bạn có thể sử dụng điều này:

import sys, inspect 
def print_classes(): 
    is_class_member = lambda member: inspect.isclass(member) and member.__module__ == __name__ 
    clsmembers = inspect.getmembers(sys.modules[__name__], is_class_member) 

Nếu bạn sử dụng câu trả lời của Nadia và bạn đã nhập các lớp khác trên mô-đun của mình, các lớp đó cũng sẽ được nhập.

Vì vậy, đó là lý do tại sao member.__module__ == __name__ đang được thêm vào vị từ được sử dụng trên is_class_member. Câu lệnh này kiểm tra xem lớp đó có thuộc về module hay không.

Vị từ là hàm (có thể gọi), trả về giá trị boolean.