from django.db import models

from apps.map.models import CaseStudy, CaseStudyDraft


class BaseFile(models.Model):
    file = models.FileField(
        upload_to='.',
    )
    case_study = models.ForeignKey(
        CaseStudy, related_name='files', blank=True, null=True,
    )
    case_study_draft = models.ForeignKey(
        CaseStudyDraft, related_name='files', blank=True, null=True
    )

    class Meta:
        abstract = True


class File(BaseFile):
    pass