Python Expense.get_allの例

コード例 #1

ファイル: global_context.py プロジェクト: ChrisPoul/Industrias-EnGo

def get_autocomplete_data(group, attribute):
    groups = dict(
        users=User.get_all(),
        products=Product.get_all(),
        customers = Customer.get_all(),
        expenses=Expense.get_all(),
        warehouses=Warehouse.get_all()
    )
    data = []
    for obj in groups[group]:
        value = getattr(obj, attribute)
        if value not in set(data):
            data.append(value)
    
    return data

コード例 #2

    def test_should_return_all_expenses(self):
        expenses = Expense.get_all()

        self.assertEqual(expenses, [self.expense])