paths.py
1 import os 2 from pathlib import Path 3 4 5 PARENT_DIR = Path("__file__").parent.resolve() 6 DATA_DIR = PARENT_DIR.joinpath("data") 7 IMAGES_DIR = PARENT_DIR.joinpath("images") 8 9 OCR_OUTPUTS = PARENT_DIR.joinpath("OCR") 10 OCR_IMAGES = OCR_OUTPUTS.joinpath("images") 11 PDFS_AFTER_OCR = OCR_OUTPUTS.joinpath("pdf") 12 TXT_AFTER_OCR = OCR_OUTPUTS.joinpath("txt") 13 14 CHROMA_DIR = PARENT_DIR.joinpath("./chroma") 15 ARCHIVE_DIR = DATA_DIR.joinpath("archive.json") 16 IMAGES_IN_DOWNLOADS = IMAGES_DIR.joinpath("images_in_downloads") 17 18 19 def make_fundamental_paths(): 20 21 paths_to_create: list[Path] = [ 22 IMAGES_DIR, 23 DATA_DIR, 24 CHROMA_DIR, 25 OCR_OUTPUTS, 26 OCR_IMAGES, 27 PDFS_AFTER_OCR, 28 TXT_AFTER_OCR, 29 IMAGES_IN_DOWNLOADS, 30 ] 31 32 for path in paths_to_create: 33 if not Path(path).exists(): 34 os.mkdir(path) 35