Support backend connection timeout configuration

app/utils/config.py

@@ -17,6 +17,7 @@ class ServerConfig:
         self.url = ""
         self.scheme = "https"
         self.port = "32400"
+        self.timeout = 9
         self.library_name = ""
         self.sync_mode = DEFAULT_SYNC_MODE
         self.local_path = "playlist"
@@ -43,6 +44,7 @@ class ServerConfig:
         self.url = config.get("server_url", "")
         self.scheme = config.get("server_scheme", "https")
         self.port = config.get("server_port", "32400")
+        self.timeout = config.get("timeout", 9)
         self.library_name = config.get("library_name", "")
         self.sync_mode = config.get("sync_mode", DEFAULT_SYNC_MODE)
         self.local_path = config.get("local_path", "playlist")
@@ -56,6 +58,7 @@ class ServerConfig:
             "server_url": self.url,
             "server_scheme": self.scheme,
             "server_port": self.port,
+            "timeout": self.timeout,
             "library_name": self.library_name,
             "sync_mode": self.sync_mode,
             "local_path": self.local_path,
@@ -74,6 +77,9 @@ class ServerConfig:
     def set_port(self, port: str) -> None:
         self.port = port
 
+    def set_timeout(self, timeout: int) -> None:
+        self.timeout = timeout if timeout and timeout > 0 else 9
+
     def set_token(self, token: str) -> None:
         self.token = token
 
@@ -103,6 +109,7 @@ class ServerConfig:
         url: str = None,
         scheme: str = None,
         port: str = None,
+        timeout: int | None = None,
         library_name: str | None = None,
         sync_mode: str | None = None,
         local_path: str | None = None,
@@ -118,6 +125,8 @@ class ServerConfig:
             self.set_scheme(scheme)
         if port is not None:
             self.set_port(port)
+        if timeout is not None:
+            self.set_timeout(timeout)
         if library_name is not None:
             self.set_library(library_name)
         if sync_mode is not None:

app/utils/plex_client.py

@@ -50,6 +50,7 @@ class PlexClient:
         scheme: str = "https",
         url: str = "",
         port: str = "32400",
+        timeout: int | None = None,
     ) -> tuple[PlexServer, str]:
         """Connect to the Plex server using username/password or token.
 
@@ -69,11 +70,11 @@ class PlexClient:
         try:
             if not str_is_empty(token):
                 self.server, self.token = self._connect_with_token(
-                    token, scheme, url, port
+                    token, scheme, url, port, timeout
                 )
             else:
                 self.server, self.token = self._connect_with_pw(
-                    username, password, scheme, url, port
+                    username, password, scheme, url, port, timeout
                 )
             # Update the base URL and connection status
             self.base_url = build_plex_url(scheme, url, port)
@@ -88,30 +89,41 @@ class PlexClient:
             raise
 
     def _connect_with_pw(
-        self, username: str, password: str, scheme: str, url: str, port: str = "32400"
+        self,
+        username: str,
+        password: str,
+        scheme: str,
+        url: str,
+        port: str = "32400",
+        timeout: int | None = None,
     ):
         """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)
+        account = MyPlexAccount(username, password, timeout=timeout)
         # token 获取
         self.token = account.authenticationToken
 
-        self.server = PlexServer(self.base_url, self.token)
+        self.server = PlexServer(self.base_url, self.token, timeout=timeout)
         logger.debug(
             f"Connected to Plex server with username: {username}, token: {self.token}"
         )
         return self.server, self.token
 
     def _connect_with_token(
-        self, token: str, scheme: str, url: str, port: str = "32400"
+        self,
+        token: str,
+        scheme: str,
+        url: str,
+        port: str = "32400",
+        timeout: int | None = None,
     ):
         """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)
+        self.server = PlexServer(self.base_url, token, timeout=timeout)
         logger.debug(f"Connected to Plex server with token: {token}")
         return self.server, token