python - 通过重用代码和传递成员名称来运行测试来减少行数

Question

我有两个来自同一类的汽车对象和一个汽车列表。

class Car:
    def __init__(self):
        pass

cars = []

car1 = Car()
car1.color = 'black'
cars.append(car1)
car2 = Car()
car2.engine = 'V8'
cars.append(car2)

我还有两个函数，用于在匹配时从 cars 中删除对象。

def rm_car_color(c):
    for car_n in cars:
        try:
            if car_n.color == c.color:
                cars.remove(car_n)
        except Exception, err:
            pass

def rm_car_engine(c):
    for car_n in cars:
        try:
            if car_n.engine == c.engine:
                cars.remove(car_n)
        except Exception, err:
            pass

如您所见，它们实际上是相同的。我想要的是能够将要检查的成员传递给函数，以便我可以删除这两个函数之一，并且只有一个在 Python 2.6.6 中处理这两个函数。这可能吗？

此代码仅用于解释。我正在处理的代码有许多这样的删除函数(总共 8 个)，如果我能做到这一点，我会大大减少代码行数。

Answer 1

首先查看代码。正如您所说，这两个功能几乎相同。这两个功能的唯一区别是什么？属性(c.color vs c.engine):

if car_n.color == c.color:

对比

if car_n.engine == c.engine:

因此，您可以使用内置的 getattr 将此代码更改为功能:

def rm_car_attribute(c, attrib):
    comparison_attrib = getattr(c, attrib)
    for car_n in cars[:]:
    # Blender mentioned a good practice: not to modify the list you are iterating
    # cars[:] creates a temporary copy of cars to iterate over
        try: 
            if getattr(car_n, attrib) == comparison_attrib:
                cars.remove(car_n)
        except Exception, err:
            pass

这可以这样使用:

rm_car_attribute(c, 'engine') # same as rm_car_engine(c) from above

---

另一方面，我不太确定您期望的异常类型。我看到的唯一可能的错误是汽车的属性是否明确定义(即 car_instance.engine = 'V8')，但对于其中一辆车，该属性未定义(在您的汽车列表中，其中一个实例没有 engine 属性)。您可以更改 try/except 以专门捕获此错误:

def rm_car_attribute(c, attrib):
        comparison_attrib = getattr(c, attrib)
        for car_n in cars[:]:
            try: 
                if getattr(car_n, attrib) == comparison_attrib:
                    cars.remove(car_n)
            except AttributeError:
                pass