from django.db import models


# Create your models here.

class TeaCategory(models.Model):
    name = models.CharField(max_length=255)
    shop_name = models.CharField(max_length=255)

    def __str__(self):
        return self.name


class TeaType(models.Model):
    name = models.CharField(max_length=255)
    shop_name = models.CharField(max_length=255)
    category = models.ForeignKey(TeaCategory,
                                 related_name='tea_types',
                                 on_delete=models.CASCADE,
                                 blank=True, null=True
                                 )
    one_item = models.BooleanField()
    preferred = models.ForeignKey('Tea',
                                  related_name='tea_type',
                                  on_delete=models.SET_NULL,
                                  blank=True, null=True
                                  )

    def __str__(self):
        return self.name


class Pic(models.Model):
    href = models.CharField(max_length=512)

    def __str__(self):
        return self.href


class Tea(models.Model):
    name = models.CharField(max_length=255)
    price = models.IntegerField()
    pic = models.ForeignKey(Pic,
                            blank=True, null=True,
                            on_delete=models.CASCADE)
    type = models.ForeignKey(TeaType,
                             related_name='teas',
                             on_delete=models.CASCADE)

    def __str__(self):
        return self.name