from django.db import models
from client.models import Client


class Project(models.Model):
    name = models.CharField(max_length=30)
    project_image = models.ImageField(upload_to='project_images', null=True, blank=True)
    client = models.ForeignKey(Client, on_delete=models.CASCADE)
    total_cost = models.DecimalField()
    start_date = models.DateField()
    end_date = models.DateField()

    def __str__(self):
        return self.name