Mixonomer/music/api/admin.py

43 lines
1.2 KiB
Python
Raw Normal View History

2019-10-20 21:07:13 +01:00
from flask import Blueprint, jsonify
import logging
from datetime import datetime
from google.cloud import tasks_v2
from music.api.decorators import login_or_jwt, admin_required, no_locked_users
2019-10-20 21:07:13 +01:00
blueprint = Blueprint('admin-api', __name__)
tasker = tasks_v2.CloudTasksClient()
task_path = tasker.queue_path('sarsooxyz', 'europe-west2', 'spotify-executions')
logger = logging.getLogger(__name__)
@blueprint.route('/tasks', methods=['GET'])
@login_or_jwt
2019-10-20 21:07:13 +01:00
@admin_required
@no_locked_users
def get_tasks(auth=None, user=None):
2019-10-20 21:07:13 +01:00
2022-08-14 19:30:45 +01:00
tasks = list(tasker.list_tasks(parent=task_path))
2019-10-20 21:07:13 +01:00
urls = {}
for task in tasks:
if urls.get(task.app_engine_http_request.relative_uri):
urls[task.app_engine_http_request.relative_uri] += 1
else:
urls[task.app_engine_http_request.relative_uri] = 1
response = {
'tasks': [{
'url': i,
'count': j,
'scheduled_times': [datetime.fromtimestamp(k.schedule_time.seconds) for k in tasks
if k.app_engine_http_request.relative_uri == i]
} for i, j in urls.items()],
'total_tasks': len(tasks),
}
return jsonify(response), 200