"""Unpack the provided synthetic training data before training."""
from pathlib import Path
import zipfile
ROOT=Path(__file__).resolve().parent
ARCHIVE=ROOT/'training_data.zip'
EXPECTED=(ROOT/'data/robotics/train.npz',ROOT/'data/gmaw/train.npz',ROOT/'data/resistance_spot_welding/train.npz',ROOT/'data/tig_welding/train.npz',ROOT/'data/fdm/train.csv')
def prepare():
    if all(x.is_file() for x in EXPECTED):
        return False
    if not ARCHIVE.is_file():
        raise FileNotFoundError('training_data.zip is missing.')
    with zipfile.ZipFile(ARCHIVE) as archive:
        bad=archive.testzip()
        if bad:raise RuntimeError(f'Training archive damaged: {bad}')
        for item in archive.infolist():
            dest=(ROOT/item.filename).resolve()
            if not dest.is_relative_to(ROOT.resolve()):raise RuntimeError('Unsafe path in training archive')
        archive.extractall(ROOT)
    if not all(x.is_file() for x in EXPECTED):
        raise RuntimeError('The training archive is incomplete.')
    print('Synthetic data extracted into data/.')
    return True
if __name__=='__main__':prepare()