48 lines
1.6 KiB
Python
48 lines
1.6 KiB
Python
import os
|
||
from typing import Dict, Optional
|
||
|
||
|
||
class Environment:
|
||
"""Класс для работы с окружением и URL-адресами."""
|
||
|
||
TEST: str = 'test'
|
||
DEVELOP: str = 'develop'
|
||
|
||
URLS: Dict[str, str] = {
|
||
TEST: 'http://192.168.2.76/',
|
||
DEVELOP: 'http://192.168.2.69/'
|
||
}
|
||
|
||
def __init__(self) -> None:
|
||
"""Инициализация объекта окружения."""
|
||
try:
|
||
self.env: str = os.getenv('ENV', self.TEST)
|
||
self.access_token: str = ""
|
||
self.token: str = ""
|
||
except KeyError:
|
||
self.env: str = self.TEST
|
||
|
||
def get_base_url(self) -> str:
|
||
"""Возвращает базовый URL для текущего окружения."""
|
||
if self.env in self.URLS:
|
||
if self.env == self.TEST:
|
||
return self.URLS[self.env] + "e-nms-ui/"
|
||
return self.URLS[self.env]
|
||
raise Exception(f"Unknown value of ENV variable {self.env}")
|
||
|
||
def get_request_url(self) -> str:
|
||
"""Возвращает URL для API-запросов."""
|
||
if self.env in self.URLS:
|
||
return self.URLS[self.env]
|
||
raise Exception(f"Unknown value of ENV variable {self.env}")
|
||
|
||
def set_access_token(self, token: str) -> None:
|
||
"""Устанавливает токен доступа."""
|
||
self.token = token
|
||
|
||
def get_access_token(self) -> str:
|
||
"""Возвращает текущий токен доступа."""
|
||
return self.token
|
||
|
||
|
||
host: Environment = Environment() |