diff options
Diffstat (limited to 'tests/modeltests/custom_pk/models.py')
| -rw-r--r-- | tests/modeltests/custom_pk/models.py | 90 |
1 files changed, 90 insertions, 0 deletions
diff --git a/tests/modeltests/custom_pk/models.py b/tests/modeltests/custom_pk/models.py new file mode 100644 index 0000000000..df127ba3a8 --- /dev/null +++ b/tests/modeltests/custom_pk/models.py @@ -0,0 +1,90 @@ +""" +14. Using a custom primary key + +By default, Django adds an ``"id"`` field to each model. But you can override +this behavior by explicitly adding ``primary_key=True`` to a field. +""" + +from django.db import models + +class Employee(models.Model): + employee_code = models.CharField(maxlength=10, primary_key=True) + first_name = models.CharField(maxlength=20) + last_name = models.CharField(maxlength=20) + class Meta: + ordering = ('last_name', 'first_name') + + def __repr__(self): + return "%s %s" % (self.first_name, self.last_name) + +class Business(models.Model): + name = models.CharField(maxlength=20, primary_key=True) + employees = models.ManyToManyField(Employee) + class Meta: + verbose_name_plural = 'businesses' + + def __repr__(self): + return self.name + +API_TESTS = """ +>>> dan = Employee(employee_code='ABC123', first_name='Dan', last_name='Jones') +>>> dan.save() +>>> Employee.objects.all() +[Dan Jones] + +>>> fran = Employee(employee_code='XYZ456', first_name='Fran', last_name='Bones') +>>> fran.save() +>>> Employee.objects.all() +[Fran Bones, Dan Jones] + +>>> Employee.objects.get(pk='ABC123') +Dan Jones +>>> Employee.objects.get(pk='XYZ456') +Fran Bones +>>> Employee.objects.get(pk='foo') +Traceback (most recent call last): + ... +DoesNotExist: Employee does not exist for {'pk': 'foo'} + +# Use the name of the primary key, rather than pk. +>>> Employee.objects.get(employee_code__exact='ABC123') +Dan Jones + +# Fran got married and changed her last name. +>>> fran = Employee.objects.get(pk='XYZ456') +>>> fran.last_name = 'Jones' +>>> fran.save() +>>> Employee.objects.filter(last_name__exact='Jones') +[Dan Jones, Fran Jones] +>>> Employee.objects.in_bulk(['ABC123', 'XYZ456']) +{'XYZ456': Fran Jones, 'ABC123': Dan Jones} + +>>> b = Business(name='Sears') +>>> b.save() +>>> b.employees.add(dan, fran) +>>> b.employees.all() +[Dan Jones, Fran Jones] +>>> fran.business_set.all() +[Sears] +>>> Business.objects.in_bulk(['Sears']) +{'Sears': Sears} + +>>> Business.objects.filter(name__exact='Sears') +[Sears] +>>> Business.objects.filter(pk='Sears') +[Sears] + +# Queries across tables, involving primary key +>>> Employee.objects.filter(business__name__exact='Sears') +[Dan Jones, Fran Jones] +>>> Employee.objects.filter(business__pk='Sears') +[Dan Jones, Fran Jones] + +>>> Business.objects.filter(employees__employee_code__exact='ABC123') +[Sears] +>>> Business.objects.filter(employees__pk='ABC123') +[Sears] +>>> Business.objects.filter(employees__first_name__startswith='Fran') +[Sears] + +""" |
