Implementation of auto_increment for MongoDB

mongodb_autoincrement.py

"""

db.seq.insert({_id: 'pacientes', seq: 1})

"""

from django.conf import settings
from pymongo import Connection

DATABASES = settings.DATABASES

connection = Connection(DATABASES['default']['HOST'], DATABASES['default']['PORT'])
mongo = connection[DATABASES['default']['NAME']]

if len(DATABASES['default']['USER']) > 0:
    mongo.authenticate(DATABASES['default']['USER'], DATABASES['default']['PASSWORD'])


def auto_inc(collection='pacientes'):
    new_id = mongo.seq.find_and_modify(
        query={'_id': collection},
        update={'$inc': {'seq': 1}},
        new=True
    )
    return new_id['seq']