class FreeTraining(models.Benefit):
class Meta:
proxy = True
description = "Free Training"
def apply(self, basket, condition, offer=None):
message = u"certification pro et training absent"
pro = 0
training = 0
for article in basket.all_lines():
if article.product_id == 212:
pro +=1
if article.product_id == 217:
training +=1
condition.consume_items(offer, basket, ())
if pro and not training:
message = "certification pro dans le panier"
if not pro and training:
message = "formation dans le panier"
if pro and training:
message = "certification pro et formation dans le panier"
return models.PostOrderAction(message)