added get_next_value to SequenceField

2013-05-16 12:50:34 -03:00
parent 5e94637adc
commit 731d8fc6be
2 changed files with 28 additions and 0 deletions
--- a/mongoengine/fields.py
+++ b/mongoengine/fields.py
@@ -1465,6 +1465,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 data['next']
+
+        return 1
+
    def get_sequence_name(self):
        if self.sequence_name:
            return self.sequence_name