/ src / setup / paths.py
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