Python - 如果仅指定它们(域),则添加条件?

时间:2014-11-24 12:25:13

标签: python

我正在编写一种方法,该方法将通过调用条件过滤记录。我的意思是,是否会检查是否满足任何记录的默认条件,然后如果指定,则检查是否需要满足这些记录的其他条件。但如果没有指定这些条件,它就会忽略它。

所以我可以这样写:

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

1 个答案:

答案 0 :(得分:3)

而不是print我会return TrueFalse,而是您可以单独检查每个项目。假设你正在测试某个对象的属性(但这可以适应你需要的任何东西):

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,并使用filtercard_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不使用任何实例或类属性或方法 - 您应该检查它是否需要是实例方法。