Libraries for Subliminal

2011-12-23 16:02:48 +01:00
parent a03c33c1c3
commit 8efc88e691
109 changed files with 26894 additions and 0 deletions
@@ -0,0 +1,121 @@
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the (LGPL) GNU Lesser General Public License as
+# published by the Free Software Foundation; either version 3 of the 
+# License, or (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Library Lesser General Public License for more details at
+# ( http://www.gnu.org/licenses/lgpl.html ).
+#
+# You should have received a copy of the GNU Lesser General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
+# written by: Jeff Ortel ( jortel@redhat.com )
+
+"""
+The I{builder} module provides an wsdl/xsd defined types factory
+"""
+
+from logging import getLogger
+from suds import *
+from suds.sudsobject import Factory
+
+log = getLogger(__name__)
+
+
+class Builder:
+    """ Builder used to construct an object for types defined in the schema """
+    
+    def __init__(self, resolver):
+        """
+        @param resolver: A schema object name resolver.
+        @type resolver: L{resolver.Resolver}
+        """
+        self.resolver = resolver
+        
+    def build(self, name):
+        """ build a an object for the specified typename as defined in the schema """
+        if isinstance(name, basestring):
+            type = self.resolver.find(name)
+            if type is None:
+                raise TypeNotFound(name)
+        else:
+            type = name
+        cls = type.name
+        if type.mixed():
+            data = Factory.property(cls)
+        else:
+            data = Factory.object(cls)
+        resolved = type.resolve()
+        md = data.__metadata__
+        md.sxtype = resolved
+        md.ordering = self.ordering(resolved)
+        history = []
+        self.add_attributes(data, resolved)
+        for child, ancestry in type.children():
+            if self.skip_child(child, ancestry):
+                continue
+            self.process(data, child, history[:])
+        return data
+            
+    def process(self, data, type, history):
+        """ process the specified type then process its children """
+        if type in history:
+            return
+        if type.enum():
+            return
+        history.append(type)
+        resolved = type.resolve()
+        value = None
+        if type.unbounded():
+            value = []
+        else:
+            if len(resolved) > 0:
+                if resolved.mixed():
+                    value = Factory.property(resolved.name)
+                    md = value.__metadata__
+                    md.sxtype = resolved
+                else:
+                    value = Factory.object(resolved.name)
+                    md = value.__metadata__
+                    md.sxtype = resolved
+                    md.ordering = self.ordering(resolved)
+        setattr(data, type.name, value)
+        if value is not None:
+            data = value
+        if not isinstance(data, list):
+            self.add_attributes(data, resolved)
+            for child, ancestry in resolved.children():
+                if self.skip_child(child, ancestry):
+                    continue
+                self.process(data, child, history[:])
+
+    def add_attributes(self, data, type):
+        """ add required attributes """
+        for attr, ancestry in type.attributes():
+            name = '_%s' % attr.name
+            value = attr.get_default()
+            setattr(data, name, value)
+                
+    def skip_child(self, child, ancestry):
+        """ get whether or not to skip the specified child """
+        if child.any(): return True
+        for x in ancestry:
+            if x.choice():
+                return True
+        return False
+    
+    def ordering(self, type):
+        """ get the ordering """
+        result = []
+        for child, ancestry in type.resolve():
+            name = child.name
+            if child.name is None:
+                continue
+            if child.isattr():
+                name = '_%s' % child.name
+            result.append(name)
+        return result
+