summaryrefslogtreecommitdiff
path: root/Pb_Api/Param/NumberRange.py
blob: fa11a2988733dc5c69bb937e12bb6eaf7ce5ab6d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
from Pb_Api.Param.MultiSelect import Pb_Api_Param_MultiSelect
import random
class Pb_Api_Param_NumberRange(Pb_Api_Param_MultiSelect): 
    def __init__(self, **kwargs):
      super(Pb_Api_Param_NumberRange, self).__init__(**kwargs)  
      self.range_min = kwargs['min']
      self.range_max = kwargs['max']
    def randomize(self):
       weights_total = sum(map(lambda x: x["weight"], self.options()) + self.range_max - self.range_min   
       choice = random.randint(0, weights_total)
       position = 0
       for elem in self.options():
          position += elem["weight"] 
          if position >= choice:
              self.value = elem["value"]
              return
       self.value = random.randint(ANGLE_MIN,ANGLE_MAX),