151 lines
5.1 KiB
Python
151 lines
5.1 KiB
Python
import spotframework.net.const as const
|
|
from spotframework.net.network import Network
|
|
from spotframework.net.user import NetworkUser
|
|
import spotframework.io.json as json
|
|
import spotframework.util.monthstrings as monthstrings
|
|
from spotframework.engine.playlistengine import PlaylistEngine
|
|
from spotframework.engine.processor.shuffle import Shuffle
|
|
from spotframework.engine.processor.sort import SortReleaseDate
|
|
from spotframework.engine.processor.deduplicate import DeduplicateByID, DeduplicateByName
|
|
|
|
import os
|
|
import datetime
|
|
import sys
|
|
import logging
|
|
|
|
import requests
|
|
|
|
|
|
logger = logging.getLogger('spotframework')
|
|
|
|
log_format = '%(asctime)s %(levelname)s %(name)s - %(funcName)s - %(message)s'
|
|
|
|
file_handler = logging.FileHandler(".spot/generate_playlists.log")
|
|
formatter = logging.Formatter(log_format)
|
|
file_handler.setFormatter(formatter)
|
|
|
|
logger.addHandler(file_handler)
|
|
|
|
stream_log_format = '%(levelname)s %(name)s:%(funcName)s - %(message)s'
|
|
stream_formatter = logging.Formatter(stream_log_format)
|
|
|
|
stream_handler = logging.StreamHandler()
|
|
stream_handler.setFormatter(stream_formatter)
|
|
|
|
logger.addHandler(stream_handler)
|
|
|
|
|
|
def update_super_playlist(engine, data_dict):
|
|
|
|
logger.info(f"makePlaylist {data_dict['name']}")
|
|
|
|
processors = [DeduplicateByID()]
|
|
|
|
if 'shuffle' in data_dict:
|
|
if data_dict['shuffle'] is True:
|
|
processors.append(Shuffle())
|
|
else:
|
|
processors.append(SortReleaseDate(reverse=True))
|
|
else:
|
|
processors.append(SortReleaseDate(reverse=True))
|
|
|
|
tracks = engine.make_playlist(data_dict['playlists'], processors)
|
|
|
|
engine.execute_playlist(tracks, data_dict['id'])
|
|
engine.change_description(data_dict['playlists'], data_dict['id'])
|
|
|
|
|
|
def update_recents_playlist(engine, data):
|
|
|
|
recents_id = data['recents']['id']
|
|
boundary_date = datetime.datetime.now() - datetime.timedelta(days=data['recents']['boundary'])
|
|
|
|
recent_parts = []
|
|
|
|
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']]
|
|
|
|
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']]
|
|
|
|
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)
|
|
|
|
processors = [DeduplicateByName(), SortReleaseDate(reverse=True)]
|
|
|
|
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'])
|
|
|
|
|
|
def go():
|
|
|
|
try:
|
|
if os.path.exists(os.path.join(const.config_path, 'config.json')):
|
|
|
|
data = json.load_json(os.path.join(const.config_path, 'config.json'))
|
|
|
|
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:
|
|
logger.error(f'arg not found {not_found}')
|
|
|
|
if len(to_execute) <= 0 and len(specials_to_execute) <= 0:
|
|
logger.critical('none to execute, terminating')
|
|
return
|
|
|
|
net = Network(NetworkUser(os.environ['SPOT_CLIENT'],
|
|
os.environ['SPOT_SECRET'],
|
|
os.environ['SPOT_REFRESH']))
|
|
net.user.refresh_access_token()
|
|
|
|
engine = PlaylistEngine(net)
|
|
|
|
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)
|
|
|
|
else:
|
|
logger.critical("config json not found")
|
|
if 'SLACKHOOK' in os.environ:
|
|
requests.post(os.environ['SLACKHOOK'], json={"text": "spot playlists: config json not found"})
|
|
|
|
except Exception as e:
|
|
logger.exception("exception occured")
|
|
if 'SLACKHOOK' in os.environ:
|
|
requests.post(os.environ['SLACKHOOK'], json={"text": f"spot playlists: exception occured {e}"})
|
|
|
|
|
|
if __name__ == '__main__':
|
|
go()
|