summaryrefslogtreecommitdiff
path: root/ricky/param/numberrange.py
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