from flask import Blueprint, request, jsonify from google.cloud import firestore from werkzeug.security import generate_password_hash import os import json import logging from datetime import datetime from music.api.decorators import login_required, login_or_basic_auth, \ admin_required, gae_cron, cloud_task, validate_json, validate_args from music.cloud import queue_run_user_playlist, offload_or_run_user_playlist from music.cloud.tasks import update_all_user_playlists, update_playlists from music.tasks.create_playlist import create_playlist from music.tasks.run_user_playlist import run_user_playlist from music.model.user import User from music.model.playlist import Playlist import music.db.database as database from spotframework.net.network import SpotifyNetworkException blueprint = Blueprint('api', __name__) db = firestore.Client() logger = logging.getLogger(__name__) @blueprint.route('/playlists', methods=['GET']) @login_or_basic_auth def all_playlists_route(user=None): assert user is not None return jsonify({ 'playlists': [i.to_dict() for i in Playlist.collection.parent(user.key).fetch()] }), 200 @blueprint.route('/playlist', methods=['GET', 'DELETE']) @login_or_basic_auth @validate_args(('name', str)) def playlist_get_delete_route(user=None): playlist = Playlist.collection.parent(user.key).filter('name', '==', request.args['name']).get() if playlist is None: return jsonify({'error': f'playlist {request.args["name"]} not found'}), 404 if request.method == "GET": return jsonify(playlist.to_dict()), 200 elif request.method == 'DELETE': Playlist.collection.parent(user.key).delete(key=playlist.key) return jsonify({"message": 'playlist deleted', "status": "success"}), 200 @blueprint.route('/playlist', methods=['POST', 'PUT']) @login_or_basic_auth @validate_json(('name', str)) def playlist_post_put_route(user=None): request_json = request.get_json() playlist_name = request_json['name'] playlist_references = [] if request_json.get('playlist_references', None): if request_json['playlist_references'] != -1: for i in request_json['playlist_references']: playlist = Playlist.collection.parent(user.key).filter('name', '==', i).get() if playlist is not None: playlist_references.append(db.document(playlist.key)) else: return jsonify({"message": f'managed playlist {i} not found', "status": "error"}), 400 if len(playlist_references) == 0 and request_json.get('playlist_references', None) != -1: playlist_references = None searched_playlist = Playlist.collection.parent(user.key).filter('name', '==', playlist_name).get() # CREATE if request.method == 'PUT': if searched_playlist is not None: return jsonify({'error': 'playlist already exists'}), 400 playlist = Playlist(parent=user.key) playlist.name = request_json['name'] for key in [i for i in Playlist.mutable_keys if i not in ['playlist_references', 'type']]: setattr(playlist, key, request_json.get(key, None)) playlist.playlist_references = playlist_references playlist.last_updated = datetime.utcnow() playlist.lastfm_stat_last_refresh = datetime.utcnow() if request_json.get('type'): playlist_type = request_json['type'].strip().lower() if playlist_type in ['default', 'recents', 'fmchart']: playlist.type = playlist_type else: playlist.type = 'default' logger.warning(f'invalid type ({playlist_type}), {user.username} / {playlist_name}') if user.spotify_linked: new_playlist = create_playlist(user, playlist_name) playlist.uri = str(new_playlist.uri) playlist.save() logger.info(f'added {user.username} / {playlist_name}') return jsonify({"message": 'playlist added', "status": "success"}), 201 # UPDATE elif request.method == 'POST': if searched_playlist is None: return jsonify({'error': "playlist doesn't exist"}), 400 playlist = Playlist.collection.parent(user.key).filter('name', '==', playlist_name).get() # ATTRIBUTES for rec_key, rec_item in request_json.items(): # type and parts require extra validation if rec_key in [k for k in Playlist.mutable_keys if k not in ['type', 'parts', 'playlist_references']]: setattr(playlist, rec_key, request_json[rec_key]) # COMPONENTS if request_json.get('parts'): if request_json['parts'] == -1: playlist.parts = [] else: playlist.parts = request_json['parts'] if playlist_references is not None: if playlist_references == -1: playlist.playlist_references = [] else: playlist.playlist_references = playlist_references # ATTRIBUTE WITH CHECKS if request_json.get('type'): playlist_type = request_json['type'].strip().lower() if playlist_type in ['default', 'recents', 'fmchart']: playlist.type = playlist_type playlist.update() logger.info(f'updated {user.username} / {playlist_name}') return jsonify({"message": 'playlist updated', "status": "success"}), 200 @blueprint.route('/user', methods=['GET', 'POST']) @login_or_basic_auth def user_route(user=None): assert user is not None if request.method == 'GET': return jsonify(user.to_dict()), 200 else: # POST request_json = request.get_json() if 'username' in request_json: if request_json['username'].strip().lower() != user.username: if user.type != "admin": return jsonify({'status': 'error', 'message': 'unauthorized'}), 401 user = User.collection.filter('username', '==', request_json['username'].strip().lower()).get() if 'locked' in request_json: if user.type == "admin": logger.info(f'updating lock {user.username} / {request_json["locked"]}') user.locked = request_json['locked'] if 'spotify_linked' in request_json: logger.info(f'deauthing {user.username}') if request_json['spotify_linked'] is False: user.access_token = None user.refresh_token = None user.spotify_linked = False if 'lastfm_username' in request_json: logger.info(f'updating lastfm username {user.username} -> {request_json["lastfm_username"]}') user.lastfm_username = request_json['lastfm_username'] user.update() logger.info(f'updated {user.username}') return jsonify({'message': 'account updated', 'status': 'succeeded'}), 200 @blueprint.route('/users', methods=['GET']) @login_or_basic_auth @admin_required def all_users_route(user=None): return jsonify({ 'accounts': [i.to_dict() for i in User.collection.fetch()] }), 200 @blueprint.route('/user/password', methods=['POST']) @login_required @validate_json(('new_password', str), ('current_password', str)) def change_password(user=None): request_json = request.get_json() if len(request_json['new_password']) == 0: return jsonify({"error": 'zero length password'}), 400 if len(request_json['new_password']) > 30: return jsonify({"error": 'password too long'}), 400 if user.check_password(request_json['current_password']): user.password = generate_password_hash(request_json['new_password']) user.update() logger.info(f'password udpated {user.username}') return jsonify({"message": 'password changed', "status": "success"}), 200 else: logger.warning(f"incorrect password {user.username}") return jsonify({'error': 'wrong password provided'}), 401 @blueprint.route('/playlist/run', methods=['GET']) @login_or_basic_auth @validate_args(('name', str)) def run_playlist(user=None): if os.environ.get('DEPLOY_DESTINATION', None) == 'PROD': queue_run_user_playlist(user.username, request.args['name']) # pass to either cloud tasks or functions else: run_user_playlist(user.username, request.args['name']) # update synchronously return jsonify({'message': 'execution requested', 'status': 'success'}), 200 @blueprint.route('/playlist/run/task', methods=['POST']) @cloud_task def run_playlist_task(): # receives cloud tasks request for update payload = request.get_data(as_text=True) if payload: payload = json.loads(payload) logger.info(f'running {payload["username"]} / {payload["name"]}') offload_or_run_user_playlist(payload['username'], payload['name']) # check whether offloading to cloud function return jsonify({'message': 'executed playlist', 'status': 'success'}), 200 logger.critical('no payload provided') @blueprint.route('/playlist/run/user', methods=['GET']) @login_or_basic_auth def run_user(user=None): if user.type == 'admin': user_name = request.args.get('username', user.username) else: user_name = user.username update_playlists(user_name) return jsonify({'message': 'executed user', 'status': 'success'}), 200 @blueprint.route('/playlist/run/user/task', methods=['POST']) @cloud_task def run_user_task(): payload = request.get_data(as_text=True) if payload: update_playlists(payload) return jsonify({'message': 'executed user', 'status': 'success'}), 200 @blueprint.route('/playlist/run/users', methods=['GET']) @login_or_basic_auth @admin_required def run_users(user=None): update_all_user_playlists() return jsonify({'message': 'executed all users', 'status': 'success'}), 200 @blueprint.route('/playlist/run/users/cron', methods=['GET']) @gae_cron def run_users_cron(): update_all_user_playlists() return jsonify({'status': 'success'}), 200 @blueprint.route('/playlist/image', methods=['GET']) @login_or_basic_auth @validate_args(('name', str)) def image(user=None): _playlist = Playlist.collection.parent(user.key).filter('name', '==', request.args['name']).get() if _playlist is None: return jsonify({'error': "playlist not found"}), 404 net = database.get_authed_spotify_network(user) try: return jsonify({'images': net.get_playlist(uri_string=_playlist.uri).images, 'status': 'success'}), 200 except SpotifyNetworkException as e: logger.exception(f'error occured during {_playlist.name} / {user.username} playlist retrieval') return jsonify({'error': f"spotify error occured: {e.http_code}"}), 404