Add proxymanager

proxy_manager.py

@@ -0,0 +1,98 @@
+from typing import Callable, Any, Optional
+from yt_dlp import YoutubeDL
+from database import get_latest_proxy, delete_proxy, format_proxy_url, mark_proxy_success
+
+class ProxyError(Exception):
+    pass
+
+def is_proxy_error(error_msg: str) -> bool:
+    proxy_error_keywords = [
+        'proxy',
+        'connection',
+        'timeout',
+        'timed out',
+        'refused',
+        'unreachable',
+        'unable to connect',
+        'network',
+        'failed to connect',
+        'connection reset',
+        'connection aborted',
+        'read timed out',
+        'http error 407',  # Proxy authentication required
+        'tunnel connection failed',
+        'connect to host',
+    ]
+
+    non_proxy_error_keywords = [
+        'requested format is not available',
+        'format not available',
+        'no video formats',
+        'video unavailable',
+        'private video',
+        'age restricted',
+    ]
+
+    error_lower = error_msg.lower()
+
+    if any(keyword in error_lower for keyword in non_proxy_error_keywords):
+        return False
+
+    return any(keyword in error_lower for keyword in proxy_error_keywords)
+
+def execute_with_proxy_retry(
+    ydl_opts: dict,
+    operation: Callable[[YoutubeDL], Any],
+    max_retries: int = 10
+) -> Any:
+    attempts = 0
+    last_error = None
+
+    while attempts < max_retries:
+        attempts += 1
+        proxy_data = None
+
+        try:
+            proxy_data = get_latest_proxy()
+
+            if proxy_data:
+                proxy_url = format_proxy_url(proxy_data)
+                ydl_opts_with_proxy = {**ydl_opts, 'proxy': proxy_url}
+                print(f"Tentativa {attempts}: Usando proxy {proxy_url} (ID: {proxy_data['id']})")
+            else:
+                if attempts == 1:
+                    print(f"Tentativa {attempts}: Nenhum proxy disponível, tentando sem proxy")
+                    ydl_opts_with_proxy = ydl_opts
+                else:
+                    raise ProxyError("Não há mais proxies disponíveis no banco de dados")
+
+            with YoutubeDL(ydl_opts_with_proxy) as ydl:
+                result = operation(ydl)
+                print(f"Operação concluída com sucesso na tentativa {attempts}")
+
+                if proxy_data:
+                    mark_proxy_success(proxy_data['id'])
+
+                return result
+
+        except Exception as e:
+            error_msg = str(e)
+            last_error = e
+
+            print(f"Erro na tentativa {attempts}: {error_msg}")
+
+            if is_proxy_error(error_msg):
+                if proxy_data:
+                    print(f"Erro identificado como erro de proxy. Removendo proxy ID {proxy_data['id']}")
+                    delete_proxy(proxy_data['id'])
+                else:
+                    print("Erro de proxy mas nenhum proxy estava sendo usado")
+
+                continue
+            else:
+                print(f"Erro não é relacionado a proxy, lançando exceção")
+                raise e
+
+    raise ProxyError(
+        f"Falha após {max_retries} tentativas. Último erro: {last_error}"
+    )