def prefixlist(ctx, as_set, output, output_format, proto, **kwargs):
""" get prefix list for specified as-sets """
if kwargs.get('debug', False):
logging.basicConfig(level=logging.DEBUG)
prefixes = PrefixList()
bgpfuc = bgpfu.client.IRRClient()
with bgpfuc as c:
if kwargs.get('sources', False):
c.set_sources(kwargs['sources'])
# prefixes = c.get_prefixes(as_set, proto)
for chunk in c.iter_prefixes(as_set, proto):
prefixes.add_iter(chunk)
if kwargs.get('aggregate', True):
prefixes = prefixes.aggregate()
with click.open_file(output, 'w') as fobj:
outfmt.write(output_format, fobj, prefixes.str_list())
print("LEN {}".format(len(prefixes)))
def test_prefixlist_str_list():
pfx0 = PrefixList(prefixes0)
assert prefixes0 == pfx0.str_list()