refactor plex_client,config

2025-07-12 19:00:35 +09:00
parent 35bd0c9956
commit 992161f9a9
3 changed files with 211 additions and 104 deletions
@@ -1,36 +1,88 @@
 import json
 import os
-from urllib.parse import urlparse

-CONFIG_PATH = os.path.join(os.path.dirname(__file__), "..", "config.json")
-
-def load_config():
-    if not os.path.exists(CONFIG_PATH):
-        return {}
-    with open(CONFIG_PATH, "r", encoding="utf-8") as f:
-        return json.load(f)
-
-def save_config(new_config):
-    with open(CONFIG_PATH, "w", encoding="utf-8") as f:
-        json.dump(new_config, f, indent=4, ensure_ascii=False)
+CONFIG_PATH = os.path.abspath(
+    os.path.join(os.path.dirname(__file__), "..", "config.json")
+)


-def get_server_settings(config):
-    """Return (scheme, host, port) using defaults when not configured."""
-    scheme = config.get("server_scheme", "https")
-    token = config.get("token", "") or ""
-    host = ""
-    port = config.get("server_port", "32400") or "32400"
+class ServerConfig:

-    url = config.get("server_url", "")
-    if url:
-        parsed = urlparse(url)
-        if parsed.scheme:
-            scheme = parsed.scheme
-            host = parsed.hostname or parsed.netloc
-            if parsed.port:
-                port = str(parsed.port)
-        else:
-            host = url
-    print(f"server_scheme: {scheme}, host: {host}, port: {port}")
-    return token, scheme, host, port
+    def __init__(self):
+        self.theme = "auto"
+        self.token = ""
+        self.url = ""
+        self.scheme = "https"
+        self.port = "32400"
+        self.load()
+
+    def load(self) -> None:
+        try:
+            with open(CONFIG_PATH, "r", encoding="utf-8") as f:
+                config = json.load(f)
+
+        except FileNotFoundError:
+            # 如果配置文件不存在，使用默认值
+            self.save()
+            return
+        except json.JSONDecodeError:
+            # 如果配置文件格式错误，使用默认值
+            self.save()
+            return
+        self.theme = config.get("theme", "auto")
+        self.token = config.get("token", "")
+        self.url = config.get("server_url", "")
+        self.scheme = config.get("server_scheme", "https")
+        self.port = config.get("server_port", "32400")
+
+    def save(self):
+        config = {
+            "theme": self.theme,
+            "token": self.token,
+            "server_url": self.url,
+            "server_scheme": self.scheme,
+            "server_port": self.port,
+        }
+        with open(CONFIG_PATH, "w", encoding="utf-8") as f:
+            json.dump(config, f, indent=4, ensure_ascii=False)
+
+    def set_url(self, url: str) -> None:
+        self.url = url
+
+    def set_scheme(self, scheme: str) -> None:
+        self.scheme = scheme
+
+    def set_port(self, port: str) -> None:
+        self.port = port
+
+    def set_token(self, token: str) -> None:
+        self.token = token
+
+    def set_theme(self, theme: str) -> None:
+        # check theme is valid
+        if theme not in ["auto", "dark", "light"]:
+            raise ValueError("Invalid theme. Must be 'auto', 'dark', or 'light'.")
+        self.theme = theme
+
+    def set_config(
+        self,
+        theme: str = None,
+        token: str = None,
+        url: str = None,
+        scheme: str = None,
+        port: str = None,
+    ) -> None:
+        if theme is not None:
+            self.set_theme(theme)
+        if token is not None:
+            self.set_token(token)
+        if url is not None:
+            self.set_url(url)
+        if scheme is not None:
+            self.set_scheme(scheme)
+        if port is not None:
+            self.set_port(port)
+        self.save()
+
+
+server_config = ServerConfig()
@@ -3,6 +3,7 @@ from plexapi.server import PlexServer
 from urllib.parse import urlparse
 from app.utils.common import str_is_empty

+
 def build_plex_url(scheme, url, port="32400"):
    """Build a full Plex URL from scheme, url, and port."""
    # 如果url不以http://或https://开头，则添加scheme
@@ -20,31 +21,77 @@ def build_plex_url(scheme, url, port="32400"):
        base_url = f"http://{url}:{port}"
    return base_url

-def connect_plex(username, password, token, scheme, url, port="32400"):
-    """Return a connected PlexServer instance and update config with token and server info."""
-    # 如果token存在且不为空，则使用token连接
-    if not str_is_empty(token):
-        return connect_plex_with_token(token, scheme, url, port)
-    else:
-        return connect_plex_with_pw(username, password, scheme, url, port)

+class PlexClient:
+    """A client for interacting with a Plex server."""

-def connect_plex_with_pw(username, password, scheme, url, port="32400"):
-    """Return a connected PlexServer instance and update config with token and server info."""
-    # url 初始化
-    base_url = build_plex_url(scheme, url, port)
-    # account 初始化
-    account = MyPlexAccount(username, password)
-    # token 获取
-    token = account.authenticationToken
+    def __init__(self) -> None:
+        self.server: PlexServer | None = None
+        self.token: str | None = None
+        self.base_url: str | None = None
+        self.connected = False

-    server = PlexServer(base_url, token)
-    return server, token
+    def connect(
+        self,
+        username: str = "",
+        password: str = "",
+        token: str = "",
+        scheme: str = "https",
+        url: str = "",
+        port: str = "32400",
+    ):
+        """Connect to the Plex server using username/password or token."""
+        # Connect Server with token if token is provided, otherwise use username/password
+        try:
+            if not str_is_empty(token):
+                self.server, self.token = self._connect_with_token(token, scheme, url, port)
+            else:
+                self.server, self.token = self._connect_with_pw(
+                    username, password, scheme, url, port
+                )
+            # Update the base URL and connection status
+            self.base_url = build_plex_url(scheme, url, port)
+            self.connected = True
+            return self.server, self.token
+        except Exception as e:
+            self.connected = False
+            raise

-def connect_plex_with_token(token, scheme, url, port="32400"):
-    """Return a connected PlexServer instance using a token."""
-    # URL 初始化
-    base_url = build_plex_url(scheme, url, port)
-    
-    server = PlexServer(base_url, token)
-    return server, token
+    def _connect_with_pw(self, username: str, password: str, scheme: str, url: str, port: str = "32400"):
+        """Return a connected PlexServer instance and update config with token and server info."""
+        # url 初始化
+        self.base_url = build_plex_url(scheme, url, port)
+        # account 初始化
+        account = MyPlexAccount(username, password)
+        # token 获取
+        self.token = account.authenticationToken
+
+        self.server = PlexServer(self.base_url, self.token)
+        return self.server, self.token
+
+    def _connect_with_token(self, token: str, scheme: str, url: str, port: str = "32400"):
+        """Return a connected PlexServer instance using a token."""
+        # URL 初始化
+        self.base_url = build_plex_url(scheme, url, port)
+
+        self.server = PlexServer(self.base_url, token)
+        return self.server, self.token
+
+    def get_server(self) -> PlexServer | None:
+        """Return the connected Plex server instance."""
+        if not self.connected:
+            raise RuntimeError("Plex client is not connected.")
+        return self.server
+
+    def get_all_playlist(self) -> list | None:
+        """Return all playlists from the Plex server."""
+        if not self.connected or not self.server:
+            raise ValueError("Plex server is not connected.")
+
+        try:
+            playlists = self.server.playlists()
+            return playlists
+        except Exception as e:
+            raise RuntimeError(f"Failed to fetch playlists: {str(e)}")
+
+plex_client = PlexClient()