60 lines
1.7 KiB
Python
Executable file
60 lines
1.7 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
import database
|
|
import argparse
|
|
import sys
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
# Parsing arguments
|
|
parser = argparse.ArgumentParser(
|
|
description="TODO")
|
|
parser.add_argument(
|
|
'-o', '--output', type=argparse.FileType('w'), default=sys.stdout,
|
|
help="TODO")
|
|
parser.add_argument(
|
|
'-f', '--first-party', action='store_true',
|
|
help="TODO")
|
|
parser.add_argument(
|
|
'-e', '--end-chain', action='store_true',
|
|
help="TODO")
|
|
parser.add_argument(
|
|
'-r', '--rules', action='store_true',
|
|
help="TODO")
|
|
parser.add_argument(
|
|
'-b', '--base-rules', action='store_true',
|
|
help="TODO implies rules")
|
|
parser.add_argument(
|
|
'-d', '--no-dupplicates', action='store_true',
|
|
help="TODO")
|
|
parser.add_argument(
|
|
'-x', '--explain', action='store_true',
|
|
help="TODO")
|
|
parser.add_argument(
|
|
'-c', '--count', action='store_true',
|
|
help="TODO")
|
|
args = parser.parse_args()
|
|
|
|
DB = database.Database()
|
|
|
|
if args.count:
|
|
assert not args.explain
|
|
print(DB.count_records(
|
|
first_party_only=args.first_party,
|
|
end_chain_only=args.end_chain,
|
|
no_dupplicates=args.no_dupplicates,
|
|
rules_only=args.base_rules,
|
|
hostnames_only=not (args.rules or args.base_rules),
|
|
))
|
|
else:
|
|
for domain in DB.list_records(
|
|
first_party_only=args.first_party,
|
|
end_chain_only=args.end_chain,
|
|
no_dupplicates=args.no_dupplicates,
|
|
rules_only=args.base_rules,
|
|
hostnames_only=not (args.rules or args.base_rules),
|
|
explain=args.explain,
|
|
):
|
|
print(domain, file=args.output)
|