import datetime

import uuid

from django.db import models

class Author(models.Model):

    name = models.CharField(max_length=100)

    class Meta:

        ordering = ("name",)

    def __str__(self):

        return self.name

class BetterAuthor(Author):

    write_speed = models.IntegerField()

class Book(models.Model):

    author = models.ForeignKey(Author, models.CASCADE)

    title = models.CharField(max_length=100)

    class Meta:

        unique_together = (("author", "title"),)

        ordering = ["id"]

    def __str__(self):

        return self.title

    def clean(self):

        # Ensure author is always accessible in clean method

        assert self.author.name is not None

class BookWithCustomPK(models.Model):

    my_pk = models.DecimalField(max_digits=5, decimal_places=0, primary_key=True)

    author = models.ForeignKey(Author, models.CASCADE)

    title = models.CharField(max_length=100)

    def __str__(self):

        return "%s: %s" % (self.my_pk, self.title)

class Editor(models.Model):

    name = models.CharField(max_length=100)

class BookWithOptionalAltEditor(models.Model):

    author = models.ForeignKey(Author, models.CASCADE)

    # Optional secondary author

    alt_editor = models.ForeignKey(Editor, models.SET_NULL, blank=True, null=True)

    title = models.CharField(max_length=100)

    class Meta:

        unique_together = (("author", "title", "alt_editor"),)

    def __str__(self):

        return self.title

class AlternateBook(Book):

    notes = models.CharField(max_length=100)

    def __str__(self):

        return "%s - %s" % (self.title, self.notes)

class AuthorMeeting(models.Model):

    name = models.CharField(max_length=100)

    authors = models.ManyToManyField(Author)

    created = models.DateField(editable=False)

    def __str__(self):

        return self.name

class CustomPrimaryKey(models.Model):

    my_pk = models.CharField(max_length=10, primary_key=True)

    some_field = models.CharField(max_length=100)

# models for inheritance tests.

class Place(models.Model):

    name = models.CharField(max_length=50)

    city = models.CharField(max_length=50)

    def __str__(self):

        return self.name

class Owner(models.Model):

    auto_id = models.AutoField(primary_key=True)

    name = models.CharField(max_length=100)

    place = models.ForeignKey(Place, models.CASCADE)

    def __str__(self):

        return "%s at %s" % (self.name, self.place)

class Location(models.Model):

    place = models.ForeignKey(Place, models.CASCADE, unique=True)

    # this is purely for testing the data doesn't matter here :)

    lat = models.CharField(max_length=100)

    lon = models.CharField(max_length=100)

class OwnerProfile(models.Model):

    owner = models.OneToOneField(Owner, models.CASCADE, primary_key=True)

    age = models.PositiveIntegerField()

    def __str__(self):

        return "%s is %d" % (self.owner.name, self.age)

class Restaurant(Place):

    serves_pizza = models.BooleanField(default=False)

class Product(models.Model):

    slug = models.SlugField(unique=True)

    def __str__(self):

        return self.slug

class Price(models.Model):

    price = models.DecimalField(max_digits=10, decimal_places=2)

    quantity = models.PositiveIntegerField()

    class Meta:

        unique_together = (("price", "quantity"),)

    def __str__(self):

        return "%s for %s" % (self.quantity, self.price)

class MexicanRestaurant(Restaurant):

    serves_tacos = models.BooleanField(default=False)

class ClassyMexicanRestaurant(MexicanRestaurant):

    the_restaurant = models.OneToOneField(

        MexicanRestaurant, models.CASCADE, parent_link=True, primary_key=True

    )

    tacos_are_yummy = models.BooleanField(default=False)

# models for testing unique_together validation when a fk is involved and

# using inlineformset_factory.

class Repository(models.Model):

    name = models.CharField(max_length=25)

    def __str__(self):

        return self.name

class Revision(models.Model):

    repository = models.ForeignKey(Repository, models.CASCADE)

    revision = models.CharField(max_length=40)

    class Meta:

        unique_together = (("repository", "revision"),)

    def __str__(self):

        return "%s (%s)" % (self.revision, str(self.repository))

# models for testing callable defaults (see bug #7975). If you define a model

# with a callable default value, you cannot rely on the initial value in a

# form.

class Person(models.Model):

    name = models.CharField(max_length=128)

class Membership(models.Model):

    person = models.ForeignKey(Person, models.CASCADE)

    date_joined = models.DateTimeField(default=datetime.datetime.now)

    karma = models.IntegerField()

# models for testing a null=True fk to a parent

class Team(models.Model):

    name = models.CharField(max_length=100)

class Player(models.Model):

    team = models.ForeignKey(Team, models.SET_NULL, null=True)

    name = models.CharField(max_length=100)

    def __str__(self):

        return self.name

# Models for testing custom ModelForm save methods in formsets and inline formsets

class Poet(models.Model):

    name = models.CharField(max_length=100)

    def __str__(self):

        return self.name

class Poem(models.Model):

    poet = models.ForeignKey(Poet, models.CASCADE)

    name = models.CharField(max_length=100)

    def __str__(self):

        return self.name

class Post(models.Model):

    title = models.CharField(max_length=50, unique_for_date="posted", blank=True)

    slug = models.CharField(max_length=50, unique_for_year="posted", blank=True)

    subtitle = models.CharField(max_length=50, unique_for_month="posted", blank=True)

    posted = models.DateField()

    def __str__(self):

        return self.title

# Models for testing UUID primary keys

class UUIDPKParent(models.Model):

    uuid = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)

    name = models.CharField(max_length=255)

class UUIDPKChild(models.Model):

    uuid = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)

    name = models.CharField(max_length=255)

    parent = models.ForeignKey(UUIDPKParent, models.CASCADE)

class ChildWithEditablePK(models.Model):

    name = models.CharField(max_length=255, primary_key=True)

    parent = models.ForeignKey(UUIDPKParent, models.CASCADE)

class AutoPKChildOfUUIDPKParent(models.Model):

    name = models.CharField(max_length=255)

    parent = models.ForeignKey(UUIDPKParent, models.CASCADE)

class AutoPKParent(models.Model):

    name = models.CharField(max_length=255)

class UUIDPKChildOfAutoPKParent(models.Model):

    uuid = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)

    name = models.CharField(max_length=255)

    parent = models.ForeignKey(AutoPKParent, models.CASCADE)

class ParentWithUUIDAlternateKey(models.Model):

    uuid = models.UUIDField(unique=True, default=uuid.uuid4, editable=False)

    name = models.CharField(max_length=50)

class ChildRelatedViaAK(models.Model):

    name = models.CharField(max_length=255)

    parent = models.ForeignKey(

        ParentWithUUIDAlternateKey, models.CASCADE, to_field="uuid"

    )