portal/lsapiclient.py

   1 import sys
   2 import urllib
   3 import urllib2
   4 import json
   5 import socket
   6 from urlparse import urlparse
   7 from database import Database
   8
   9 class LaboraSchedulerClient:
  10     """
  11     A simple rest shell to a Labora Scheduler instance
  12     This class can receive Labora API  calls to the underlying testbed
  13     """
  14
  15     direct_calls = [ 'get_testbed_info', 'get_users', 'add_user', 'delete_user', 'update_user',
  16                      'get_user_id_by_username', 'add_user_public_key', 'delete_user_public_key' ]
  17
  18     def __init__ ( self, organization ):
  19         self.url, self.key = self.getOrganizationConfigs( organization )
  20
  21
  22     def __getattr__(self, name):
  23
  24         def func(*args, **kwds):
  25             actual_name = None
  26
  27             if name in LaboraSchedulerClient.direct_calls:
  28                 actual_name = name
  29
  30             if not actual_name:
  31                 raise Exception, "Method %s not found in Labora Scheduler"%(name)
  32                 return
  33
  34             if not self.url or not self.key:
  35                 raise Exception, "Missing Labora Scheduler island url and/or key."
  36                 return
  37
  38             address = self.url + "?method=" + actual_name + "&key=" + self.key
  39
  40             # get the direct_call parameters
  41             method_parameters = []
  42
  43             if actual_name == "get_users":
  44                 method_parameters.extend(['filter'])
  45             elif actual_name == "update_user":
  46                 method_parameters.extend(['user_id', 'new_user_data'])
  47             elif actual_name == "delete_user" or actual_name == "delete_user_public_key":
  48                 method_parameters.extend(['user_id'])
  49             elif actual_name == "get_user_id_by_username":
  50                 method_parameters.extend(['username'])
  51             elif actual_name == "add_user":
  52                 method_parameters.extend(['username', 'email', 'password', 'name', 'gidnumber',
  53                                           'homedirectory', 'created_by'])
  54             elif actual_name == "add_user_public_key":
  55                 method_parameters.extend(['user_id', 'public_key'])
  56
  57             for parameter in args:
  58                 if isinstance(parameter, (frozenset, list, set, tuple, dict)):
  59                     for key_name in parameter.keys():
  60
  61                         if key_name in method_parameters:
  62                             param_value = parameter[key_name]
  63
  64                             if param_value == None:
  65                                 continue
  66
  67                             if isinstance(param_value, (frozenset, list, set, tuple, dict)):
  68                                 param_value = json.dumps(param_value)
  69
  70                             param_value = urllib.quote(param_value.encode('utf-8'))
  71
  72                             address += "&" + key_name + "=" + param_value
  73
  74             api_call = urllib2.urlopen(address)
  75             api_call = json.load(api_call)
  76
  77             if not api_call['call_status']:
  78                 result = api_call['method_result']
  79             else:
  80                 result = False
  81
  82
  83             return result
  84
  85         return func
  86
  87
  88     def getOrganizationConfigs( self, organization ):
  89         ls_url = None
  90         ls_key = None
  91
  92         databaseConfig = {
  93             'dbHost'        : '10.128.11.200',
  94             'dbUser'        : 'postgres',
  95             'dbPassword'    : '5e6b70f2e9dc',
  96             'dbName'        : 'LaboraSchedulerNOC'
  97         }
  98
  99         databaseConnection = Database( databaseConfig )
 100
 101         query = "SELECT * FROM islands WHERE domain ='" + organization + "'"
 102         orgConfig = databaseConnection.fetchRows( query )
 103
 104         if orgConfig:
 105             ls_url = orgConfig[0]["ls_url"]
 106             ls_key = orgConfig[0]["ls_key"]
 107
 108         return ls_url, ls_key