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