paths.py

import os


def get_tmp_thread_files(uid):
    return [get_thread_filepath(uid, UPLOAD_DIR), get_thread_filepath(uid, RESULTS_DIR)]


def get_thread_filepath(uid, filepath):
    base_prefix = TMP_FILE_DIR + "/"
    new_prefix = f"{TMP_FILE_DIR}/{uid}"
    return filepath.replace(base_prefix, new_prefix)


DOCKER_DIR = os.path.join(os.getcwd(), "docker")
BASE_DIR = os.path.join(os.getcwd(), "backend")
TMP_FILE_DIR = os.path.join(BASE_DIR, "tmp")

RESOURCES_DIR = os.path.join(BASE_DIR, "resources")
UPLOAD_DIR = os.path.join(TMP_FILE_DIR, "uploads")
RESULTS_DIR = os.path.join(TMP_FILE_DIR, "results")

DATASET_FILE = os.path.join(RESOURCES_DIR, "dataset.json")
LAUNCH_INFO_FILE = os.path.join(UPLOAD_DIR, "launch_info.csv")
MODEL_ONNX_FILE = os.path.join(UPLOAD_DIR, "model.onnx")
METHODS_TS_FILE = os.path.join(
    BASE_DIR, "../frontend/src/components/components/Methods.ts"
)
COMPSTRAT_RESULTS_DIR = os.path.join(RESULTS_DIR, "compstrat_results")
ARTIFACTS_AI_CSV_FILE = os.path.join(RESULTS_DIR, "artifacts_run_ai/AI.csv")
ARTIFACTS_BASELINE_CSV_FILE = os.path.join(
    RESULTS_DIR, "artifacts_run_baseline/ExecutionTreeContributedCoverage.csv"
)