blob: 85314f8ca021bf3a7a99227b509b44ea1e098a35 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
|
import random
from ricky.param import Param
class NumberRange(Param):
def __init__(self, **kwargs):
super(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(self.range_min, self.range_max)
@property
def value(self):
return super(NumberRange, self).value_get()
@value.setter
def value(self, value):
self._value = value
if self._value < self.range_min or self._value > self.range_max:
raise ValueError(
"Value must be between %s and %s\n" % (
self.range_min, self.range_max
)
)
if self._value is not None:
self.is_ready = 1
self.set_by_user = 1
|