spotframework/generateplaylists.py

133 lines
4.5 KiB
Python
Raw Normal View History

import spotframework.net.const as const
2019-06-11 11:59:24 +01:00
from spotframework.net.network import Network
from spotframework.net.user import User
import spotframework.log.log as log
import spotframework.io.json as json
import spotframework.util.monthstrings as monthstrings
from spotframework.engine.playlistengine import PlaylistEngine
from spotframework.engine.filter.shuffle import Shuffle
from spotframework.engine.filter.sortreversereleasedate import SortReverseReleaseDate
from spotframework.engine.filter.deduplicatebyid import DeduplicateByID
from spotframework.engine.filter.deduplicatebyname import DeduplicateByName
2019-06-12 15:55:46 +01:00
import os
import datetime
2019-06-12 15:55:46 +01:00
import sys
import requests
2019-06-12 15:55:46 +01:00
def update_super_playlist(engine, data_dict):
2019-06-12 15:55:46 +01:00
log.log("makePlaylist", data_dict['name'])
2019-06-12 15:55:46 +01:00
processors = [DeduplicateByID()]
2019-06-12 15:55:46 +01:00
if 'shuffle' in data_dict:
if data_dict['shuffle'] is True:
processors.append(Shuffle())
else:
processors.append(SortReverseReleaseDate())
else:
processors.append(SortReverseReleaseDate())
2019-06-12 15:55:46 +01:00
tracks = engine.make_playlist(data_dict['playlists'], processors)
2019-06-12 15:55:46 +01:00
engine.execute_playlist(tracks, data_dict['id'])
engine.change_description(data_dict['playlists'], data_dict['id'])
2019-06-12 15:55:46 +01:00
def update_recents_playlist(engine, data):
2019-06-12 15:55:46 +01:00
recents_id = data['recents']['id']
boundary_date = datetime.datetime.now() - datetime.timedelta(days=data['recents']['boundary'])
recent_parts = []
2019-06-12 15:55:46 +01:00
if 'use_marked_playlists' in data['recents']:
if data['recents']['use_marked_playlists']:
for playlist in [i for i in data['playlists'] if 'include_in_recents' in i]:
if playlist['include_in_recents']:
recent_parts += [i for i in playlist['playlists']]
2019-06-12 15:55:46 +01:00
else:
for playlist in [i for i in data['playlists'] if 'include_in_recents' in i]:
if playlist['include_in_recents']:
recent_parts += [i for i in playlist['playlists']]
2019-06-12 15:55:46 +01:00
if 'playlists' in data['recents']:
recent_parts += data['recents']['playlists']
if 'exclude' in data['recents']:
for exclusion in data['recents']['exclude']:
recent_parts.remove(exclusion)
2019-06-12 15:55:46 +01:00
processors = [DeduplicateByName(), SortReverseReleaseDate()]
2019-06-12 15:55:46 +01:00
recent_tracks = engine.get_recent_playlist(boundary_date, recent_parts, processors)
engine.execute_playlist(recent_tracks, recents_id)
engine.change_description([monthstrings.get_this_month(), monthstrings.get_last_month()], data['recents']['id'])
2019-06-10 22:23:03 +01:00
2019-06-12 15:55:46 +01:00
def go():
2019-06-12 15:55:46 +01:00
try:
if os.path.exists(os.path.join(const.config_path, 'config.json')):
2019-06-12 15:55:46 +01:00
data = json.load_json(os.path.join(const.config_path, 'config.json'))
2019-06-12 15:55:46 +01:00
to_execute = []
not_found = []
available_specials = ['recents']
specials_to_execute = []
if len(sys.argv) > 1:
for arg in sys.argv[1:]:
for playlist in data['playlists']:
if arg.lower() == playlist['name'].lower():
to_execute.append(playlist)
break
else:
if arg in available_specials:
specials_to_execute.append(arg)
else:
not_found.append(arg)
else:
to_execute = data['playlists']
specials_to_execute = ['recents']
if len(not_found) > 0:
log.log('arg not found', not_found)
if len(to_execute) <= 0 and len(specials_to_execute) <= 0:
log.log('none to execute, terminating')
return
net = Network(User())
engine = PlaylistEngine(net)
engine.load_user_playlists()
for super_playlist in to_execute:
update_super_playlist(engine, super_playlist)
if 'recents' in data and 'recents' in specials_to_execute:
update_recents_playlist(engine, data)
2019-05-14 22:58:15 +01:00
else:
log.log("config json not found")
if 'SLACKHOOK' in os.environ:
requests.post(os.environ['SLACKHOOK'], json={"text": "spot playlists: config json not found"})
2019-05-14 22:58:15 +01:00
2019-06-11 11:59:24 +01:00
log.dump_log()
except Exception as e:
log.log("exception occured", e)
if 'SLACKHOOK' in os.environ:
requests.post(os.environ['SLACKHOOK'], json={"text": f"spot playlists: exception occured {e}"})
2019-06-11 11:59:24 +01:00
log.dump_log()
2019-06-12 15:55:46 +01:00
if __name__ == '__main__':
go()