[wifi] dns

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