Merge pull request #35 from fayazkhan/level

Category level

Author: uralbash

sqlalchemy_mptt/events.py

@@ -426,7 +426,7 @@ def after_flush_postexec(self, session, context):
                 session.expire(parent, ['left', 'right'])
                 parent = self.get_parent_value(parent)
             else:
-                session.expire(instance, ['tree_id', ])
+                session.expire(instance, ['tree_id', 'level'])
 
     @staticmethod
     def get_parent_value(instance):

sqlalchemy_mptt/tests/tree_testing_base.py

@@ -1734,6 +1734,7 @@ def test_session_expire_for_move_after_to_new_tree(self):
         node.move_after('1')
         self.session.flush()
         self.assertEqual(node.tree_id, 2)
+        self.assertEqual(node.level, 1)
         self.assertEqual(node.parent_id, None)
 
         children = self.session.query(self.model)\