hacamer
/
dnsmasq-china-list
mirror of https://github.com/felixonmars/dnsmasq-china-list.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

updater: handle more than one domains and ignore dups

This commit is contained in:
Felix Yan 2015-04-23 21:27:08 +08:00
parent 89af325a6b
commit a463bfdf23
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
updater.py
View File

@ -6,8 +6,9 @@ if __name__ == "__main__":
parser = ArgumentParser(description="dnsmasq-china-list updater")
parser.add_argument(
'-a', '--add',
nargs='?',
help='Add a new domain (implies -s)',
metavar="DOMAIN",
nargs="+",
help='Add one or more new domain(s) (implies -s)',
)
parser.add_argument(
'-s', '--sort',
@ -23,7 +24,14 @@ if __name__ == "__main__":
if options.add:
options.sort = True
lines.append("server=/%s/114.114.114.114\n" % options.add)
for domain in options.add:
new_line = "server=/%s/114.114.114.114\n" % domain
if new_line in lines:
print("Domain already exists: " + domain)
else:
print("New domain added: " + domain)
lines.append(new_line)
if options.sort:
lines.sort(key=lambda x: x.lstrip("#"))