from django.db import models

class Greenhouse(models.Model):
    name = models.CharField(max_length=100)
    location = models.CharField(max_length=200)
    climate_zone = models.CharField(
        max_length=20,
        choices=[("tropical", "Тропики"), ("temperate", "Умеренная"), ("arid", "Аридная")]
    )

    def __str__(self):
        return self.name


class Plant(models.Model):
    name = models.CharField(max_length=150)
    species = models.CharField(max_length=150)
    family = models.CharField(max_length=100, blank=True)
    greenhouse = models.ForeignKey(Greenhouse, on_delete=models.CASCADE, related_name="plants")
    height_cm = models.FloatField()
    watering_frequency_days = models.PositiveIntegerField()
    is_flowering = models.BooleanField(default=False)
    planted_at = models.DateField()
    origin_country = models.CharField(max_length=100, blank=True)
    endangered = models.BooleanField(default=False)

    def __str__(self):
        return self.name