140 lines
4.1 KiB
Python
140 lines
4.1 KiB
Python
import os
|
|
from dotenv import load_dotenv
|
|
|
|
from src.functions import logger, str_to_bool
|
|
from src.plex import Plex
|
|
from src.jellyfin import Jellyfin
|
|
from src.emby import Emby
|
|
|
|
load_dotenv(override=True)
|
|
|
|
|
|
def jellyfin_emby_server_connection(server_baseurl, server_token, server_type):
|
|
servers = []
|
|
|
|
server_baseurl = server_baseurl.split(",")
|
|
server_token = server_token.split(",")
|
|
|
|
if len(server_baseurl) != len(server_token):
|
|
raise Exception(
|
|
f"{server_type.upper()}_BASEURL and {server_type.upper()}_TOKEN must have the same number of entries"
|
|
)
|
|
|
|
for i, baseurl in enumerate(server_baseurl):
|
|
baseurl = baseurl.strip()
|
|
if baseurl[-1] == "/":
|
|
baseurl = baseurl[:-1]
|
|
|
|
if server_type == "jellyfin":
|
|
server = Jellyfin(baseurl=baseurl, token=server_token[i].strip())
|
|
servers.append(
|
|
(
|
|
"jellyfin",
|
|
server,
|
|
)
|
|
)
|
|
|
|
elif server_type == "emby":
|
|
server = Emby(baseurl=baseurl, token=server_token[i].strip())
|
|
servers.append(
|
|
(
|
|
"emby",
|
|
server,
|
|
)
|
|
)
|
|
else:
|
|
raise Exception("Unknown server type")
|
|
|
|
logger(f"{server_type} Server {i} info: {server.info()}", 3)
|
|
|
|
return servers
|
|
|
|
|
|
def generate_server_connections():
|
|
servers = []
|
|
|
|
plex_baseurl = os.getenv("PLEX_BASEURL", None)
|
|
plex_token = os.getenv("PLEX_TOKEN", None)
|
|
plex_username = os.getenv("PLEX_USERNAME", None)
|
|
plex_password = os.getenv("PLEX_PASSWORD", None)
|
|
plex_servername = os.getenv("PLEX_SERVERNAME", None)
|
|
ssl_bypass = str_to_bool(os.getenv("SSL_BYPASS", "False"))
|
|
|
|
if plex_baseurl and plex_token:
|
|
plex_baseurl = plex_baseurl.split(",")
|
|
plex_token = plex_token.split(",")
|
|
|
|
if len(plex_baseurl) != len(plex_token):
|
|
raise Exception(
|
|
"PLEX_BASEURL and PLEX_TOKEN must have the same number of entries"
|
|
)
|
|
|
|
for i, url in enumerate(plex_baseurl):
|
|
server = Plex(
|
|
baseurl=url.strip(),
|
|
token=plex_token[i].strip(),
|
|
username=None,
|
|
password=None,
|
|
servername=None,
|
|
ssl_bypass=ssl_bypass,
|
|
)
|
|
|
|
logger(f"Plex Server {i} info: {server.info()}", 3)
|
|
|
|
servers.append(
|
|
(
|
|
"plex",
|
|
server,
|
|
)
|
|
)
|
|
|
|
if plex_username and plex_password and plex_servername:
|
|
plex_username = plex_username.split(",")
|
|
plex_password = plex_password.split(",")
|
|
plex_servername = plex_servername.split(",")
|
|
|
|
if len(plex_username) != len(plex_password) or len(plex_username) != len(
|
|
plex_servername
|
|
):
|
|
raise Exception(
|
|
"PLEX_USERNAME, PLEX_PASSWORD and PLEX_SERVERNAME must have the same number of entries"
|
|
)
|
|
|
|
for i, username in enumerate(plex_username):
|
|
server = Plex(
|
|
baseurl=None,
|
|
token=None,
|
|
username=username.strip(),
|
|
password=plex_password[i].strip(),
|
|
servername=plex_servername[i].strip(),
|
|
ssl_bypass=ssl_bypass,
|
|
)
|
|
|
|
logger(f"Plex Server {i} info: {server.info()}", 3)
|
|
servers.append(
|
|
(
|
|
"plex",
|
|
server,
|
|
)
|
|
)
|
|
|
|
jellyfin_baseurl = os.getenv("JELLYFIN_BASEURL", None)
|
|
jellyfin_token = os.getenv("JELLYFIN_TOKEN", None)
|
|
|
|
if jellyfin_baseurl and jellyfin_token:
|
|
servers.extend(
|
|
jellyfin_emby_server_connection(
|
|
jellyfin_baseurl, jellyfin_token, "jellyfin"
|
|
)
|
|
)
|
|
|
|
emby_baseurl = os.getenv("EMBY_BASEURL", None)
|
|
emby_token = os.getenv("EMBY_TOKEN", None)
|
|
|
|
if emby_baseurl and emby_token:
|
|
servers.extend(
|
|
jellyfin_emby_server_connection(emby_baseurl, emby_token, "emby")
|
|
)
|
|
|
|
return servers
|