Source code for database

# SPDX-License-Identifier: MIT
# SPDX-FileCopyrightText: 2023 Felix Zailskas <felixzailskas@gmail.com>

from config import DATABASE_TYPE
from logger import get_logger

from .leads import LocalRepository, Repository, S3Repository

_database = None
log = get_logger()




def get_database() -> Repository:
    global _database
    if _database is None:
        if DATABASE_TYPE == "S3":
            _database = S3Repository()
        elif DATABASE_TYPE == "Local":
            _database = LocalRepository()
        else:
            log.error("Database type not initialised")
            raise ValueError

    return _database