diff --git a/biobricks/downloader.py b/biobricks/downloader.py index c39cb5a..1af89c1 100644 --- a/biobricks/downloader.py +++ b/biobricks/downloader.py @@ -7,6 +7,15 @@ from pathlib import Path from tqdm import tqdm # Import tqdm for the progress bar +from dataclasses import dataclass, field + +@dataclass +class Downloader: + remote_url_prefix: str = field(default = 'https://dvc.biobricks.ai/files/md5/') + + def _md5_to_remote_url( self, md5 ): + return self.remote_url_prefix + md5[:2] + "/" + md5[2:] + def _download_outdir(url, dest_path: Path): with requests.get(url, headers={'BBToken': token()}, stream=True) as r: r.raise_for_status()