sfa/senslab/OARrestapi.py

   1 # import modules used here -- sys is a very standard one
   2 import sys
   3 import httplib
   4 import json
   5 from sfa.senslab.parsing import *
   6 from sfa.senslab.SenslabImportUsers import *
   7
   8 #OARIP='10.127.255.254'
   9 OARIP='192.168.0.109'
  10
  11
  12 OARrequests_list = ["GET_version", "GET_timezone", "GET_jobs", "GET_jobs_table", "GET_jobs_details",
  13 "GET_resources_full", "GET_resources"]
  14
  15 OARrequests_uri_list = ['/oarapi/version.json','/oarapi/timezone.json', '/oarapi/jobs.json',
  16 '/oarapi/jobs/details.json', '/oarapi/resources/full.json', '/oarapi/resources.json']
  17
  18 OARrequests_get_uri_dict = { 'GET_version': '/oarapi/version.json',
  19                         'GET_timezone':'/oarapi/timezone.json' ,
  20                         'GET_jobs': '/oarapi/jobs.json',
  21                         'GET_jobs_table': '/oarapi/jobs/table.json',
  22                         'GET_jobs_details': '/oarapi/jobs/details.json',
  23                         'GET_resources_full': '/oarapi/resources/full.json',
  24                         'GET_resources':'/oarapi/resources.json',
  25 }
  26
  27 POSTformat = {  #'yaml': {'content':"text/yaml", 'object':yaml}
  28 'json' : {'content':"application/json",'object':json},
  29 #'http': {'content':"applicaton/x-www-form-urlencoded",'object': html},
  30 }
  31
  32 class OARrestapi:
  33         def __init__(self):
  34                 self.oarserver= {}
  35                 self.oarserver['ip'] = OARIP
  36                 self.oarserver['port'] = 80
  37                 self.oarserver['uri'] = None
  38                 self.oarserver['postformat'] = None
  39
  40         def GETRequestToOARRestAPI(self, request ):
  41                 self.oarserver['uri'] = OARrequests_get_uri_dict[request]
  42                 try :
  43                         conn = httplib.HTTPConnection(self.oarserver['ip'],self.oarserver['port'])
  44                         conn.request("GET",self.oarserver['uri'] )
  45                         resp = ( conn.getresponse()).read()
  46                         conn.close()
  47                 except:
  48                         raise ServerError("GET_OAR_SRVR : Could not reach OARserver")
  49                 try:
  50                         js = json.loads(resp)
  51                         return js
  52
  53                 except ValueError:
  54                         raise ServerError("Failed to parse Server Response:" + js)
  55
  56
  57
  58         def POSTRequestToOARRestAPI(self, uri,format, data):
  59                 self.oarserver['uri'] = uri
  60                 if format in POSTformat:
  61                         try :
  62                                 conn = httplib.HTTPConnection(self.oarserver['ip'],self.oarserver['port'])
  63                                 conn.putrequest("POST",self.oarserver['uri'] )
  64                                 self.oarserver['postformat'] = POSTformat[format]
  65                                 conn.putheader('content-type', self.oarserver['postformat']['content'])
  66                                 conn.putheader('content-length', str(len(data)))
  67                                 conn.endheaders()
  68                                 conn.send(data)
  69                                 resp = ( conn.getresponse()).read()
  70                                 conn.close()
  71
  72                         except:
  73                                 raise ServerError("POST_OAR_SRVR : error")
  74
  75                         try:
  76                                 answer = self.oarserver['postformat']['object'].loads(resp)
  77                                 return answer
  78
  79                         except ValueError:
  80                                 raise ServerError("Failed to parse Server Response:" + answer)
  81                 else:
  82                         print>>sys.stderr, "\r\n POSTRequestToOARRestAPI : ERROR_POST_FORMAT"
  83
  84
  85 class OARGETParser:
  86
  87         #Insert a new node into the dictnode dictionary
  88         def AddNodeId(self,dictnode,value):
  89                 #Inserts new key. The value associated is a tuple list.
  90                 node_id = int(value)
  91                 dictnode[node_id] = [('node_id',node_id) ]
  92                 return node_id
  93
  94         def AddNodeNetworkAddr(self,tuplelist,value):
  95                 #tuplelist.append(('hostname',str(value)))
  96                 tuplelist.append(('hostname',str(value)))
  97
  98
  99         def AddNodeSite(self,tuplelist,value):
 100                 tuplelist.append(('site_login_base',str(value)))
 101
 102         def AddNodeRadio(self,tuplelist,value):
 103                 tuplelist.append(('radio',str(value)))
 104
 105
 106         def AddMobility(self,tuplelist,value):
 107                 if value :
 108                         tuplelist.append(('mobile',int(value)))
 109                 return 0
 110
 111
 112         def AddPosX(self,tuplelist,value):
 113                 tuplelist.append(('posx',value))
 114
 115
 116         def AddPosY(self,tuplelist,value):
 117                 tuplelist.append(('posy',value))
 118
 119
 120
 121         def ParseVersion(self) :
 122                 print self.raw_json
 123                 #print >>sys.stderr, self.raw_json
 124                 if 'oar_version' in self.raw_json :
 125                         self.version_json_dict.update(api_version=self.raw_json['api_version'] ,
 126                                         apilib_version=self.raw_json['apilib_version'],
 127                                         api_timezone=self.raw_json['api_timezone'],
 128                                         api_timestamp=self.raw_json['api_timestamp'],
 129                                         oar_version=self.raw_json['oar_version'] )
 130                 else :
 131                         self.version_json_dict.update(api_version=self.raw_json['api'] ,
 132                                         apilib_version=self.raw_json['apilib'],
 133                                         api_timezone=self.raw_json['api_timezone'],
 134                                         api_timestamp=self.raw_json['api_timestamp'],
 135                                         oar_version=self.raw_json['oar'] )
 136
 137                 print self.version_json_dict['apilib_version']
 138
 139         def ParseTimezone(self) :
 140                 print " ParseTimezone"
 141
 142         def ParseJobs(self) :
 143                 self.jobs_list = []
 144                 print " ParseJobs "
 145
 146         def ParseJobsTable(self) :
 147                 print "ParseJobsTable"
 148
 149         def ParseJobsDetails (self):
 150                 print "ParseJobsDetails"
 151
 152         def ParseResources(self) :
 153                 print>>sys.stderr, " \r\n  \t\t\t ParseResources__________________________ "
 154                 #resources are listed inside the 'items' list from the json
 155                 self.raw_json = self.raw_json['items']
 156                 self.ParseNodes()
 157                 self.ParseSites()
 158
 159
 160
 161         def ParseResourcesFull(self ) :
 162                 print>>sys.stderr, " \r\n \t\t\t  ParseResourcesFull_____________________________ "
 163                 #print self.raw_json[1]
 164                 #resources are listed inside the 'items' list from the json
 165                 if self.version_json_dict['apilib_version'] != "0.2.10" :
 166                         self.raw_json = self.raw_json['items']
 167                 self.ParseNodes()
 168                 self.ParseSites()
 169
 170
 171         #Parse nodes properties from OAR
 172         #Put them into a dictionary with key = node id and value is a dictionary
 173         #of the node properties and properties'values.
 174         def ParseNodes(self):
 175                 node_id = None
 176                 for dictline in self.raw_json:
 177                         for k in dictline.keys():
 178                                 if k in self.resources_fulljson_dict:
 179                                         # dictionary is empty and/or a new node has to be inserted
 180                                         if node_id is None :
 181                                                 node_id = self.resources_fulljson_dict[k](self,self.node_dictlist, dictline[k])
 182                                         else:
 183                                                 ret = self.resources_fulljson_dict[k](self,self.node_dictlist[node_id], dictline[k])
 184                                                 #If last property has been inserted in the property tuple list, reset node_id
 185                                                 if ret == 0:
 186                                                         #Turn the property tuple list (=dict value) into a dictionary
 187                                                         self.node_dictlist[node_id] = dict(self.node_dictlist[node_id])
 188                                                         node_id = None
 189
 190                                 else:
 191                                         pass
 192
 193         #Retourne liste de dictionnaires contenant attributs des sites
 194         def ParseSites(self):
 195                 nodes_per_site = {}
 196                 # Create a list of nodes per  site_id
 197                 for node_id in self.node_dictlist.keys():
 198                         node  = self.node_dictlist[node_id]
 199                         if node['site_id'] not in nodes_per_site.keys():
 200                                 nodes_per_site[node['site_id']] = []
 201                                 nodes_per_site[node['site_id']].append(node['node_id'])
 202                         else:
 203                                 if node['node_id'] not in nodes_per_site[node['site_id']]:
 204                                         nodes_per_site[node['site_id']].append(node['node_id'])
 205                 #Create a site dictionary with key is site_login_base (name of the site)
 206                 # and value is a dictionary of properties, including the list of the node_ids
 207                 for node_id in self.node_dictlist.keys():
 208                         node  = self.node_dictlist[node_id]
 209                         if node['site_id'] not in self.site_dict.keys():
 210                                 self.site_dict[node['site_login_base']] = [('site_id',node['site_id']),\
 211                                                                          ('login_base', node['site_login_base']),\
 212                                                                         ('node_ids',nodes_per_site[node['site_id']]),\
 213                                                                         ('latitude',"48.83726"),\
 214                                                                         ('longitude',"- 2.10336"),('name',"demolab"),\
 215                                                                         ('pcu_ids', []), ('max_slices', None), ('ext_consortium_id', None),\
 216                                                                         ('max_slivers', None), ('is_public', True), ('peer_site_id', None),\
 217                                                                         ('abbreviated_name', "demolab"), ('address_ids', []),\
 218                                                                         ('url', "http,//www.sdemolab.fr"), ('person_ids', []),\
 219                                                                         ('site_tag_ids', []), ('enabled', True),  ('slice_ids', []),\
 220                                                                         ('date_created', None), ('peer_id', None),]
 221                                 self.site_dict[node['site_login_base']] = dict(self.site_dict[node['site_login_base']])
 222
 223                 #print>>sys.stderr, "\r\n \r\n =============\t\t ParseSites site dict %s \r\n"%(self.site_dict)
 224
 225
 226         def GetNodesFromOARParse(self):
 227                 #print>>sys.stderr, " \r\n =========GetNodesFromOARParse: node_dictlist %s "%(self.node_dictlist)
 228                 return self.node_dictlist
 229
 230         def GetSitesFromOARParse(self):
 231                 return self.site_dict
 232
 233         def GetJobsFromOARParse(self):
 234                 return self.jobs_list
 235
 236         OARrequests_uri_dict = {
 237                 'GET_version': {'uri':'/oarapi/version.json', 'parse_func': ParseVersion},
 238                 'GET_timezone':{'uri':'/oarapi/timezone.json' ,'parse_func': ParseTimezone },
 239                 'GET_jobs': {'uri':'/oarapi/jobs.json','parse_func': ParseJobs},
 240                 'GET_jobs_table': {'uri':'/oarapi/jobs/table.json','parse_func': ParseJobsTable},
 241                 'GET_jobs_details': {'uri':'/oarapi/jobs/details.json','parse_func': ParseJobsDetails},
 242                 'GET_resources_full': {'uri':'/oarapi/resources/full.json','parse_func': ParseResourcesFull},
 243                 'GET_resources':{'uri':'/oarapi/resources.json' ,'parse_func': ParseResources},
 244                 }
 245         resources_fulljson_dict= {
 246                 'resource_id' : AddNodeId,
 247                 'network_address' : AddNodeNetworkAddr,
 248                 'site': AddNodeSite,
 249                 'radio': AddNodeRadio,
 250                 'mobile': AddMobility,
 251                 'posx': AddPosX,
 252                 'posy': AddPosY,
 253                 }
 254
 255
 256         def __init__(self, srv ):
 257                 self.version_json_dict= { 'api_version' : None , 'apilib_version' :None,  'api_timezone': None, 'api_timestamp': None, 'oar_version': None ,}
 258                 self.timezone_json_dict = { 'timezone': None, 'api_timestamp': None, }
 259                 self.jobs_json_dict = { 'total' : None, 'links' : [] , 'offset':None , 'items' : [] , }
 260                 self.jobs_table_json_dict = self.jobs_json_dict
 261                 self.jobs_details_json_dict = self.jobs_json_dict
 262                 self.server = srv
 263                 self.node_dictlist = {}
 264                 self.site_dict = {}
 265                 self.SendRequest("GET_version")
 266
 267         def SendRequest(self,request):
 268                 if request in OARrequests_get_uri_dict:
 269                         self.raw_json = self.server.GETRequestToOARRestAPI(request)
 270                         self.OARrequests_uri_dict[request]['parse_func'](self)
 271                 else:
 272                         print>>sys.stderr, "\r\n OARGetParse __init__ : ERROR_REQUEST " ,request
 273
 274 class OARapi:
 275
 276         def __init__(self):
 277                 self.server = OARrestapi()
 278                 self.parser = OARGETParser(self.server)
 279
 280         def GetNodes(self,node_filter= None, return_fields=None):
 281                 #print>>sys.stderr, " \r\n GetNodes node_filter %s return_fields %s" %(node_filter,return_fields)
 282                 self.parser.SendRequest("GET_resources_full")
 283                 node_dict = self.parser.GetNodesFromOARParse()
 284                 return_node_list = []
 285                 #print>>sys.stderr, " \r\n GetNodes   node_dict %s" %(node_dict)
 286                 if not (node_filter or return_fields):
 287                         return_node_list = node_dict.values()
 288                         return return_node_list
 289
 290                 return_node_list= parse_filter(node_dict.values(),node_filter ,'node', return_fields)
 291                 return return_node_list
 292
 293
 294         def GetSites(self, site_filter= None, return_fields=None):
 295                 print>>sys.stderr, " \r\n GetSites+++++++++++++++++"
 296                 self.parser.SendRequest("GET_resources_full")
 297                 site_dict = self.parser.GetSitesFromOARParse()
 298                 return_site_list = []
 299                 site = site_dict.values()[0]
 300                 Users = SenslabImportUsers()
 301
 302                 #print>>sys.stderr, " \r\n  GetSites sites_dict %s site_filter %s  \r\n \r\n  \r\n \r\n------site %s" %(site_dict,site_filter,site )
 303                 admins_dict ={'person_ids': Users.GetPIs(site['site_id'])}
 304                 site.update(admins_dict)
 305
 306                 slice_list = Users.GetSlices()
 307                 for sl in slice_list:
 308                         #print>>sys.stderr, " \r\n  GetSites sl %s" %(sl)
 309                         if sl['site_id'] == site['site_id']:
 310                                 site['slice_ids'].append(sl['slice_id'])
 311                 #print>>sys.stderr, " \r\n  GetSites -site['site_id'] %s --slice_list %s" %(site['site_id'],slice_list )
 312
 313
 314                 #print>>sys.stderr, " \r\n  GetSites -site['site_id'] %s --admins_dict %s---site %s" %(site['site_id'],admins_dict,site )
 315                 if not (site_filter or return_fields):
 316                         return_site_list = site_dict.values()
 317                         return return_site_list
 318
 319                 return_site_list = parse_filter(site_dict.values(),site_filter ,'site', return_fields)
 320                 return return_site_list
 321
 322
 323         def GetJobs(self):
 324                 print>>sys.stderr, " \r\n GetJobs"
 325                 self.parser.SendRequest("GET_jobs")
 326                 return self.parser.GetJobsFromOARParse()
 327