Merge remote-tracking branch 'origin/pr/319'

2013-05-20 08:10:37 +00:00
parent 967da7944f bc92f78afb
commit e402157b4d
3 changed files with 44 additions and 1 deletions
--- a/mongoengine/fields.py
+++ b/mongoengine/fields.py
@@ -1442,6 +1442,17 @@ class SequenceField(BaseField):
                                             upsert=True)
        return self.value_decorator(counter['next'])

+    def get_next_value(self):
+        sequence_name = self.get_sequence_name()
+        sequence_id = "%s.%s" % (sequence_name, self.name)
+        collection = get_db(alias=self.db_alias)[self.collection_name]
+        data = collection.find_one({"_id": sequence_id})
+
+        if data:
+            return self.value_decorator(data['next']+1)
+
+        return self.value_decorator(1)
+
    def get_sequence_name(self):
        if self.sequence_name:
            return self.sequence_name