toolbox/os/environment.py

#!/usr/bin/python3
# -*- coding: utf-8 -*-
import os

from dotenv import load_dotenv
from dotenv.main import DotEnv


class EnvironmentManager(object):
    def __init__(self, filename, override=False):
        self.filename = filename

        load_dotenv(
            dotenv_path=filename,
            override=override
        )

    def open_dotenv(self, filename: str = None):
        filename = filename or self.filename
        dotenv = DotEnv(
            dotenv_path=filename,
            stream=None,
            verbose=False,
            interpolate=False,
            override=False,
            encoding="utf-8",
        )
        result = dotenv.dict()
        return result

    def get(self, key, default=None, dtype=str):
        result = os.environ.get(key)
        if result is None:
            if default is None:
                result = None
            else:
                result = default
        else:
            result = dtype(result)
        return result


def main():
    return


if __name__ == '__main__':
    main()