spotify-lib-vis/api/models.py

from django.db import models
from login.models import User

# id's are 22 in length in examples but set to 30 for buffer
MAX_ID = 30

#  Genre {{{ # 

class Genre(models.Model):
            
    class Meta:
        verbose_name = "Genre"
        verbose_name_plural = "Genres"

    name = models.CharField(primary_key=True, max_length=50)
    num_songs = models.PositiveIntegerField()

    def __str__(self):
        return self.name

#  }}} Genre # 

#  Artist {{{ # 

class Artist(models.Model):
    class Meta:
        verbose_name = "Artist"
        verbose_name_plural = "Artists"

    id = models.CharField(primary_key=True, max_length=MAX_ID)
    name = models.CharField(max_length=50)
    genres = models.ManyToManyField(Genre, blank=True)
    # genre = models.ForeignKey(Genre, on_delete=models.CASCADE, blank=True,
    #                           null=True)

    def __str__(self):
        return self.name

#  }}} Artist # 

#  Track {{{ # 

class Track(models.Model):
    
    class Meta:
        verbose_name = "Track"
        verbose_name_plural = "Tracks"

    id = models.CharField(primary_key=True, max_length=MAX_ID)
    artists = models.ManyToManyField(Artist, blank=True)
    year = models.PositiveSmallIntegerField(blank=True, null=True)
    popularity = models.PositiveSmallIntegerField()
    runtime = models.PositiveSmallIntegerField()
    name = models.CharField(max_length=200)
    users = models.ManyToManyField(User, blank=True)
    genre = models.ForeignKey(Genre, on_delete=models.CASCADE, blank=True,
                              null=True)

    def __str__(self):
        track_str = "{}, genre: {}, artists: [".format(self.name, self.genre)
        for artist in self.artists.all():
            track_str += "{}, ".format(artist.name)
        track_str += "]"
        return track_str

#  }}} Track # 

#  AudioFeatures {{{ # 

class AudioFeatures(models.Model):
        
    class Meta:
        verbose_name = "AudioFeatures"
        verbose_name_plural = "AudioFeatures"

    track = models.OneToOneField(Track, on_delete=models.CASCADE, primary_key=True,)
    acousticness = models.DecimalField(decimal_places=3, max_digits=3)
    danceability = models.DecimalField(decimal_places=3, max_digits=3)
    energy = models.DecimalField(decimal_places=3, max_digits=3)
    instrumentalness = models.DecimalField(decimal_places=3, max_digits=3)
    loudness = models.DecimalField(decimal_places=3, max_digits=6)
    speechiness = models.DecimalField(decimal_places=3, max_digits=3)
    tempo = models.DecimalField(decimal_places=3, max_digits=6)
    valence = models.DecimalField(decimal_places=3, max_digits=3)

    def __str__(self):
        return super(AudioFeatures, self).__str__()

#  }}} AudioFeatures #

#  History {{{ # 

class History(models.Model):
        
    class Meta:
        verbose_name = "History"
        verbose_name_plural = "History"
        unique_together = (("user", "timestamp"),)

    id = models.AutoField(primary_key=True)
    user = models.ForeignKey(User, on_delete=models.CASCADE)
    timestamp = models.DateTimeField()
    track = models.ForeignKey(Track, on_delete=models.CASCADE)

    def __str__(self):
        return " - ".join((str(self.user), str(self.timestamp), str(self.track)))

    def get_artists(self):
        artist_names = [artist.name for artist in self.track.artists.all()]
        return ', '.join(artist_names)

    def get_iso_timestamp(self):
        return self.timestamp.isoformat()

#  }}} #