157 lines
4.4 KiB
Python
157 lines
4.4 KiB
Python
from flask import Blueprint, jsonify, request
|
|
|
|
import logging
|
|
import os
|
|
import json
|
|
|
|
from music.api.decorators import login_or_jwt, cloud_task, no_locked_users
|
|
from music.cloud.function import update_tag as serverless_update_tag
|
|
from music.tasks.update_tag import update_tag
|
|
|
|
from music.model.tag import Tag
|
|
|
|
blueprint = Blueprint('task', __name__)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@blueprint.route('/tag', methods=['GET'])
|
|
@login_or_jwt
|
|
@no_locked_users
|
|
def tags(auth=None, user=None):
|
|
logger.info(f'retrieving tags for {user.username}')
|
|
return jsonify({
|
|
'tags': [i.to_dict() for i in Tag.collection.parent(user.key).fetch()]
|
|
}), 200
|
|
|
|
|
|
@blueprint.route('/tag/<tag_id>', methods=['GET', 'PUT', 'POST', "DELETE"])
|
|
@login_or_jwt
|
|
@no_locked_users
|
|
def tag_route(tag_id, auth=None, user=None):
|
|
if request.method == 'GET':
|
|
return get_tag(tag_id, user)
|
|
elif request.method == 'PUT':
|
|
return put_tag(tag_id, user)
|
|
elif request.method == 'POST':
|
|
return post_tag(tag_id, user)
|
|
elif request.method == 'DELETE':
|
|
return delete_tag(tag_id, user)
|
|
|
|
|
|
def get_tag(tag_id, user):
|
|
logger.info(f'retrieving {tag_id} for {user.username}')
|
|
|
|
db_tag = Tag.collection.parent(user.key).filter('tag_id', '==', tag_id).get()
|
|
if db_tag is not None:
|
|
return jsonify({
|
|
'tag': db_tag.to_dict()
|
|
}), 200
|
|
else:
|
|
return jsonify({"error": 'tag not found'}), 404
|
|
|
|
|
|
def put_tag(tag_id, user):
|
|
logger.info(f'updating {tag_id} for {user.username}')
|
|
|
|
db_tag = Tag.collection.parent(user.key).filter('tag_id', '==', tag_id).get()
|
|
|
|
if db_tag is None:
|
|
return jsonify({"error": 'tag not found'}), 404
|
|
|
|
request_json = request.get_json()
|
|
|
|
if name := request_json.get('name'):
|
|
db_tag.name = name.strip()
|
|
|
|
if time_objects := request_json.get('time_objects') is not None:
|
|
db_tag.time_objects = time_objects
|
|
|
|
if tracks := request_json.get('tracks') is not None:
|
|
db_tag.tracks = [
|
|
{
|
|
'name': track['name'].strip(),
|
|
'artist': track['artist'].strip()
|
|
}
|
|
for track in tracks
|
|
if track.get('name') and track.get('artist')
|
|
]
|
|
|
|
if albums := request_json.get('albums') is not None:
|
|
db_tag.albums = [
|
|
{
|
|
'name': album['name'].strip(),
|
|
'artist': album['artist'].strip()
|
|
}
|
|
for album in albums
|
|
if album.get('name') and album.get('artist')
|
|
]
|
|
|
|
if artists := request_json.get('artists') is not None:
|
|
db_tag.artists = [
|
|
{
|
|
'name': artist['name'].strip()
|
|
}
|
|
for artist in artists
|
|
if artist.get('name')
|
|
]
|
|
|
|
db_tag.update()
|
|
|
|
return jsonify({"message": 'tag updated', "status": "success"}), 200
|
|
|
|
|
|
def post_tag(tag_id, user):
|
|
logger.info(f'creating {tag_id} for {user.username}')
|
|
|
|
tag_id = tag_id.replace(' ', '_').strip()
|
|
|
|
existing_ids = [i.tag_id for i in Tag.collection.parent(user.key).fetch()]
|
|
while tag_id in existing_ids:
|
|
tag_id += '_'
|
|
|
|
tag = Tag(parent=user.key)
|
|
tag.tag_id = tag_id
|
|
tag.name = tag_id
|
|
tag.username = user.username
|
|
tag.save()
|
|
|
|
return jsonify({"message": 'tag added', "status": "success"}), 201
|
|
|
|
|
|
def delete_tag(tag_id, user):
|
|
logger.info(f'deleting {tag_id} for {user.username}')
|
|
|
|
db_tag = Tag.collection.parent(user.key).filter('tag_id', '==', tag_id).get()
|
|
Tag.collection.parent(user.key).delete(key=db_tag.key)
|
|
|
|
return jsonify({"message": 'tag deleted', "status": "success"}), 201
|
|
|
|
|
|
@blueprint.route('/tag/<tag_id>/update', methods=['GET'])
|
|
@login_or_jwt
|
|
@no_locked_users
|
|
def tag_refresh(tag_id, auth=None, user=None):
|
|
logger.info(f'updating {tag_id} tag for {user.username}')
|
|
|
|
if os.environ.get('DEPLOY_DESTINATION', None) == 'PROD':
|
|
serverless_update_tag(username=user.username, tag_id=tag_id)
|
|
else:
|
|
update_tag(user=user, tag=tag_id)
|
|
|
|
return jsonify({"message": 'tag updated', "status": "success"}), 200
|
|
|
|
|
|
@blueprint.route('/tag/update/task', methods=['POST'])
|
|
@cloud_task
|
|
def run_tag_task():
|
|
|
|
payload = request.get_data(as_text=True)
|
|
if payload:
|
|
payload = json.loads(payload)
|
|
|
|
logger.info(f'running {payload["username"]} / {payload["tag_id"]}')
|
|
|
|
serverless_update_tag(username=payload['username'], tag_id=payload['tag_id'])
|
|
|
|
return jsonify({'message': 'executed playlist', 'status': 'success'}), 200
|