Changeset 987aaa1 for fedd/fedd_util.py

Timestamp:

Sep 9, 2008 2:07:18 PM (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:

45ebc4d

Parents:

4fc2250

Message:

get topo and vis data, persistent state

File:

• fedd/fedd_util.py (modified) (3 diffs)

fedd/fedd_util.py

@@ -7 +7 @@
 from M2Crypto import SSL, X509, EVP
 from pyasn1.codec.der import decoder
+
+from xmlrpclib import Binary
 
 
@@ -89 +91 @@
 
     def pack_soap(self):
+        return self.buf
+
+    def pack_xmlrpc(self):
         return self.buf
 
@@ -227 +232 @@
     else: 
         return element
+
+def encapsulate_binaries(e, tags):
+    """Walk through a message and encapsulate any dictionary entries in 
+    tags into a binary object."""
+    dict_type = type(dict())
+    list_type = type(list())
+    str_type = type(str())
+
+    if isinstance(e, dict_type):
+        for k in e.keys():
+            if k in tags:
+                if isinstance(e[k], list_type):
+                    bin_list = []
+                    for ee in e[k]:
+                        if getattr(ee, 'pack_xmlrpc', None):
+                            bin_list.append(Binary(ee.pack_xmlrpc()))
+                        else:
+                            bin_list.append(Binary(ee))
+                    e[k] = bin_list
+                elif getattr(e[k],'pack_xmlrpc', None):
+                    e[k] = Binary(e[k].pack_xmlrpc())
+                else:
+                    e[k] = Binary(e[k])
+            elif isinstance(e[k], dict_type):
+                encapsulate_binaries(e[k], tags)
+            elif isinstance(e[k], list_type):
+                for ee in e[k]:
+                    encapsulate_binaries(ee, tags)
+    # Other types end the recursion - they should be leaves
+    return e
+
+def decapsulate_binaries(e, tags):
+    """Walk through a message and encapsulate any dictionary entries in 
+    tags into a binary object."""
+    dict_type = type(dict())
+    list_type = type(list())
+    str_type = type(str())
+
+    if isinstance(e, dict_type):
+        for k in e.keys():
+            if k in tags:
+                if isinstance(e[k], list_type):
+                    e[k] = [ b.data for b in e[k]]
+                else:
+                    e[k] = e[k].data
+            elif isinstance(e[k], dict_type):
+                decapsulate_binaries(e[k], tags)
+            elif isinstance(e[k], list_type):
+                for ee in e[k]:
+                    decapsulate_binaries(ee, tags)
+    # Other types end the recursion - they should be leaves
+    return e
+
 
 def generate_fedid(subj, bits=2048, trace=None, dir=None):