mirror of
https://github.com/kristoferssolo/Traffic-Light-Detector.git
synced 2025-10-21 20:00:36 +00:00
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
from pathlib import Path
|
|
from shutil import rmtree
|
|
|
|
from loguru import logger
|
|
|
|
|
|
BASE_PATH = Path(__file__).resolve().parent.parent
|
|
LOGS_PATH = BASE_PATH.joinpath(".logs")
|
|
ASSETS_PATH = BASE_PATH.joinpath("assets")
|
|
IMAGES_IN_PATH = ASSETS_PATH.joinpath("images_in")
|
|
IMAGES_OUT_PATH = ASSETS_PATH.joinpath("images_out")
|
|
HAAR_PATH = ASSETS_PATH.joinpath("haar").joinpath("TrafficLights.xml")
|
|
SOUND_PATH = ASSETS_PATH.joinpath("sound")
|
|
|
|
PATHS = (LOGS_PATH, IMAGES_IN_PATH, IMAGES_OUT_PATH, SOUND_PATH)
|
|
|
|
|
|
log_level = "DEBUG" if BASE_PATH.joinpath("debug").exists() else "INFO"
|
|
# Set up logging
|
|
logger.add(LOGS_PATH.joinpath("detection.log"), format="{time} | {level} | {message}", level=log_level, rotation="10 MB", compression="zip")
|
|
|
|
|
|
@logger.catch
|
|
def create_dirs(fresh: bool = False) -> None:
|
|
if fresh:
|
|
rmtree(ASSETS_PATH)
|
|
rmtree(LOGS_PATH)
|
|
create_dirs()
|
|
logger.info("Deleted all dirs")
|
|
else:
|
|
for path in PATHS:
|
|
if not path.exists():
|
|
path.mkdir(parents=True, exist_ok=True)
|
|
logger.info(f"Created directory {path}")
|