python获取父类的子类(遍历,递归),并循环执行所有子类的某一方法

时间:2021-06-05 14:31:05

前言

换了新工作,踏足于python语言的开发,也把自己的学习过程记录下来。

一,递归获取某一父类的所有子类

all_subclasses = {'0': '0'}

def get_all_classes(model):
"""
获取父类的所有子类
""" for subclass in model.__subclasses__():
# print(subclass._meta.abstract)
if (not (subclass.__name__) in all_subclasses.keys()) and (not subclass._meta.abstract):
all_subclasses[subclass.__name__] = subclass
get_all_classes(subclass)
return all_subclasses

  这里主要用到了__subclasses__() 这个方法,这个方法返回的是这个类的子类的集合,用递归的方法,去获取传入类型的所有子类。返回给全局变量 all_subclasses这个字典集合。

二,遍历子类集合,执行某一方法

def do_collection(model=None, date=None):
"""
执行收集数据的方法
"""
allclasses = get_all_classes(StatBaseModel)
if date:
date = datetime.date.today() + datetime.timedelta(days=-1)
if model:
fn_collect = getattr(allclasses[model], 'collect', None)
if callable(fn_collect):
fn_collect(date)
print("执行{0}的collect 方法".format(model))
return
for item, value in allclasses.items():
fn_collect = getattr(value, 'collect', None)
if callable(fn_collect):
fn_collect(date)
print("执行{0}的collect 方法".format(item))

  通过getattr()这个方法获取类型的某一个属性,然后通过callback方法判断这个属性是不是可执行的,然后执行这个方法。

三,总结和反思

这个方法我该来改去的,总是想把它搞得更面向对象些,刚开始封装成了一个类,把all_subclasses这个字断作为一个property,但是看着是很别扭,同事说那样做也不是很好,但是具体怎么不好也没说出来,我想更大的一个原因是还是我对python的不够熟悉吧,希望看到的大神们能给提出改进的建议。