求助, 这段代码怎么复用

[plko345]

麻烦大佬看看, 注 1 的函数是变化的, 其它部分基本一样, 有多个类似的统计, 菜鸟一枚, 先谢过

def cal_category_ecs_cpu_core_amount_by_tags(**tags):
    """合计机器 CPU 核心总数, 通过标签过滤"""
    amount = 0
    for instance_id in get_category_instance_id_by_tags(**tags):
        ecs_obj = ECSInstance(instance_id)
        cpu_core = ecs_obj.get_cpu_core()    # 注 1
        amount += cpu_core

    return amount

def cal_category_ecs_memory_amount_by_tags(**tags):
    """合计机器内存总数, 通过标签过滤"""
    amount = 0
    for instance_id in get_category_instance_id_by_tags(**tags):
        ecs_obj = ECSInstance(instance_id)
        memory = ecs_obj.get_memory()    # 注 1
        amount += memory

    return amount

我的想法是将这个类的实例方法作为参数传进去, 但这样处理似乎不是很合适, 而且这样处理的话方法是需要改成 类方法 或者静态方法吗?

[hebin]

python 是可以直接函数做为参数传递的的

def stat(stat_fn, **tags) 类似这种

[ipwx]

为啥不一起统计？（手动狗头

def cal_category_ecs_amount_by_tags(stat_names: Sequence[str], **tags) -> Dict[str, int]:
....stat_values = {n: 0 for n in stat_names}
....for instance_id in get_category_instance_id_by_tags(**tags):
........ecs_obj = ECSInstance(instance_id)
........for n in stat_names:
............stat_values[n] += getattr(ecs_obj, f'get_{n}')()
....return stat_values

[wuwukai007]

怎么发代码？

[tankren]

@wuwukai007 新建主题的时候选 markdown 语法

[lihongjie0209]

```

def get_category_ecs_data_by_tags(**tags):
"""合计机器内存总数, 通过标签过滤"""
//amount = 0
datas = []
for instance_id in get_category_instance_id_by_tags(getDatafn, handleDatafn, **tags):
ecs_obj = ECSInstance(instance_id)
data= fn1(ecs_obj) # 注 1
datas.add(data)

return handleDatafn(datas)




调用方:

计算内存: get_category_ecs_data_by_tags(lamda ecs: ecs.get_memory(), lambda datas: sum(datas), tag1, tag2 )
计算 CPU: get_category_ecs_data_by_tags(lamda ecs: ecs.get_cpu_core(), lambda datas: sum(datas), tag1, tag2 )


```

[lihongjie0209]

data= fn1(ecs_obj) # 注 1
改为
data = getDatafn(ecs_obj)

[wuwukai007]

```python
def cal_category_ecs_memory_or_cpu_amount_by_tags(_type,**tags):
'''

:param _type: 类型
:param tags: other
:return: 数量
'''
"""合计机器内存总数, 通过标签过滤"""
amount = 0
for instance_id in get_category_instance_id_by_tags(**tags):
ecs_obj = ECSInstance(instance_id)
count = ecs_obj,get_memory() if _type == 'memory' else ecs_obj.get_cpu_core()
amount += count
return amount
```

[Orenoid]

这种模式还有很多的话，用楼上的 if-else 显然是不现实的。
如果你的 get_memory/get_cpu_core 都不需要操作 self 的话，我觉得转成静态方法没啥问题啊，然后把方法作为参数传进来就好了。
如果必须要用实例方法的形式调用的话，就用内置的 getattr 函数，把方法名传进来，比如：
data_fetcher = getattr(ecs_obj, 'get_memory')
data = data_fetcher()
然后把函数名和一些变量改为更通用的名字就行了。

[plko345]

@ipwx 因为并不是全都需要统计, 有的只需要其中的一个, 有的需要多个

[plko345]

谢谢各位
@ipwx
@Orenoid
我试试这个 getattr 方法, 应该是我想要的, 这个类里的方法有 self 操作, 必须是实例方法

[ipwx]

@plko345 你像我写的那样一起统计，要统计的量通过 stat_names 传入，不是最优解么。

不然你要统计多个指标，就要查多次数据库，特别浪费时间。

[plko345]

@ipwx 有道理, 我想想看, 主要我需要对不同的 tag 匹配, 不知道这样全统计会不会不灵活, 因为有的 tag 类型的不需要某些数据, 有的又需要, 不过我没多少数据, 这样全统计好像也没什么大问题

`-> Dict[str, int]` 你代码里的这个是什么意思

[mind3x]

Lambda 了解一下。

[arvinsilm]

不太懂 Python，参数加个 tag，循环里加个 switch ？

[luozic]

@ipwx 這個 python 魔法用得有點多啊。
@plko345 這個是返回值注解， 不過有點點問題想請教一下，這裏面的 ecs_obj = ECSInstance(instance_id)默認是去重了麽？

[ipwx]

@plko345 python 3.6 以上的类型注解，配合 ide 智能提示的体验简直飞起。因为习惯所以给你加上了

[ipwx]

@luozic 没觉得，3.6+ 这不都是基本操作么

[zwzmzd]

把 lambda 作为函数参数传进去

[plko345]

@luozic 类里没有去重, 默认传入的是唯一的, 不过似乎应该加上, 多谢提醒

[luozic]

@ipwx python 后面都是在向 typescript 学习类型推断和标记，这个才是最近 python 的主要方向吧。

[cassidyhere]

def stats(callback_name, **tags):
for instance_id in get_category_instance_id_by_tags(**tags):
ecs_obj = ECSInstance(instance_id)
yield getattr(ecs_obj, callback_name)


def cal_category_ecs_cpu_core_amount_by_tags(**tags):
return sum(stats('get_cpu_core', **tags))


def cal_category_ecs_memory_amount_by_tags(**tags):
return sum(stats('get_memory', **tags))


其实应该修改的是 ECSInstance