Update of flyweight pattern

Author: mlyundin

flyweight.py

@@ -14,7 +14,7 @@ class Card(object):
     """Flyweight implementation. If the object exists in the
     pool just return it (instead of creating a new one)"""
     def __new__(cls, value, suit):
-        obj = Card._CardPool.get(value + suit, None)
+        obj = Card._CardPool.get(value + suit)
         if not obj:
             obj = object.__new__(cls)
             Card._CardPool[value + suit] = obj
@@ -27,16 +27,24 @@ def __new__(cls, value, suit):
     def __repr__(self):
         return "<Card: %s%s>" % (self.value, self.suit)
 
-
 if __name__ == '__main__':
     # comment __new__ and uncomment __init__ to see the difference
     c1 = Card('9', 'h')
     c2 = Card('9', 'h')
     print(c1, c2)
-    print(c1 == c2)
+    print(c1 == c2, c1 is c2)
     print(id(c1), id(c2))
 
+    c1.temp = None
+    c3 = Card('9', 'h')
+    print(hasattr(c3, 'temp'))
+    c1 = c2 = c3 = None
+    c3 = Card('9', 'h')
+    print(hasattr(c3, 'temp'))
+
 ### OUTPUT ###
-# <Card: 9h> <Card: 9h>
+# (<Card: 9h>, <Card: 9h>)
+# (True, True)
+# (31903856, 31903856)
 # True
-# 140368617673296 140368617673296
+# False