def test_price_display_options_more():
options = PriceDisplayOptions(show_prices=False)
assert options.show_prices is False
assert options.hide_prices is True
assert options.include_taxes is None
options = PriceDisplayOptions(include_taxes=True)
assert options.show_prices is True
assert options.include_taxes is True
options = PriceDisplayOptions(include_taxes=False)
assert options.show_prices is True
assert options.include_taxes is False
def get_price_display_options(self):
"""
Get price display options of the contact.
If the default group (`get_default_group`) defines price display
options and the contact is member of it, return it.
If contact is not (anymore) member of the default group or the
default group does not define options, return one of the groups
which defines options. If there is more than one such groups,
it is undefined which options will be used.
If contact is not a member of any group that defines price
display options, return default constructed
`PriceDisplayOptions`.
Subclasses may still override this default behavior.
:rtype: PriceDisplayOptions
"""
groups_with_options = self.groups.with_price_display_options()
if groups_with_options:
default_group = self.get_default_group()
if groups_with_options.filter(pk=default_group.pk).exists():
group_with_options = default_group
else:
# Contact was removed from the default group.
group_with_options = groups_with_options.first()
return group_with_options.get_price_display_options()
return PriceDisplayOptions()
def test_price_display_options_default():
options = PriceDisplayOptions()
assert options.show_prices is True
assert options.hide_prices is False
assert options.include_taxes is None
def get_price_display_options(self):
return PriceDisplayOptions(
include_taxes=self.show_prices_including_taxes,
show_prices=(not self.hide_prices),
)
