Refactor

Moves from_json to utils and splits the models to files in a module

playlist/models/__init__.py

@@ -0,0 +1,3 @@
+from .track import Track
+from .artist import Artist
+from .album import Album

playlist/models/album.py

@@ -0,0 +1,12 @@
+from django.db import models
+from .artist import Artist
+
+class Album(models.Model):
+    def __str__(self):
+        return self.name
+    id = models.CharField(primary_key=True, max_length=128)
+    name = models.CharField(max_length=255)
+    artists = models.ManyToManyField(Artist)
+    album_type = models.CharField(max_length=255)
+    total_tracks = models.PositiveSmallIntegerField(null=True)
+    image = models.ImageField(upload_to="albums/")

playlist/models/artist.py

@@ -0,0 +1,9 @@
+from django.db import models
+
+class Artist(models.Model):
+    def __str__(self):
+        return self.name
+    id = models.CharField(primary_key=True, max_length=128)
+    name = models.CharField(max_length=255)
+    popularity = models.SmallIntegerField(null=True, blank=True)
+    genres = models.JSONField(null=True)

playlist/models/track.py

@@ -0,0 +1,32 @@
+from django.db import models
+from .album import Album
+from .artist import Artist
+from .utils import from_json
+
+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}'"
+        if self.artists.count() > 0:
+            artists = list(map(lambda a:a.name, self.artists.all()))
+            s += " by "
+            s += " & ".join(artists)
+        return s
+    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)
+
+    old = models.BooleanField(default=False)
+
+    objects = TrackManager()
+    all_tracks = models.Manager()

playlist/models/utils/__init__.py

@@ -0,0 +1 @@
+from .json import from_json

playlist/models/utils/json.py

@@ -0,0 +1,8 @@
+def from_json(cls, json):
+    try:
+        return cls.objects.get(pk=json["id"])
+    except cls.DoesNotExist:
+        keys = [f.name for f in cls._meta.get_fields()]
+        subset = {key:json[key] for key in set(keys) & set(json.keys())}
+        return cls.objects.create(**subset)
+