PyCS/pycs/management/result.py

import click
import flask

from flask.cli import AppGroup

from pycs import app
from pycs.database.Project import Project
from pycs.interfaces.MediaStorage import MediaStorage

result_cli = AppGroup("result", short_help="Result operations")



@result_cli.command("export")
@click.argument("project_id")
@click.argument("indent", required=False)
@click.argument("output", required=False)
def export(project_id, output, indent):
    """ Export results for a specific project or for all projects """
    if project_id == "all":
        projects = Project.query.all()
        app.logger.info(f"Exporting results for all projects ({len(projects)})!")
        if output is None:
            output = "output.json"

    else:
        project = Project.query.get(project_id)
        if project is None:
            app.logger.error(f"Could not find project with ID {project_id}!")
            return
        app.logger.info(f"Exporting results for project {project}!")
        projects = [project]
        if output is None:
            output = f"output_project_{int(project_id):04d}.json"

    app.logger.info(f"Exporting to {output}")

    results = []

    for project in projects:
        storage = MediaStorage(project.id, None)
        project_files = [f.serialize() for f in storage.files().iter() if len(f.results()) != 0]
        results.append(dict(
            project_id=project.id,
            files=project_files,
        ))


    if indent is not None:
        indent = int(indent)

    with open(output, "w", encoding="utf-8") as out_f:
        flask.json.dump(results, out_f, app=app, indent=indent)