from django.contrib.auth.models import User
from django.db import models
from pycs_api.models import base
from pycs_api.models.model import Model
from pycs_api.models.label_provider import LabelProvider

class Project(base.BaseModel):

    user = models.ForeignKey(
        User,
        on_delete=models.CASCADE,
        related_name="projects",
        related_query_name="project",
    )

    description = models.TextField()

    created = models.DateTimeField(auto_now_add=True)

    model = models.ForeignKey(
        Model,
        null=True,
        blank=True,
        on_delete=models.SET_NULL
    )

    label_provider = models.ForeignKey(
        LabelProvider,
        null=True,
        blank=True,
        on_delete=models.SET_NULL
    )

    root_folder = models.CharField(max_length=255, unique=True)

    data_folder = models.CharField(max_length=255)

    external_data = models.BooleanField()