Support GetLinkSpecs

[sfa.git] / geni / util / specdict.py

diff --git a/geni/util/specdict.py b/geni/util/specdict.py
index 1b0a6ba..8f0b597 100644 (file)
--- a/geni/util/specdict.py
+++ b/geni/util/specdict.py
@@ -46,11 +46,14 @@ class SpecDict(dict):
                     plc_field = self.plc_fields[field]
                     if spec_dict.has_key(plc_field):
                         value = [expected(value) for value in spec_dict[plc_field]]
-                    
-             
             spec[field] = value
         return {self.type: spec}
-    
+
+#
+# fields = { geni_field:  type.  Could be class for nested classes, otherwise prob str}
+# plc_fields = {geni_field : plc_field}
+#
+ 
 class IfSpecDict(SpecDict):
     type = 'IfSpec'
     fields = {'name': '',
@@ -61,9 +64,29 @@ class IfSpecDict(SpecDict):
               'max_rate': '',
               'max_kbyte': '',
               'ip_spoof': ''}
-    plc_fields = {'name': 'is_primary',
+    plc_fields = {'name': 'is_primary', # XXX needs munging to return name instead of True or False
                  'addr': 'ip',
                  'type': 'type'}
+ 
+class LinkSpecDict(SpecDict):
+    type = 'LinkSpec'
+    fields = {'min_alloc': '', 
+              'max_alloc': '', 
+              'type': '', 
+              'start_time': '', 
+              'bw': '', 
+              'duration': '', 
+              'init_params': '',
+              'endpoints': [IfSpecDict]}
+    plc_fields = {'min_alloc': 'min_alloc',
+              'max_alloc': 'max_alloc', 
+              'type': 'type', 
+              'start_time': 'start_time', 
+              'bw': 'bw', 
+              'duration': 'duration', 
+              'init_params': 'init_params',
+              'endpoints': 'endpoints'}
+  
             
 class NodeSpecDict(SpecDict):
     type = 'NodeSpec'
@@ -89,6 +112,8 @@ class NetSpecDict(SpecDict):
               'nodes': [NodeSpecDict],
              }
     plc_fields = {'name': 'name',
+                  'start_time': 'start_time',
+                  'duration': 'duration',
                   'nodes': 'nodes'}
 
 class RspecDict(SpecDict):
@@ -97,6 +122,9 @@ class RspecDict(SpecDict):
               'duration': '',
               'networks': [NetSpecDict]
              }
-    plc_fields = {'networks': 'networks'}
+    plc_fields = {'networks': 'networks',
+                  'start_time': 'start_tim',
+                  'duration': 'duration'
+                 }
 
 # vim:ts=4:expandtab