Как упростить код?

Question

[Никита Pewget]

Приветствую!

Как упростить этот код и сделать его компактнее?
Мне кажется, что есть способ делать такие вещи как-то элегантно, или я ошибаюсь?
Заранее спасибо.

totals = {
    'qty': 0,
    'total_amount': 0,
    'cost': 0,
    'profit': 0,
}

for order in orders:
    totals['qty'] += order.qty
    totals['total_amount'] += order.total_amount
    totals['cost'] += order.cost
    totals['profit'] += order.profit

Answer 1

[lega]

Как то так:

totals = defaultdict(int)
for order in orders:
    for k in ['qty', 'total_amount', 'cost', 'profit']:
        totals[k] += getattr(order, k)

Comments

[lega]

Если total будет из вашего примера, то вместо списка ключи можно взять из totals.keys()

[Никита Pewget]

Спасибо, так намного приятнее :)

[Владислав]

@lega откуда взялся defaultdict? Где-то выше вроде как функцию объявляешь такую?

[Никита Pewget]

@Hateman31 from collections import defaultdict

[Владислав]

@Pewget ясн

Answer 2

[Артем Спиридонов]

Можно перебирать свойства в цикле?

Не помню, как это должно быть на питоне, но на PHP как-то так:

foreach ($orders as $order)
{
	foreach ($order as $property => $value)
	{
		$totals[$property] += $value;
	}
}