import json import os import tempfile import uuid import boto3 from flask import abort, Flask, request, send_file from globus_sdk import ConfidentialAppAuthClient from container_handler import convert_definition_file, pull_container from pg_utils import build_schema, create_table_entry, prep_database, select_by_column, table_exists, update_table_entry from sqs_queue_utils import put_message from task_manager import TaskManager application = Flask(__name__) manager = TaskManager(max_threads=11, kill_time=10) manager.start_prune_thread(10) @application.route("/change_thread", methods=["POST"]) def change_thread(): global manager manager = TaskManager(max_threads=request.json["threads"]) return "k" @application.before_first_request def config(): if not (table_exists("definition") and table_exists('build')): prep_database() @application.route('/thread')