Source code for tasks.task_lister

import os
from os.path import join
import json

CURRENT_FOLDER = os.path.abspath(os.path.dirname(__file__))


def process_file(task_file_contents, folder_path):
    task_data = json.loads(task_file_contents)
    result = []
    for task in task_data["tasks"]:
        assert all(x in task for x in ("name", "tester", "solution"))
        t_result = dict(task)  # copy
        # join paths that are needed
        t_result["tester"] = join(folder_path, t_result["tester"])
        t_result["solution"] = join(folder_path, t_result["solution"])
        if "assets" in t_result:
            t_result["assets"] = [join(folder_path, x) for x in t_result["assets"]]
        result.append(t_result)
    return result


def find_files_with_name(searched_name, path):
    for filename in os.listdir(path):
        full_path = join(path, filename)
        if os.path.isdir(full_path):
            yield from find_files_with_name(searched_name, full_path)
        elif filename == searched_name:
            yield full_path, path


def extend_nameset(collection, nameset):
    for c in collection:
        assert c["name"] not in nameset, c["name"]
        nameset.add(c["name"])


def find_all_tasks(path=CURRENT_FOLDER):
    tasks = []
    seen_names = set()

    for file_path, folder in find_files_with_name("tasks.json", path):
        result = process_file(open(file_path).read(), folder)
        extend_nameset(result, seen_names)
        tasks.extend(result)
    return tasks


def transform_assets(path):
    from grader.utils import read_code
    return dict(
        filename=os.path.basename(path),
        contents=read_code(path)
    )


def format_submit_data(task_json):
    from grader.utils import read_code

    submit_data = dict(
        #post=dict(
            solution_code=read_code(task_json["solution"]),
            tester_code=read_code(task_json["tester"]),
        #),
        name=task_json["name"],
        assets=list(map(transform_assets, task_json.get("assets", [])))
    )
    return submit_data


def submit_task(task_json, endpoint):
    import requests
    submit_data = format_submit_data(task_json)
    headers = {'Content-Type': 'application/json'}
    answer = requests.post(endpoint, data=json.dumps(submit_data), headers=headers)
    return answer.json()


def save_task(task_json, endpoint):
    import requests
    from grader.utils import read_code
    submit_data = dict(
        post=dict(
            solution_code=read_code(task_json["solution"]),
            tester_code=read_code(task_json["tester"]),
        ),
        name=task_json["name"],
        assets=list(map(transform_assets, task_json.get("assets", [])))
    )
    headers = {'Content-Type': 'application/json'}
    answer = requests.post(endpoint, data=json.dumps(submit_data), headers=headers)
    try:
        return answer.json()
    except:
        return answer.text

if __name__ == "__main__":
    tasks = find_all_tasks()
    result = submit_task(tasks[4], "http://localhost:5000/api/grade")
    print(json.dumps(result, indent=4))