今天开会的时间,翻了翻《流畅的Python》,还是很有启发的。
例如,一个订单优惠业务需求是这样子的,电商会有很多优惠方式,一个订单依据不同的优惠方式,会有不同的折扣金额。算法需要自动计算出最大折扣金额,推荐给用户。
按照函数是一等公民的思想,我们用不同的函数来表征每个优惠策略。
# 策略1
def fidelity_promo(order):
"""订单 order 产生的优惠金额"""
return 1
# 策略2
def bulk_item_promo(order):
return 2
用户下了一个订单,需要给出最优惠的策略,这里用
# 策略集合
promos = [fidelity_promo, bulk_item_promo]
# 返回最大折扣
def best_promo(order):
return max(promo(order) for promo in promos)
功能上,没有问题。但从可维护的角度来看,如果不断增减策略的话,开发者既要编写策略函数,又要维护策略集合promos。
能不能让策略集合promos 自动获取全部的策略函数呢?
方法一 内省
promos = [globals()[name] for name in globals()
if name.endswith('_promo')
and name != 'best_promo'
]
这个方法写起来不够优雅,同时要求函数名称必须用_promo
结尾。
方法二 模块
import promotions
import inspect
promps = [func for name, func in
inspect.getmembers(promotions, inspect.isfunction)
]
这个方法对函数名称没有要求,但要求promotions模块中的所有函数,都必须是策略函数。
方法三 装饰器
promos = []
def promotion(func):
promos.append(func)
return func
@promotion
def fidelity_promo(order):
"""订单 order 产生的优惠金额"""
return 1
@promotion
def bulk_item_promo(order):
return 2
这个方法最为优雅,对函数名没有要求,在任何地方都可以声明策略函数。同时,如何临时禁用某个策略,只用注释掉 '@promotion' 即可。