Changeset ef36c1e for fedd/fedd_allocate_project.py

Timestamp:

Aug 1, 2008 11:41:39 AM (16 years ago)

Author:

Ted Faber <faber@…>

Branches:

axis_example, compt_changes, info-ops, master, version-1.30, version-2.00, version-3.01, version-3.02

Children:

808889e

Parents:

7da9da6

Message:

split out project creation

File:

• fedd/fedd_allocate_project.py (modified) (5 diffs)

fedd/fedd_allocate_project.py

@@ -21 +21 @@
 from fedd_util import *
 import parse_detail
+from service_error import *
 
 class fedd_allocate_project_local:
-    def __init__(self, dp=False, url=None):
+    def __init__(self, dp=False, url=None, certs=None):
         """
         Initializer.  Parses a configuration if one is given.
@@ -58 +59 @@
 
 
-    def dynamic_project(self, req):
+    def dynamic_project(self, req, fedid=None):
         """
         Create a dynamic project with ssh access
@@ -70 +71 @@
                 dir="/tmp")
 
-        print userfile
-        proj = req['project']
+        if req.has_key('AllocateProjectRequestBody') and \
+                req['AllocateProjectRequestBody'].has_key('project'):
+            proj = req['AllocateProjectRequestBody']['project']
+        else:
+            raise service_error(service_error.req, 
+                    "Badly formed allocation request")
         # Take the first user and ssh key
         name = proj.get('name', None) or self.random_string("proj",4)
@@ -91 +96 @@
             access = user.get('access', None)
             if access != None:
-                ssh = access.get('sshPubkey', None)
+                ssh = access[0].get('sshPubkey', None)
                 if ssh == None:
                     raise fedd_proj.service_error(fedd_proj.service_error.req, 
@@ -177 +182 @@
         }
         return rv
+
+
+class fedd_allocate_project_remote:
+    def __init__(self, dp=False, url=None, certs=None):
+        """
+        Initializer.  Parses a configuration if one is given.
+        """
+
+        self.dynamic_projects = dp
+        self.url = url
+
+        if certs != None and isinstance(certs, type(tuple())):
+            self.cert_file, self.trusted_certs, self.cert_pwd = certs
+            # self.cert_file = certs[0]
+            # self.trusted_certs = certs[1]
+            # self.cert_pwd = certs[2]
+        else:
+            self.cert_file = None
+            self.trusted_certs = None
+            self.cert_pwd = None
+
+    def dynamic_project(self, req, fedid=None):
+        """
+        Send req on to a remote project instantiator.
+
+        Req is just the projectAllocType object.  This function re-wraps it.
+        It also rethrows any faults.
+        """
+        # No retry loop here.  Proxy servers must correctly authenticate
+        # themselves without help
+        try:
+            ctx = fedd_ssl_context(self.cert_file, self.trusted_certs,
+                    password=self.cert_pwd)
+        except SSL.SSLError:
+            raise service_error(service_error.server_config, 
+                    "Server certificates misconfigured")
+
+        loc = feddServiceLocator();
+        port = loc.getfeddPortType(self.url,
+                transport=M2Crypto.httpslib.HTTPSConnection, 
+                transdict={ 'ssl_context' : ctx })
+
+        if req.has_key('AllocateProjectRequestBody'):
+            req = req['AllocateProjectRequestBody']
+        else:
+            raise service_error(service_error.req, "Bad formated request");
+
+        # Reconstruct the full request message
+        msg = AllocateProjectRequestMessage()
+        msg.set_element_AllocateProjectRequestBody(
+                pack_soap(msg, "AllocateProjectRequestBody", req))
+        try:
+            resp = port.AllocateProject(msg)
+        except ZSI.ParseException, e:
+            raise service_error(service_error.proxy,
+                    "Bad format message (XMLRPC??): %s" % str(e))
+        r = unpack_soap(resp)
+
+        if r.has_key('AllocateProjectResponseBody'):
+            return r['AllocateProjectResponseBody']
+        else:
+            raise service_error(service_error.proxy, "Bad proxy response")
+