summaryrefslogtreecommitdiff
path: root/ricky/param/numberrange.py
blob: 626d128c992b06ed423be7f68fe1ea4cd3064aa1 (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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
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.selections())
            )# + self.range_max - self.range_min
        if weights_total < 100:
            weights_total = 100;
        choice = random.randint(0, weights_total)
        import sys
        sys.stderr.write("choosing %s: random_int: %s, selections: %s\n" % (
            self.name,
            choice,
            self.selections()))
        position = 0
        for elem in self.selections():
            position += elem["weight"]
            if position >= choice:
                self.value = elem["value"]
                return
        
        val = random.randint(self.range_min, self.range_max)
        self.value = val
        
    @property
    def value(self):
        return super(NumberRange, self).value_get()

    @value.setter
    def value(self, value):
        self._value = value
        import sys
        sys.stderr.write("%s \n" % self.name)
        sys.stderr.write("%s \n" % self._value)
        if self._value < self.range_min:
            sys.stderr.write("PROBLEM HERE1") 
        if self._value > self.range_max:
            sys.stderr.write("%s > %s\n" % (self._value, self.range_max))
            sys.stderr.write("%s\n" % type(self._value))
            sys.stderr.write("%s\n" % type(self.range_max))
            sys.stderr.write("PROBLEM HERE2") 
        if self._value and \
            (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