updater: exit 1 on invalid operation

This commit is contained in:
Felix Yan 2019-01-20 19:26:54 +08:00
parent 882bf125a4
commit 8a1ea4e8fb
1 changed files with 8 additions and 0 deletions

View File

@ -1,6 +1,7 @@
#!/usr/bin/env python
from __future__ import unicode_literals
from argparse import ArgumentParser
import sys
if __name__ == "__main__":
parser = ArgumentParser(description="dnsmasq-china-list updater")
@ -34,6 +35,8 @@ if __name__ == "__main__":
with open(options.file[0]) as f:
lines = list(f)
changed = False
if options.add:
options.sort = True
@ -44,6 +47,7 @@ if __name__ == "__main__":
else:
print("New domain added: " + domain)
lines.append(new_line)
changed = True
if options.delete:
options.sort = True
@ -55,6 +59,10 @@ if __name__ == "__main__":
else:
print("Domain removed: " + domain)
lines.remove(target_line)
changed = True
if (options.add or options.delete) and not changed:
sys.exit(1)
if options.sort:
lines.sort(key=lambda x: x.lstrip("#"))