django/tests/forms_tests/models.py

import datetime
import itertools
import tempfile

from django.core.files.storage import FileSystemStorage
from django.db import models

callable_default_counter = itertools.count()


def callable_default():
    return next(callable_default_counter)


temp_storage = FileSystemStorage(location=tempfile.mkdtemp())


class BoundaryModel(models.Model):
    positive_integer = models.PositiveIntegerField(null=True, blank=True)


class Defaults(models.Model):
    name = models.CharField(max_length=255, default="class default value")
    def_date = models.DateField(default=datetime.date(1980, 1, 1))
    value = models.IntegerField(default=42)
    callable_default = models.IntegerField(default=callable_default)


class ChoiceModel(models.Model):
    """For ModelChoiceField and ModelMultipleChoiceField tests."""

    CHOICES = [
        ("", "No Preference"),
        ("f", "Foo"),
        ("b", "Bar"),
    ]

    INTEGER_CHOICES = [
        (None, "No Preference"),
        (1, "Foo"),
        (2, "Bar"),
    ]

    STRING_CHOICES_WITH_NONE = [
        (None, "No Preference"),
        ("f", "Foo"),
        ("b", "Bar"),
    ]

    name = models.CharField(max_length=10)
    choice = models.CharField(max_length=2, blank=True, choices=CHOICES)
    choice_string_w_none = models.CharField(
        max_length=2, blank=True, null=True, choices=STRING_CHOICES_WITH_NONE
    )
    choice_integer = models.IntegerField(choices=INTEGER_CHOICES, blank=True, null=True)


class ChoiceOptionModel(models.Model):
    """
    Destination for ChoiceFieldModel's ForeignKey.
    Can't reuse ChoiceModel because error_message tests require that it have no
    instances.
    """

    name = models.CharField(max_length=10)

    class Meta:
        ordering = ("name",)

    def __str__(self):
        return "ChoiceOption %d" % self.pk


def choice_default():
    return ChoiceOptionModel.objects.get_or_create(name="default")[0].pk


def choice_default_list():
    return [choice_default()]


def int_default():
    return 1


def int_list_default():
    return [1]


class ChoiceFieldModel(models.Model):
    """Model with ForeignKey to another model, for testing ModelForm
    generation with ModelChoiceField."""

    choice = models.ForeignKey(
        ChoiceOptionModel,
        models.CASCADE,
        blank=False,
        default=choice_default,
    )
    choice_int = models.ForeignKey(
        ChoiceOptionModel,
        models.CASCADE,
        blank=False,
        related_name="choice_int",
        default=int_default,
    )
    multi_choice = models.ManyToManyField(
        ChoiceOptionModel,
        blank=False,
        related_name="multi_choice",
        default=choice_default_list,
    )
    multi_choice_int = models.ManyToManyField(
        ChoiceOptionModel,
        blank=False,
        related_name="multi_choice_int",
        default=int_list_default,
    )


class OptionalMultiChoiceModel(models.Model):
    multi_choice = models.ManyToManyField(
        ChoiceOptionModel,
        blank=False,
        related_name="not_relevant",
        default=choice_default,
    )
    multi_choice_optional = models.ManyToManyField(
        ChoiceOptionModel,
        blank=True,
        related_name="not_relevant2",
    )


class FileModel(models.Model):
    file = models.FileField(storage=temp_storage, upload_to="tests")


class Article(models.Model):
    content = models.TextField()