Update app.py

app.py

@@ -31,28 +31,69 @@ logging.basicConfig(
 logger = logging.getLogger("BK_GTA_ARCHITECT")
 
 # ==============================================================================
-# 0.5 [HuggingFace 专用] DNS 稳定性修复
+# 0.5 [HuggingFace 专用] DNS 稳定性修复 (增强版)
 # ==============================================================================
 def fix_dns():
     """
-    HuggingFace Spaces 容器默认 DNS 解析器不稳定。
-    注入 Google (8.8.8.8) 和 Cloudflare (1.1.1.1) 作为备用 DNS。
-    原 Replit 环境无此问题，此为 HuggingFace 迁移专用适配。
+    HuggingFace Spaces 容器 DNS 解析器无法解析 discord.com。
+    方案: 用 dnspython 通过 Google DNS (8.8.8.8) 直接查询 IP，
+    然后写入 /etc/hosts 完全绕过容器的 DNS。
     """
+    # 第一步：尝试注入 resolv.conf 备用 DNS
     try:
         with open('/etc/resolv.conf', 'r') as f:
             current = f.read()
         if '8.8.8.8' not in current:
             with open('/etc/resolv.conf', 'a') as f:
                 f.write('\nnameserver 8.8.8.8\nnameserver 1.1.1.1\n')
-            logger.info("✅ [DNS] Injected Google & Cloudflare DNS servers.")
+            logger.info("✅ [DNS] Injected backup DNS to resolv.conf.")
+    except Exception as e:
+        logger.warning(f"⚠️ [DNS] resolv.conf write failed: {e}")
+
+    # 第二步（核心）：用 dnspython 解析并写入 /etc/hosts
+    hosts_to_resolve = [
+        'discord.com',
+        'gateway.discord.gg',
+        'cdn.discordapp.com',
+        'api.discord.com',
+    ]
+
+    try:
+        import dns.resolver
+
+        resolver = dns.resolver.Resolver(configure=False)
+        resolver.nameservers = ['8.8.8.8', '1.1.1.1', '8.8.4.4']
+        resolver.timeout = 10
+        resolver.lifetime = 10
+
+        hosts_lines = []
+        for hostname in hosts_to_resolve:
+            try:
+                answers = resolver.resolve(hostname, 'A')
+                ip = answers[0].to_text()
+                hosts_lines.append(f"{ip} {hostname}")
+                logger.info(f"✅ [DNS] Resolved {hostname} → {ip}")
+            except Exception as e:
+                logger.warning(f"⚠️ [DNS] Failed to resolve {hostname}: {e}")
+
+        if hosts_lines:
+            try:
+                with open('/etc/hosts', 'a') as f:
+                    f.write('\n# BK-GTA Discord DNS Fix\n')
+                    for line in hosts_lines:
+                        f.write(line + '\n')
+                logger.info(f"✅ [DNS] Wrote {len(hosts_lines)} entries to /etc/hosts.")
+            except PermissionError:
+                logger.warning("⚠️ [DNS] /etc/hosts is read-only.")
+            except Exception as e:
+                logger.warning(f"⚠️ [DNS] hosts write error: {e}")
         else:
-            logger.info("✅ [DNS] Reliable DNS servers already present.")
-    except PermissionError:
-        os.environ['RES_OPTIONS'] = 'timeout:3 attempts:5'
-        logger.warning("⚠️ [DNS] resolv.conf read-only. Set RES_OPTIONS fallback.")
+            logger.error("❌ [DNS] Could not resolve any Discord hosts.")
+
+    except ImportError:
+        logger.warning("⚠️ [DNS] dnspython not installed, skipping hosts injection.")
     except Exception as e:
-        logger.warning(f"⚠️ [DNS] Fix attempt failed: {e}")
+        logger.error(f"❌ [DNS] Resolution failed: {e}")
 
 fix_dns()