Mixonomer/music/api/admin.py

from flask import Blueprint, jsonify

import logging
from datetime import datetime

from google.cloud import tasks_v2

from music.api.decorators import login_or_basic_auth, admin_required

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_basic_auth
@admin_required
def get_tasks(user=None):

    tasks = list(tasker.list_tasks(task_path))

    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