from pathlib import Path

from backend.condition.processor import process_file


def main() -> None:
    try:
        # Works in scripts / debugger / pytest
        ROOT_DIR = Path(__file__).resolve().parents[1]
    except NameError:
        # __file__ is not defined in notebooks
        ROOT_DIR = Path.cwd()

    path: Path = ROOT_DIR / "condition" / "sample_data"

    # TODO: get these from s3, maybe as part of devcontainer init
    lbwf_path: Path = path / "lbwf" / "LBWF - Example Asset Data September 2025.xlsx"
    peabody_path: Path = (
        path
        / "peabody"
        / "2026_01_06 - Peabody - Stock Condition Data - Survey Records - D  Lower.xlsx"
    )
    filepaths = [lbwf_path, peabody_path]

    for fp in filepaths:
        with fp.open("rb") as f:
            process_file(
                file_stream=f,
                source_key=fp.as_posix(),
            )


if __name__ == "__main__":
    main()