2 # Implements support for geni records
4 # TODO: Use existing PLC database methods? or keep this separate?
11 # The GeniRecord class implements a Geni Record. A GeniRecord is a tuple
12 # (Name, GID, Type, Info).
14 # Name specifies the HRN of the object
15 # GID is the GID of the object
16 # Type is user | sa | ma | slice | component
18 # Info is comprised of the following sub-fields
19 # pointer = a pointer to the record in the PL database
20 # pl_info = planetlab-specific info (when talking to client)
21 # geni_info = geni-specific info (when talking to client)
23 # The pointer is interpreted depending on the type of the record. For example,
24 # if the type=="user", then pointer is assumed to be a person_id that indexes
25 # into the persons table.
27 # A given HRN may have more than one record, provided that the records are
28 # of different types. For example, planetlab.us.arizona may have both an SA
29 # and a MA record, but cannot have two SA records.
34 # Create a Geni Record
36 # @param name if !=None, assign the name of the record
37 # @param gid if !=None, assign the gid of the record
38 # @param type one of user | sa | ma | slice | component
39 # @param pointer is a pointer to a PLC record
40 # @param dict if !=None, then fill in this record from the dictionary
42 def __init__(self, name=None, gid=None, type=None, pointer=None, dict=None):
53 self.set_pointer(pointer)
55 self.set_name(dict['name'])
56 self.set_gid(dict['gid'])
57 self.set_type(dict['type'])
58 self.set_pointer(dict['pointer'])
60 self.set_pl_info(dict["pl_info"])
61 if "geni_info" in dict:
62 self.set_geni_info(dict["geni_info"])
65 # Set the name of the record
67 # @param name is a string containing the HRN
69 def set_name(self, name):
74 # Set the GID of the record
76 # @param gid is a GID object or the string representation of a GID object
78 def set_gid(self, gid):
79 if isinstance(gid, str):
82 self.gid = gid.save_to_string(save_parents=True)
86 # Set the type of the record
88 # @param type is a string: user | sa | ma | slice | component
90 def set_type(self, type):
95 # Set the pointer of the record
97 # @param pointer is an integer containing the ID of a PLC record
99 def set_pointer(self, pointer):
100 self.pointer = pointer
104 # Set the PLC info of the record
106 # @param pl_info is a dictionary containing planetlab info
108 def set_pl_info(self, pl_info):
109 self.pl_info = pl_info
113 # Set the geni info the record
115 # @param geni_info is a dictionary containing geni info
117 def set_geni_info(self, geni_info):
118 self.geni_info = geni_info
122 # Return the pl_info of the record, or an empty dictionary if none exists
124 def get_pl_info(self):
131 # Return the geni_info of the record, or an empty dictionary if none exists
133 def get_geni_info(self):
135 return self.geni_info
140 # Return the name (HRN) of the record
146 # Return the type of the record
152 # Return the pointer of the record. The pointer is an integer that may be
153 # used to look up the record in the PLC database. The evaluation of pointer
154 # depends on the type of the record
156 def get_pointer(self):
160 # Return the GID of the record, in the form of a GID object
161 # TODO: not the best name for the function, because we have things called
162 # gidObjects in the Cred
164 def get_gid_object(self):
165 return GID(string=self.gid)
168 # Return a key that uniquely identifies this record among all records in
169 # Geni. This key is used to uniquely identify the record in the Geni
173 return self.name + "#" + self.type
176 # Returns a list of field names in this record. pl_info, geni_info are not
177 # included because they are not part of the record that is stored in the
178 # database, but are rather computed values from other entities
180 def get_field_names(self):
181 return ["name", "gid", "type", "pointer"]
184 # Given a field name ("name", "gid", ...) return the value of that field.
186 # @param name is the name of field to be returned
188 def get_field_value_string(self, fieldname):
189 if fieldname == "key":
192 val = getattr(self, fieldname)
193 if isinstance(val, str):
194 return "'" + str(val) + "'"
199 # Given a list of field names, return a list of values for those fields.
201 # @param fieldnames is a list of field names
203 def get_field_value_strings(self, fieldnames):
205 for fieldname in fieldnames:
206 strs.append(self.get_field_value_string(fieldname))
210 # Return the record in the form of a dictionary
214 names = self.get_field_names()
216 dict[name] = getattr(self, name)
219 dict['pl_info'] = self.pl_info
222 dict['geni_info'] = self.geni_info
227 # Dump the record to stdout
229 # @param dump_parents if true, then the parents of the GID will be dumped
231 def dump(self, dump_parents=False):
232 print "RECORD", self.name
233 print " hrn:", self.name
234 print " type:", self.type
236 self.get_gid_object().dump(8, dump_parents)
237 print " pointer:", self.pointer
240 geni_info = getattr(self, "geni_info", {})
242 for key in geni_info.keys():
243 print " ", key, ":", geni_info[key]
246 pl_info = getattr(self, "pl_info", {})
248 for key in pl_info.keys():
249 print " ", key, ":", pl_info[key]