summaryrefslogtreecommitdiff
path: root/ricky/params.py
blob: 304ad2d5db246a23fd677f48f399a0f8063be7c4 (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
import sys
import pprint
class Params(object):
  def __init__(self):
    self._api = None
    self._params = []

  def param(self, name):
    for p in self._params:
      if p.name == name:
        return p
    return None

  def __str__(self):
    return pprint.pformat({ "params": map(lambda x: vars(x), self._params) })

  def randomize(self):
    for el in self._params:
      if el.set_by_user:
        continue
      el.randomize()

  @property
  def api(self):
    return self._api

  @api.setter
  def api(self, cls):
    self._api = cls


  def execute(self):
    return self.api.call(self)

  def is_ready(self):
    for p in self._params:
      if not p.is_ready and not p.default:
        sys.stderr.write("param not ready: {}".format(p))
        return 0
    return 1

  def __dict__(self):
    result = {}
    for p in self._params:
      result[p.name] = p.value
    return result

  def as_dict(self):
    return self.__dict__()