my_environ.py

"""Class Env for reading variables from enviroment file."""

import os


class Env(object):
    """Contains enviroment variables."""

    def __init__(self, **kwargs):
        """Save **kwargs to self.VARS as enviroment variables."""

        self.VARS = {key: value[0] for key, value in kwargs.items()}
        self.TYPES = {key: value[1] for key, value in kwargs.items()}

    def read_env(self, file_name='.env'):
        if os.path.exists(file_name):
            with open(file_name, 'r') as env_file:
                for line in env_file.readlines():
                    line = line.strip()
                    if '=' in line:
                        key, value = line.split('=', maxsplit=1)
                        if key in self.TYPES:
                            self.VARS[key] = self.TYPES[key](value)
                        else:
                            self.VARS[key] = value

    def __getitem__(self, key):
        if key in self.VARS:
            return self.VARS[key]