improve merge script

This commit is contained in:
Lord_Alfred
2021-07-29 18:04:19 +03:00
parent f4d0df3234
commit 5dce3b8635

View File

@@ -3,21 +3,10 @@ import argparse
import netaddr
def read(fh):
for line in fh:
yield line.rstrip()
fh.close()
def merge(fh):
for addr in netaddr.cidr_merge(read(fh)):
print(addr)
if __name__ == '__main__':
parser = argparse.ArgumentParser(description='Merge IP addresses into the smallest possible list of CIDRs.')
parser.add_argument('--source', nargs='?', type=argparse.FileType('r'), required=True, help='Source file path')
args = parser.parse_args()
merge(args.source)
for addr in netaddr.cidr_merge(args.source.readlines()):
print(addr)