Marown/hosts
1
0
Fork 0
mirror of https://github.com/StevenBlack/hosts.git synced 2024-07-04 03:26:12 +02:00
Code Issues Packages Projects Releases Wiki Activity

Import argparse, and add an --auto flag to run the update and merge with no prompts.

Browse Source
This commit is contained in:
Steven Black 2016-02-10 22:35:09 -05:00
parent f75e0f06da
commit 1eb1509e41
1 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
updateHostsFile.py
View File

@ -20,6 +20,7 @@ import subprocess
import sys
import tempfile
import glob
import argparse
# zip files are not used actually, support deleted
# StringIO is not needed in Python 3
# Python 3 works differently with urlopen
@ -91,7 +92,16 @@ COMMON_EXCLUSIONS = ['hulu.com']
exclusionRegexs = []
numberOfRules = 0
auto = False
def main():
parser = argparse.ArgumentParser(description="Creates an amalgamated hosts file from hosts stored in data subfolders.")
parser.add_argument("--auto", "-a", dest="auto", default=False, action='store_true', help="Run without prompting.")
args = parser.parse_args()
global auto
auto = args.auto
promptForUpdate()
promptForExclusions()
mergeFile = createInitialFile()
@ -113,14 +123,14 @@ def promptForUpdate():
except:
printFailure("ERROR: No 'hosts' file in the folder, try creating one manually")
response = query_yes_no("Do you want to update all data sources?")
response = "yes" if auto else query_yes_no("Do you want to update all data sources?")
if response == "yes":
updateAllSources()
else:
print ("OK, we\'ll stick with what we\'ve got locally.")
def promptForExclusions():
response = query_yes_no("Do you want to exclude any domains?\n" +
response = "no" if auto else query_yes_no("Do you want to exclude any domains?\n" +
"For example, hulu.com video streaming must be able to access " +
"its tracking and ad servers in order to play video.")
if response == "yes":
@ -136,7 +146,7 @@ def promptForMoreCustomExclusions():
return False
def promptForMove(finalFile):
response = query_yes_no("Do you want to replace your existing hosts file " +
response = "no" if auto else query_yes_no("Do you want to replace your existing hosts file " +
"with the newly generated file?")
if response == "yes":
moveHostsFileIntoPlace(finalFile)