gistfile1.txt

import logging
import os
import pathlib
import sys
from datetime import datetime
import time

from dotenv import load_dotenv

# env = load_dotenv()
env = load_dotenv('env.example')

PACKAGE_ROOT = pathlib.Path(__file__).resolve().parent.parent

LOG_DIR = PACKAGE_ROOT / 'logs'
LOG_DIR.mkdir(exist_ok=True)
LOG_FILE = LOG_DIR / 'app.log'

ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg'}

FORMATTER = logging.Formatter("%(message)s")


def get_console_handler():
    console_handler = logging.StreamHandler(sys.stdout)
    console_handler.setFormatter(FORMATTER)
    return console_handler


def get_logger(*, logger_name):
    """Get logger with prepared handlers."""
    logger = logging.getLogger(logger_name)
    logger.setLevel(logging.DEBUG)

    file_handler = logging.FileHandler(LOG_FILE)
    file_handler.setLevel(logging.DEBUG)
    file_handler.setFormatter(FORMATTER)

    logger.addHandler(file_handler)
    logger.addHandler(get_console_handler())
    logger.propagate = False

    return logger


logger = get_logger(logger_name=__name__)


class ApplicationConfig:
    NUM_OF_STREAM = int(os.environ.get('NUM_OF_STREAM'))
    STREAMS_RAW = os.environ.get('STREAMS').strip(';')
    ROIS_RAW = os.environ.get('ROIS').strip(';')
    CAMERAS_MAC = os.environ.get('CAMERAS_MAC').strip(';')

    stream_raw_data = STREAMS_RAW.split(';')
    rois_raw_data = ROIS_RAW.split(';')
    cameras_mac = CAMERAS_MAC.split(';')
    STREAMS = []
    ROIS = []
    MACS = []

    if NUM_OF_STREAM != len(stream_raw_data) or NUM_OF_STREAM != len(rois_raw_data) or NUM_OF_STREAM != len(cameras_mac):
        print("[ERROR] NUMBER OF STREAM DIFFERENT LENGHT OF LIST STREAM (ROI) IN VAULT")
        quit(0)

    for i in range(NUM_OF_STREAM):
        STREAMS.append(stream_raw_data[i])
        ROIS.append(rois_raw_data[i])
        MACS.append(cameras_mac[i])
    
    print('STREAMS = ', STREAMS)
    print('ROIS = ',ROIS)
    print('MACS = ', MACS)
    
    FACE_DETECTION_MODEL_PATH = os.environ.get('FACE_DETECTION_MODEL_PATH')
    FACE_DETECTION_CONF_THRES = float(os.environ.get('FACE_DETECTION_CONF_THRES'))
    FACE_DETECTION_INPUT_SIZE = int(os.environ.get('FACE_DETECTION_INPUT_SIZE'))
    FACIAL_RECOGNITION_URL = os.environ.get('FACIAL_RECOGNITION_URL')

    PERSON_DETECTION_MODEL_PATH = os.environ.get('PERSON_DETECTION_MODEL_PATH')
    PERSON_DETECTION_CONF_THRES = float(os.environ.get('PERSON_DETECTION_CONF_THRES'))
    PERSON_DETECTION_IOU_THRES = float(os.environ.get('PERSON_DETECTION_IOU_THRES'))
    PERSON_DETECTION_INPUT_SIZE = int(os.environ.get('PERSON_DETECTION_INPUT_SIZE'))

    UNIFORM_DETECTION_MODEL_PATH = os.environ.get('UNIFORM_DETECTION_MODEL_PATH')
    UNIFORM_DETECTION_CONF_THRES = float(os.environ.get('UNIFORM_DETECTION_CONF_THRES'))
    UNIFORM_DETECTION_IOU_THRES = float(os.environ.get('UNIFORM_DETECTION_IOU_THRES'))
    UNIFORM_DETECTION_INPUT_SIZE = int(os.environ.get('UNIFORM_DETECTION_INPUT_SIZE'))

    # device < 0: cpu, > 0: gpu
    DEVICE_GPU = int(os.environ.get('DEVICE_GPU'))
    if DEVICE_GPU < 0:
        USE_GPU = False
    else:
        USE_GPU = True
    
    PUSH_LOG_IOT_URL = os.environ.get('PUSH_LOG_IOT_URL')

config = ApplicationConfig()