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()