Ticket #198: caching-class-alias.diff

pyamf/__init__.py

@@ -131 +131 @@
         self.objects = []
         self.rev_objects = {}
 
+        self.class_aliases = {}
+
     def getObject(self, ref):
         """
         Gets an object based on a reference.
@@ -170 +172 @@
 
         return idx
 
+    def getClassAlias(self, klass):
+        """
+        Gets a class alias based on the supplied C{klass}.
+        """
+        if klass not in self.class_aliases:
+            try:
+                self.class_aliases[klass] = get_class_alias(klass)
+            except UnknownClassAlias:
+                self.class_aliases[klass] = None
+
+        return self.class_aliases[klass]
+
     def __copy__(self):
         raise NotImplementedError
 

pyamf/amf0.py

@@ -699 +699 @@
         except pyamf.ReferenceError:
             self.context.addObject(o)
 
-        try:
-            alias = pyamf.get_class_alias(o)
-        except pyamf.UnknownClassAlias:
-            alias = None
+        alias = self.context.getClassAlias(o.__class__)
 
         if alias is None:
             self.writeType(ASTypes.OBJECT)

pyamf/amf3.py

@@ -1492 +1492 @@
         """
         encoding = ObjectEncoding.DYNAMIC
 
-        try:
-            alias = pyamf.get_class_alias(obj)
-        except pyamf.UnknownClassAlias:
-            alias = None
+        alias = self.context.getClassAlias(obj.__class__)
 
         if alias:
             if 'dynamic' in alias.metadata: