Skip to content

Commit

Permalink
Merge branch 'master' of https://github.com/nikitinsm/mongoengine
Browse files Browse the repository at this point in the history
  • Loading branch information
rozza committed Aug 24, 2012
2 parents ea666d4 + 7e64bb2 commit 89c67bf
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions mongoengine/fields.py
Original file line number Diff line number Diff line change
Expand Up @@ -1324,17 +1324,18 @@ class SequenceField(IntField):
.. versionadded:: 0.5
"""
def __init__(self, collection_name=None, db_alias = None, *args, **kwargs):
def __init__(self, collection_name=None, db_alias = None, sequence_name = None, *args, **kwargs):
self.collection_name = collection_name or 'mongoengine.counters'
self.db_alias = db_alias or DEFAULT_CONNECTION_NAME
self.sequence_name = sequence_name
return super(SequenceField, self).__init__(*args, **kwargs)

def generate_new_value(self):
"""
Generate and Increment the counter
"""
sequence_id = "%s.%s" % (self.owner_document._get_collection_name(),
self.name)
sequence_name = self.sequence_name or self.owner_document._get_collection_name()
sequence_id = "%s.%s" % (sequence_name, self.name)
collection = get_db(alias=self.db_alias)[self.collection_name]
counter = collection.find_and_modify(query={"_id": sequence_id},
update={"$inc": {"next": 1}},
Expand Down

0 comments on commit 89c67bf

Please sign in to comment.