sarsoo/spotframework
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

added recents processing

Browse Source
This commit is contained in:
aj 2019-08-03 12:19:59 +01:00
parent c221025c5f
commit a93ea5c27d
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
spotframework/google/run_user_playlist.py
View File

@ -1,5 +1,7 @@
from google.cloud import firestore from google.cloud import firestore
import datetime
from spotframework.engine.playlistengine import PlaylistEngine from spotframework.engine.playlistengine import PlaylistEngine
from spotframework.engine.filter.shuffle import Shuffle from spotframework.engine.filter.shuffle import Shuffle
from spotframework.engine.filter.sortreversereleasedate import SortReverseReleaseDate from spotframework.engine.filter.sortreversereleasedate import SortReverseReleaseDate
@ -13,8 +15,7 @@ db = firestore.Client()
def run_user_playlist(username, playlist_name): def run_user_playlist(username, playlist_name):
users = db.collection(u'spotify_users').where(u'username', u'==', username).stream() users = [i for i in db.collection(u'spotify_users').where(u'username', u'==', username).stream()]
users = [i for i in users]
if len(users) == 1: if len(users) == 1:
@ -53,6 +54,10 @@ def run_user_playlist(username, playlist_name):
else: else:
processors.append(SortReverseReleaseDate()) processors.append(SortReverseReleaseDate())
if playlist_dict['type'] == 'recents':
boundary_date = datetime.datetime.now() - datetime.timedelta(days=playlist_dict['day_boundary'])
tracks = engine.get_recent_playlist(boundary_date, playlist_dict['parts'], processors)
else:
tracks = engine.make_playlist(playlist_dict['parts'], processors) tracks = engine.make_playlist(playlist_dict['parts'], processors)
engine.execute_playlist(tracks, playlist_dict['playlist_id']) engine.execute_playlist(tracks, playlist_dict['playlist_id'])
@ -62,4 +67,4 @@ def run_user_playlist(username, playlist_name):
raise Exception('multiple playlists found') raise Exception('multiple playlists found')
else: else:
raise Exception('more than one user found') raise Exception('not one user found')