dnsmasq-china-list/updater.py

33 lines
895 B
Python
Raw Normal View History

2015-04-23 21:17:12 +08:00
#!/usr/bin/env python
from __future__ import unicode_literals
from argparse import ArgumentParser
if __name__ == "__main__":
parser = ArgumentParser(description="dnsmasq-china-list updater")
parser.add_argument(
'-a', '--add',
nargs='?',
help='Add a new domain (implies -s)',
)
parser.add_argument(
'-s', '--sort',
action='store_true',
default=True,
help='Sort the list (default action)',
)
options = parser.parse_args()
with open("accelerated-domains.china.conf") as f:
lines = list(f)
if options.add:
options.sort = True
lines.append("server=/%s/114.114.114.114\n" % options.add)
if options.sort:
lines.sort(key=lambda x: x.lstrip("#"))
with open("accelerated-domains.china.conf", "w") as f:
f.write(''.join(filter(lambda line: line.strip(), lines)))