import os class Environment: TEST = 'test' DEVELOP = 'develop' URLS = { TEST: 'http://192.168.2.76/', DEVELOP: 'http://192.168.50.69/' } def __init__(self): try: self.env = os.getenv('ENV') self.access_token = "" except KeyError: self.env = self.TEST def get_base_url(self): if self.env in self.URLS: return self.URLS[self.env] else: raise Exception(f"Unknown value of ENV variable {self.env}") def set_access_token(self, token): self.token = token def get_access_token(self): return self.token host = Environment()