Change hosts permission to writeable

derekhe · Nov 16, 2021 · f4f8392 · f4f8392
1 parent 7308ea7
commit f4f8392
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/runner.py b/src/runner.py
@@ -1,4 +1,6 @@
 import dns
+import os
+import stat
 import subprocess
 import dns.resolver
 import traceback
@@ -35,16 +37,20 @@ def get_hosts_origin_ips():
 
 def override_hosts():
     print("Overriding hosts")
+    os.chmod(host_path, stat.S_IWRITE)
     with open(host_path, "a") as f:
         f.writelines(host_entries)
+    print("Hosts override")
 
 
 def restore_hosts():
     print("Restoring hosts")
+    os.chmod(host_path, stat.S_IWRITE)
     with open(host_path, "r+") as f:
         host = f.read()
         for line in host_entries:
             host = host.replace(line, "")
         f.seek(0)
         f.write(host)
         f.truncate()
+    print("Restored hosts")