Files
muzak/playlist/models/track.py
T
2024-02-19 21:18:53 +01:00

49 lines
1.5 KiB
Python

from django.db import models
from random import choice
from .album import Album
from .artist import Artist
import re
class TrackManager(models.Manager):
def get_queryset(self):
return super().get_queryset().filter(old=False)
class Track(models.Model):
def __str__(self):
s = f"'{self.name}'"
s += f" AKA '{self.name_sanitized}'"
if self.artists.count() > 0:
artists = list(map(lambda a:a.name, self.artists.all()))
s += " door "
s += " & ".join(artists)
return s
@property
def artist(self):
if self.artists.count() > 0:
artists = list(map(lambda a:a.name, self.artists.all()))
return " & ".join(artists)
@property
def name_sanitized(self):
return re.match("[^\-\(]*\w", self.name)[0]
def is_duplicate_of(self):
return Track.objects.exclude(id=self.id).filter(name__startswith=self.name_sanitized).first()
id = models.CharField(primary_key=True, max_length=128)
name = models.CharField(max_length=255)
artists = models.ManyToManyField(Artist)
album = models.ForeignKey(Album, null=True, on_delete=models.SET_NULL)
explicit = models.BooleanField(default=False)
duration_ms = models.PositiveIntegerField(null=True)
popularity = models.PositiveIntegerField(null=True)
banter = models.TextField(blank=True)
banter_done = models.BooleanField(default=False)
old = models.BooleanField(default=False)
objects = TrackManager()
all_tracks = models.Manager()