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