From c0536c06daa141a19288edddfdb92366193c800c Mon Sep 17 00:00:00 2001
From: IrvingGao <1729854488@qq.com>
Date: Mon, 17 Jun 2024 20:26:56 +0800
Subject: [PATCH] [wifi] dns

---
 wifi_manager.py | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/wifi_manager.py b/wifi_manager.py
index aed264c..6b4cbcd 100644
--- a/wifi_manager.py
+++ b/wifi_manager.py
@@ -97,6 +97,24 @@ def save_local_ip():
         logging.info(f"Error getting private IP: {e}")
         return None
 
+def edit_dns():
+    # 编辑 /etc/resolv.conf 文件，添加 DNS 服务器
+    # nameserver 8.8.8.8
+    # nameserver 8.8.4.4
+    # 读取文件内容
+    dns_exist = False
+    with open('/etc/resolv.conf', 'r+') as file:
+        lines = file.readlines()
+        for line in lines:
+            if 'nameserver' in line:
+                dns_exist = True
+        if not dns_exist:
+            file.write('nameserver 8.8.8.8' + '\n')
+            file.write('nameserver 8.8.4.4' + '\n')
+            logging.info(f"DNS server added.")
+        else:
+            logging.info(f"DNS server already exist.")
+
 app = Flask(__name__)
 
 def close_app():
@@ -457,6 +475,8 @@ if __name__ == '__main__':
 
     logging.info(f"Starting Wi-Fi Manager...")
     
+    edit_dns()
+    
     init_networkmanager_file()  # 初始化 NetworkManager 配置文件
     
     # 扫描Wi-Fi