tubearchivist/jellyfin
1
0
Fork 0
mirror of https://github.com/tubearchivist/jellyfin.git synced 2025-01-15 13:20:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

simplify env check for multiple environments

Browse Source
This commit is contained in:
Simon 2023-07-27 22:02:38 +07:00
parent 3281532856
commit 319fa1713d
No known key found for this signature in database
GPG Key ID: 2C15AA5E89985DD4
1 changed files with 12 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
src/connect.py
View File

@ -9,6 +9,7 @@ from src.config import get_config
from src.static_types import ConfigType, TAVideo from src.static_types import ConfigType, TAVideo
CONFIG: ConfigType = get_config() CONFIG: ConfigType = get_config()
EXPECTED_ENV = {"ta_url", "ta_token", "jf_url", "jf_token", "ta_video_path"}
class Jellyfin: class Jellyfin:
@ -99,31 +100,20 @@ class TubeArchivist:
def env_check() -> None: def env_check() -> None:
"""check if ta_video_path is accessible""" """check for expected environment"""
# if not os.path.exists("config.json"):
# raise FileNotFoundError("config.json file not found")
if not CONFIG["ta_url"]:
raise ValueError("TA_URL not set")
else:
print("TA_URL =", CONFIG["ta_url"])
if not CONFIG["ta_token"]: if not CONFIG:
raise ValueError("TA_TOKEN not set") raise ValueError("could not build config.")
else:
print("TA_TOKEN =", CONFIG["ta_token"])
if not CONFIG["jf_url"]:
raise ValueError("JF_URL not set")
else:
print("JF_URL =", CONFIG["jf_url"])
if not CONFIG["jf_token"]:
raise ValueError("JF_TOKEN not set")
else:
print("JF_TOKEN =", CONFIG["jf_token"])
if not os.path.exists(CONFIG["ta_video_path"]): if not os.path.exists(CONFIG["ta_video_path"]):
raise FileNotFoundError("failed to access ta_video_path", CONFIG["ta_video_path"]) raise FileNotFoundError(
"failed to access ta_video_path", CONFIG["ta_video_path"]
)
if not set(CONFIG) == EXPECTED_ENV:
raise ValueError(
f"expected environment {EXPECTED_ENV} but got {set(CONFIG)}"
)
def clean_overview(overview_raw: str) -> str: def clean_overview(overview_raw: str) -> str: