diff --git a/app/api/playlist.py b/app/api/playlist.py index f73a247e..89b1a49d 100644 --- a/app/api/playlist.py +++ b/app/api/playlist.py @@ -177,6 +177,7 @@ def get_playlist(playlistid: str): playlist.last_updated = date_string_to_time_passed(playlist.last_updated) playlist.set_duration(duration) + playlist.set_count(len(tracks)) if not playlist.has_image: playlist.images = get_first_4_images(tracks) diff --git a/app/db/sqlite/utils.py b/app/db/sqlite/utils.py index 7127f165..69b1d5c5 100644 --- a/app/db/sqlite/utils.py +++ b/app/db/sqlite/utils.py @@ -2,7 +2,6 @@ Helper functions for use with the SQLite database. """ -from pprint import pprint import sqlite3 from sqlite3 import Connection, Cursor import time @@ -46,7 +45,6 @@ def tuple_to_playlist(playlist: tuple): """ Takes a tuple and returns a Playlist object """ - pprint(playlist) return Playlist(*playlist) diff --git a/app/migrations/__init__.py b/app/migrations/__init__.py index 95e07cb7..08faa26b 100644 --- a/app/migrations/__init__.py +++ b/app/migrations/__init__.py @@ -41,5 +41,5 @@ def apply_migrations(): log.info("Running new migration: %s", m.name) m.migrate() - # update migration version + # bump migration version MigrationManager.set_version(len(migrations)) diff --git a/app/migrations/v1_3_0/__init__.py b/app/migrations/v1_3_0/__init__.py index a7a49bfa..6776e72d 100644 --- a/app/migrations/v1_3_0/__init__.py +++ b/app/migrations/v1_3_0/__init__.py @@ -110,6 +110,7 @@ class AddSettingsToPlaylistTable: # (my first coroutine) pipeline = insert_playlist() transform_playlists(pipeline, playlists) + pipeline.close() cur.close() diff --git a/app/models/playlist.py b/app/models/playlist.py index ed7569c3..9c15befc 100644 --- a/app/models/playlist.py +++ b/app/models/playlist.py @@ -44,6 +44,9 @@ class Playlist: def set_duration(self, duration: int): self.duration = duration + + def set_count(self, count: int): + self.count = count def clear_lists(self): """ diff --git a/app/requests/artists.py b/app/requests/artists.py index dc329821..2da19c2c 100644 --- a/app/requests/artists.py +++ b/app/requests/artists.py @@ -1,7 +1,6 @@ """ Requests related to artists """ -from pprint import pprint import requests from app import settings