Animal_Disease_Classification/template.py at main · rohit-chandra/Animal_Disease_Classification · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
import os
from pathlib import Path
import logging

# log information level to console
# asctime = current sys time
logging.basicConfig(level = logging.INFO, format = "[%(asctime)s]: %(message)s")


# project name
project_name = "chicken_disease_classification"

# list of directories of the project
# NOTE: .gitkeep  --> where an empty folder needs to be tracked via Git.
list_of_files = [
    ".github/workflows/.gitkeep",
    f"src/{project_name}/__init__.py",
    f"src/{project_name}/components/__init__.py",
    f"src/{project_name}/utils/__init__.py",
    f"src/{project_name}/config/__init__.py",
    f"src/{project_name}/config/configuration.py",
    f"src/{project_name}/pipeline/__init__.py",
    f"src/{project_name}/entity/__init__.py",
    f"src/{project_name}/constants/__init__.py",
    "config/config.yaml",
    "dvc.yaml",
    "params.yaml",
    "requirements.txt",
    "setup.py",
    "research/trails.ipynb",
    "templates/index.html"
]


for file_path in list_of_files:
    # Path returns OS based file path
    file_path = Path(file_path)
    file_dir, file_name = os.path.split(file_path)

    if file_dir != "":
        os.makedirs(file_dir, exist_ok = True)
        logging.info(f"Creating  directory: {file_dir} for the file: {file_name}")

    # if file does not exist or file size is 0 (empty)
    if (not os.path.exists(file_path)) or (os.path.getsize(file_path) == 0):
        with open(file_path, "w") as f:
            pass
            logging.info(f"Creating empty file: {file_path}")

    else:
        logging.info(f"File already exists: {file_name}")