Added query to user for moving the hosts file into place.

diff --git a/updateHostsFile.py b/updateHostsFile.py
index 805f9ad74c816eb8b91cf1f6dd6b038713f479c3..231ea090d14df7c7c4a734bd1d014227e75e61b5 100644 (file)
--- a/updateHostsFile.py
+++ b/updateHostsFile.py
@@ -42,7 +42,7 @@ def main():
        finalizeFile(finalFile)
        updateReadme(numberOfRules)
        printSuccess('Success! Your shiny new hosts file has been prepared.\nIt contains ' + str(numberOfRules) + ' unique entries.')
-       moveHostsFileIntoPlace(finalFile)
+       promptForMove(finalFile)
 
 # Prompt the User
 def promptForUpdate():
@@ -67,6 +67,13 @@ def promptForMoreCustomExclusions():
                return True
        else:
                return False
+
+def promptForMove(finalFile):
+  response = query_yes_no("Do you want to replace your existing hosts file with the newly generated file?")
+  if (response == "yes"):
+    moveHostsFileIntoPlace(finalFile)
+  else:
+    return False
 # End Prompt the User
 
 # Exclusion logic