refactor plex_client,config

app/utils/plex_client.py

@@ -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()