37 lines
1.5 KiB
Python
37 lines
1.5 KiB
Python
import random
|
|
|
|
|
|
class Promotioner:
|
|
"""
|
|
Promotioner is used to select promotion randomly based on weights of every promotion.
|
|
"""
|
|
def __init__(
|
|
self,
|
|
promotions: list[dict],
|
|
default_promotion_index: int = 0,
|
|
promotion_vars: dict = None,
|
|
):
|
|
self.promotions = promotions
|
|
self.default_promotion_index = default_promotion_index
|
|
if not promotion_vars:
|
|
promotion_vars = {}
|
|
self.promotion_vars = promotion_vars
|
|
self.partial_sums: list = [self.promotions[0]['weight']]
|
|
for promotion in self.promotions[1:]:
|
|
self.partial_sums.append(promotion['weight'] + self.partial_sums[-1])
|
|
|
|
def choose_promotion(self, language: str = 'en') -> str:
|
|
pivot = random.randrange(self.partial_sums[-1])
|
|
for partial_sum, promotion in zip(self.partial_sums, self.promotions):
|
|
if partial_sum <= pivot:
|
|
continue
|
|
if language in promotion['texts']:
|
|
return promotion['texts'][language].format(**self.promotion_vars)
|
|
elif promotion.get('local', False):
|
|
default_promotion = self.promotions[self.default_promotion_index]
|
|
if language in default_promotion['texts']:
|
|
return default_promotion['texts'][language].format(**self.promotion_vars)
|
|
return default_promotion['texts']['en'].format(**self.promotion_vars)
|
|
else:
|
|
return promotion['texts']['en'].format(**self.promotion_vars)
|