e-nms_qa_automation.1/data/environment.py

32 lines
695 B
Python

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()