diff options
Diffstat (limited to 'tests/admin_widgets/models.py')
-rw-r--r-- | tests/admin_widgets/models.py | 130 |
1 files changed, 130 insertions, 0 deletions
diff --git a/tests/admin_widgets/models.py b/tests/admin_widgets/models.py new file mode 100644 index 0000000000..2977b86f3e --- /dev/null +++ b/tests/admin_widgets/models.py @@ -0,0 +1,130 @@ +from __future__ import unicode_literals + +from django.db import models +from django.contrib.auth.models import User +from django.utils.encoding import python_2_unicode_compatible + + +class MyFileField(models.FileField): + pass + +@python_2_unicode_compatible +class Member(models.Model): + name = models.CharField(max_length=100) + birthdate = models.DateTimeField(blank=True, null=True) + gender = models.CharField(max_length=1, blank=True, choices=[('M','Male'), ('F', 'Female')]) + + def __str__(self): + return self.name + +@python_2_unicode_compatible +class Band(models.Model): + name = models.CharField(max_length=100) + style = models.CharField(max_length=20) + members = models.ManyToManyField(Member) + + def __str__(self): + return self.name + +@python_2_unicode_compatible +class Album(models.Model): + band = models.ForeignKey(Band) + name = models.CharField(max_length=100) + cover_art = models.FileField(upload_to='albums') + backside_art = MyFileField(upload_to='albums_back', null=True) + + def __str__(self): + return self.name + +class HiddenInventoryManager(models.Manager): + def get_query_set(self): + return super(HiddenInventoryManager, self).get_query_set().filter(hidden=False) + +@python_2_unicode_compatible +class Inventory(models.Model): + barcode = models.PositiveIntegerField(unique=True) + parent = models.ForeignKey('self', to_field='barcode', blank=True, null=True) + name = models.CharField(blank=False, max_length=20) + hidden = models.BooleanField(default=False) + + # see #9258 + default_manager = models.Manager() + objects = HiddenInventoryManager() + + def __str__(self): + return self.name + +class Event(models.Model): + band = models.ForeignKey(Band, limit_choices_to=models.Q(pk__gt=0)) + start_date = models.DateField(blank=True, null=True) + start_time = models.TimeField(blank=True, null=True) + description = models.TextField(blank=True) + link = models.URLField(blank=True) + min_age = models.IntegerField(blank=True, null=True) + +@python_2_unicode_compatible +class Car(models.Model): + owner = models.ForeignKey(User) + make = models.CharField(max_length=30) + model = models.CharField(max_length=30) + + def __str__(self): + return "%s %s" % (self.make, self.model) + +class CarTire(models.Model): + """ + A single car tire. This to test that a user can only select their own cars. + """ + car = models.ForeignKey(Car) + +class Honeycomb(models.Model): + location = models.CharField(max_length=20) + +class Bee(models.Model): + """ + A model with a FK to a model that won't be registered with the admin + (Honeycomb) so the corresponding raw ID widget won't have a magnifying + glass link to select related honeycomb instances. + """ + honeycomb = models.ForeignKey(Honeycomb) + +class Individual(models.Model): + """ + A model with a FK to itself. It won't be registered with the admin, so the + corresponding raw ID widget won't have a magnifying glass link to select + related instances (rendering will be called programmatically in this case). + """ + name = models.CharField(max_length=20) + parent = models.ForeignKey('self', null=True) + +class Company(models.Model): + name = models.CharField(max_length=20) + +class Advisor(models.Model): + """ + A model with a m2m to a model that won't be registered with the admin + (Company) so the corresponding raw ID widget won't have a magnifying + glass link to select related company instances. + """ + name = models.CharField(max_length=20) + companies = models.ManyToManyField(Company) + + +@python_2_unicode_compatible +class Student(models.Model): + name = models.CharField(max_length=255) + + def __str__(self): + return self.name + + class Meta: + ordering = ('name',) + +@python_2_unicode_compatible +class School(models.Model): + name = models.CharField(max_length=255) + students = models.ManyToManyField(Student, related_name='current_schools') + alumni = models.ManyToManyField(Student, related_name='previous_schools') + + def __str__(self): + return self.name |