Skip to content

cordery/django-autosequence

Repository files navigation

Django AutoSequence

https://badge.fury.io/py/django-autosequence.png https://travis-ci.org/cordery/django-autosequence.png?branch=master

A model field for Django that provides for a configurable automatic sequence of values

Documentation

The full documentation is at https://django-autosequence.readthedocs.org.

Quickstart

Install Django AutoSequence:

pip install django-autosequence

Add the AutoSequenceField to your models like so:

class ModelWithStartAt(Model):
    sequence = AutoSequenceField(start_at=100)


class ModelWithUnique(Model):
    name = CharField(max_length=200)
    sequence = AutoSequenceField(unique_with='name')


class ModelWithUniqueCombo(Model):
    name = CharField(max_length=200)
    simple_model = ForeignKey(SimpleModel, on_delete=CASCADE)
    sequence = AutoSequenceField(unique_with=('simple_model', 'name'))

Description

AutoSequenceField is an AutoField that is available for non primary keys and can be configured with unique_with to have separate sequences based on other model fields.

For example, say you have a model Invoices that tracks Invoices you have issued to different customers and you would like the invoice number to be autoincrementing but unique for each customer. You would add a field like:

class Invoice(models.Model):
    customer = models.ForeignKey('Customer')
    invoice_number = AutoSequenceField(unique_with='customer')

It does this but running a select query against the model to find the highest sequence number for the given set of unique_with fields. This means that if you delete the last object created, the sequence number of that object will be reused next time you create a new object.

This behavior may be undesirable in some circumstances however in general this field type is intended to be used with models where deletions are infrequent or disabled entirely.

It is non editable and unique, either unique for all rows in the table or unique_with another field or fields in the table.

The following keywords may be passed to AutoSequenceField:

start_at: integer: the starting number of the sequence, defaults to 1. Ex: start_at=100

unique_with: string or tuple of strings: the name or names of other fields on this model that this sequence will be unique with. Ex: unique_with='category'

Race Conditions

Due to the simple way this field is implemented (it is not a real SERIAL field), if you will be creating objects concurrently you will need to lock the table before the save.

If you are using postgres try the helpers.lock_table context manager like so:

with lock_table(MyAutoSequenceFieldModel):
    MyAutoSequenceFieldModel.objects.create(..)

This will start a transaction using transaction.atomic() and then execute LOCK TABLE {tbl} IN SHARE ROW EXCLUSIVE MODE against the model's db_table.

Tests have been added that verify the race condition issue and verify the lock_table solution.

Running Tests

Does the code actually work?

$ pip install poetry
$ poetry install
$ poetry run pytest

About

Simple autoincrementing field for Django

Topics

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages