From 96dbeea171cc7d081ccd269705752722bad146bc Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=D0=90=D0=BD=D1=85=D0=B1=D0=B0=D1=8F=D1=80=20=D0=9B=D1=85?=
 =?UTF-8?q?=D0=B0=D0=B3=D0=B2=D0=B0=D0=B4=D0=BE=D1=80=D0=B6?=
 <L.Ankhbayar@gmail.com>
Date: Tue, 12 Apr 2011 20:23:16 +0800
Subject: [PATCH] Added __hash__, __ne__ with test.

---
 mongoengine/base.py | 11 +++++++++++
 tests/document.py   | 48 +++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 59 insertions(+)

diff --git a/mongoengine/base.py b/mongoengine/base.py
index 6b74cb07..6340e319 100644
--- a/mongoengine/base.py
+++ b/mongoengine/base.py
@@ -489,6 +489,17 @@ class BaseDocument(object):
                 return True
         return False
 
+    def __ne__(self, other):
+        return not self.__eq__(other)
+
+    def __hash__(self):
+        """ For list, dic key  """
+        if self.pk is None:
+            # For new object
+            return super(BaseDocument,self).__hash__()
+        else:
+            return hash(self.pk)
+
 if sys.version_info < (2, 5):
     # Prior to Python 2.5, Exception was an old-style class
     def subclass_exception(name, parents, unused):
diff --git a/tests/document.py b/tests/document.py
index c0567632..280b671e 100644
--- a/tests/document.py
+++ b/tests/document.py
@@ -627,6 +627,54 @@ class DocumentTest(unittest.TestCase):
     def tearDown(self):
         self.Person.drop_collection()
 
+    def test_document_hash(self):
+        """Test document in list, dict, set
+        """
+        class User(Document):
+            pass
+
+        class BlogPost(Document):
+            pass
+        
+        # Clear old datas
+        User.drop_collection()
+        BlogPost.drop_collection()
+
+        u1 = User.objects.create()
+        u2 = User.objects.create()
+        u3 = User.objects.create()
+        u4 = User() # New object
+
+        b1 = BlogPost.objects.create()
+        b2 = BlogPost.objects.create()
+
+        # in List
+        all_user_list = list(User.objects.all())
+
+        self.assertTrue(u1 in all_user_list)
+        self.assertTrue(u2 in all_user_list)
+        self.assertTrue(u3 in all_user_list)
+        self.assertFalse(u4 in all_user_list) # New object
+        self.assertFalse(b1 in all_user_list) # Other object
+        self.assertFalse(b2 in all_user_list) # Other object
+
+        # in Dict
+        all_user_dic = {}
+        for u in User.objects.all():
+            all_user_dic[u] = "OK"
+
+        self.assertEqual(all_user_dic.get(u1, False), "OK" )
+        self.assertEqual(all_user_dic.get(u2, False), "OK" )
+        self.assertEqual(all_user_dic.get(u3, False), "OK" )
+        self.assertEqual(all_user_dic.get(u4, False), False ) # New object
+        self.assertEqual(all_user_dic.get(b1, False), False ) # Other object
+        self.assertEqual(all_user_dic.get(b2, False), False ) # Other object
+
+        # in Set
+        all_user_set = set(User.objects.all())
+        
+        self.assertTrue(u1 in all_user_set )
+        
 
 if __name__ == '__main__':
     unittest.main()