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

duplicate removal of submit_parts, sorting for description

Browse Source
This commit is contained in:
aj 2019-08-05 23:08:57 +01:00
parent f17ac71168
commit 2768854a9e
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
spotframework/google/run_user_playlist.py
View File

@ -59,32 +59,37 @@ def run_user_playlist(username, playlist_name):
submit_parts = playlist_dict['parts'] + generate_parts(users[0].id, playlist_dict['name']) submit_parts = playlist_dict['parts'] + generate_parts(users[0].id, playlist_dict['name'])
submit_parts = [i for i in {j for j in submit_parts}]
if playlist_dict['type'] == 'recents': if playlist_dict['type'] == 'recents':
boundary_date = datetime.datetime.now() - datetime.timedelta(days=int(playlist_dict['day_boundary'])) boundary_date = datetime.datetime.now() - datetime.timedelta(days=int(playlist_dict['day_boundary']))
tracks = engine.get_recent_playlist(boundary_date, tracks = engine.get_recent_playlist(boundary_date,
submit_parts, submit_parts,
processors, processors,
include_recommendations=playlist_dict['include_recommendations'], include_recommendations=playlist_dict['include_recommendations'],
recommendation_limit=playlist_dict['recommendation_sample']) recommendation_limit=int(playlist_dict['recommendation_sample']))
else: else:
tracks = engine.make_playlist(submit_parts, tracks = engine.make_playlist(submit_parts,
processors, processors,
include_recommendations=playlist_dict['include_recommendations'], include_recommendations=playlist_dict['include_recommendations'],
recommendation_limit=playlist_dict['recommendation_sample']) recommendation_limit=int(playlist_dict['recommendation_sample']))
engine.execute_playlist(tracks, playlist_dict['playlist_id']) engine.execute_playlist(tracks, playlist_dict['playlist_id'])
engine.change_description({i for i in submit_parts}, playlist_dict['playlist_id']) engine.change_description(sorted(submit_parts), playlist_dict['playlist_id'])
else: else:
raise Exception('multiple playlists found') raise Exception('multiple/no playlists found')
else: else:
raise Exception('not one user found') raise Exception('multiple/no user(s) found')
def generate_parts(user_id, name): def generate_parts(user_id, name):
playlist_doc = [i.to_dict() for i in db.document(u'spotify_users/{}'.format(user_id)).collection(u'playlists').where(u'name', '==', name).stream()][0] playlist_doc = [i.to_dict() for i in
db.document(u'spotify_users/{}'.format(user_id))
.collection(u'playlists')
.where(u'name', '==', name).stream()][0]
return_parts = playlist_doc['parts'] return_parts = playlist_doc['parts']