我正在编写一种方法,该方法将通过调用条件过滤记录。我的意思是,是否会检查是否满足任何记录的默认条件,然后如果指定,则检查是否需要满足这些记录的其他条件。但如果没有指定这些条件,它就会忽略它。
所以我可以这样写:
def test_conditions(a, b=None)
#b - this one is additional condition that might be specified or not
if a > 5 and (b > 10 if b else True):
print 'record passed'
我需要这样做,但我认为它看起来非常优雅,我需要使用多个附加条件,这些条件在调用方法时可能会或可能不会指定,然后这样的方法可能不会看起来不错。
所以也许在Python中有一些练习如何最好地解决这些问题(我指的是当你不知道在调用方法之前是否需要检查它时,处理这些附加条件的最佳方法)?
我被要求提供我正在编写的真实代码,所以这里是:
def countEmergency(self, card_ids, duration=None):
cnt = 0
for card in card_ids:
if card.emergency_level == 'first' and (card.duration > duration if duration else True):
cnt += 1
return cnt
答案 0 :(得分:3)
而不是print
我会return
True
或False
,而是您可以单独检查每个项目。假设你正在测试某个对象的属性(但这可以适应你需要的任何东西):
def test_conditions(obj, a=5, b=None):
if obj.a <= a:
return False
if b is not None and obj.b <= b:
return False
return True
或者,使用**kwargs
来获取任意关键字参数:
def test_conditions(obj, a=5, **kwargs):
if obj.a <= a:
return False
for k, v in kwargs.items():
if getattr(obj, k) <= v:
return False
return True
或
def test_conditions(obj, **kwargs):
if obj.a <= kwargs.get('a', 5):
return False
for k, v in kwargs.items():
if getattr(obj, k) <= v:
return False
return True
鉴于您实际做了什么,请使用continue
而不是返回:
def countEmergency(self, card_ids, duration=None):
cnt = 0
for card in card_ids:
if card.emergency_level != 'first':
continue
if duration is not None and card.duration <= duration:
continue
cnt += 1
return cnt
或者,重构使用上述函数,需要一个card
,并使用filter
从card_ids
获取有效元素的列表。
from functools import partial
class WhateverThisIs(object):
...
@staticmethod
def validate_card(card, duration=None):
if card.emergency_level != 'first':
return False
if duration is not None and card.duration <= duration:
return False
return True
def countEmergency(self, card_ids, duration=None):
return len(filter(partial(self.validate_card, duration=duration), card_ids))
这使用functools.partial
来创建适当的过滤功能。请注意,您的countEmergency
不使用任何实例或类属性或方法 - 您应该检查它是否需要是实例方法。