45 lines
1.1 KiB
Python
45 lines
1.1 KiB
Python
"""handle config file"""
|
|
|
|
import json
|
|
import os
|
|
from typing import Literal
|
|
|
|
from src.static_types import ConfigType
|
|
|
|
|
|
def get_config() -> ConfigType:
|
|
"""get connection config"""
|
|
config_content: ConfigType | Literal[False] = (
|
|
get_config_file() or get_config_env()
|
|
)
|
|
if not config_content:
|
|
raise ValueError("No config.json or environment variable found")
|
|
|
|
return config_content
|
|
|
|
|
|
def get_config_file() -> ConfigType | Literal[False]:
|
|
"""read config file if available"""
|
|
if os.path.exists("./config.json"):
|
|
with open("./config.json", "r", encoding="utf-8") as f:
|
|
config_content: ConfigType = json.loads(f.read())
|
|
|
|
return config_content
|
|
|
|
return False
|
|
|
|
|
|
def get_config_env() -> ConfigType | Literal[False]:
|
|
"""read config from environment"""
|
|
if "TA_URL" in os.environ:
|
|
config_content: ConfigType = {
|
|
"ta_video_path": "/youtube",
|
|
"ta_url": os.environ["TA_URL"],
|
|
"ta_token": os.environ["TA_TOKEN"],
|
|
"jf_url": os.environ["JF_URL"],
|
|
"jf_token": os.environ["JF_TOKEN"],
|
|
}
|
|
return config_content
|
|
|
|
return False
|