from django.db import models
from core.models import BaseModel


class Artist(BaseModel):
    name = models.CharField(max_length=200)

    def __str__(self):
        return f"{self.name}"


class Album(BaseModel):
    artist = models.ForeignKey(Artist, on_delete=models.CASCADE)
    name = models.CharField(max_length=200)

    def __str__(self):
        return f"{self.artist} - {self.name}"


class Track(BaseModel):
    title = models.CharField(max_length=200)
    album = models.ForeignKey(Album, blank=True, null=True, default=None, on_delete=models.SET_DEFAULT)
    file = models.FileField(upload_to="music/")

    def __str__(self):
        return f"{self.album.artist} - {self.title}"