geni/util/specdict.py

   1 ##
   2 # SpecDict
   3 #
   4 # SpecDict defines a means for converting a dictionary with plc specific keys
   5 # to a dict with rspec specific keys.
   6 #
   7 # SpecDict.fields dict defines all the rspec specific attribute and their
   8 # expected type.
   9 #
  10 # SpecDict.plc_fields defines a one to one mapping of plc attribute to rspec
  11 # attribute
  12
  13 from types import StringTypes, ListType
  14
  15
  16 class SpecDict(dict):
  17     """
  18     Base class of SpecDict objects.
  19     """
  20     fields = {}
  21     plc_fields = {}
  22     type = None
  23
  24     def __init__(self, spec_dict):
  25         # convert plc dict and initialize self
  26         sdict = self.plcToSpec(spec_dict)
  27         dict.__init__(self, sdict)
  28
  29
  30     def plcToSpec(self, spec_dict):
  31         """
  32         Defines how to convert a plc dict to rspec dict
  33         """
  34         spec = {}
  35         for field in self.fields:
  36             value = ""
  37             expected = self.fields[field]
  38             if isinstance(expected, StringTypes):
  39                 if self.plc_fields.has_key(field):
  40                     plc_field = self.plc_fields[field]
  41                     if spec_dict.has_key(plc_field):
  42                         value = spec_dict[plc_field]
  43             elif isinstance(expected, ListType):
  44                 expected = expected[0]
  45                 if self.plc_fields.has_key(field):
  46                     plc_field = self.plc_fields[field]
  47                     if spec_dict.has_key(plc_field):
  48                         value = [expected(value) for value in spec_dict[plc_field]]
  49             spec[field] = value
  50         return {self.type: spec}
  51
  52 #
  53 # fields = { geni_field:  type.  Could be class for nested classes, otherwise prob str}
  54 # plc_fields = {geni_field : plc_field}
  55 #
  56
  57 class IfSpecDict(SpecDict):
  58     type = 'IfSpec'
  59     fields = {'name': '',
  60               'addr': '',
  61               'type': '',
  62               'init_params': '',
  63               'min_rate': '',
  64               'max_rate': '',
  65               'max_kbyte': '',
  66               'ip_spoof': ''}
  67     plc_fields = {'name': 'is_primary', # XXX needs munging to return name instead of True or False
  68                  'addr': 'ip',
  69                  'type': 'type'}
  70
  71 class LinkSpecDict(SpecDict):
  72     type = 'IfSpec'
  73     fields = {'name': '',
  74               'addr': '',
  75               'type': '',
  76               'init_params': '',
  77               'min_rate': '',
  78               'max_rate': '',
  79               'max_kbyte': ''}
  80     plc_fields = {}
  81
  82
  83 class NodeSpecDict(SpecDict):
  84     type = 'NodeSpec'
  85     fields = {'name': '',
  86               'type': '',
  87               'init_params': '',
  88               'cpu_min': '',
  89               'cpu_share': '',
  90               'cpu_pct': '',
  91               'disk_max': '',
  92               'start_time': '',
  93               'duration': '',
  94               'net_if': [IfSpecDict]}
  95
  96     plc_fields = {'name': 'hostname',
  97                   'net_if': 'interfaces'}
  98
  99 class NetSpecDict(SpecDict):
 100     type = 'NetSpec'
 101     fields = {'name': '',
 102               'start_time': '',
 103               'duration': '',
 104               'nodes': [NodeSpecDict],
 105              }
 106     plc_fields = {'name': 'name',
 107                   'start_time': 'start_time',
 108                   'duration': 'duration',
 109                   'nodes': 'nodes'}
 110
 111 class RspecDict(SpecDict):
 112     type = 'RSpec'
 113     fields = {'start_time': '',
 114               'duration': '',
 115               'networks': [NetSpecDict]
 116              }
 117     plc_fields = {'networks': 'networks',
 118                   'start_time': 'start_tim',
 119                   'duration': 'duration'
 120                  }
 121
 122 # vim:ts=4:expandtab