From 16e922cdd627046642897add13fe6888cc6cec5c Mon Sep 17 00:00:00 2001 From: Ben Limmer Date: Tue, 30 Jul 2013 10:59:31 -0600 Subject: [PATCH] Added query to user for moving the hosts file into place. --- updateHostsFile.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/updateHostsFile.py b/updateHostsFile.py index 805f9ad74..231ea090d 100644 --- 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