From 738b275959a6d2ecbe5c62ab9b6faa248ff5e4e7 Mon Sep 17 00:00:00 2001 From: Rob Austein Date: Tue, 2 Nov 2010 13:25:54 +0000 Subject: Allow alternate input file names to be given on command line. svn path=/scripts/testbed-rootcert.py; revision=3529 --- scripts/testbed-rootcert.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'scripts/testbed-rootcert.py') diff --git a/scripts/testbed-rootcert.py b/scripts/testbed-rootcert.py index 6c75da9c..e4cf304f 100644 --- a/scripts/testbed-rootcert.py +++ b/scripts/testbed-rootcert.py @@ -24,8 +24,8 @@ PERFORMANCE OF THIS SOFTWARE. import csv, rpki.myrpki, sys -if len(sys.argv) != 2: - raise RuntimeError, "Usage: %s [holder]" % sys.argv[0] +if len(sys.argv) not in (2, 4): + sys.exit("Usage: %s holder [asns.csv prefixes.csv]" % sys.argv[0]) print '''\ [req] @@ -51,7 +51,7 @@ sbgp-ipAddrBlock = critical,@rfc3997_addrs ''' % { "holder" : sys.argv[1].lower(), "HOLDER" : sys.argv[1].upper() } -for i, asn in enumerate(asn for handle, asn in rpki.myrpki.csv_reader("asns.csv", columns = 2)): +for i, asn in enumerate(asn for handle, asn in rpki.myrpki.csv_reader(sys.argv[2] if len(sys.argv) > 2 else "asns.csv", columns = 2)): print "AS.%d = %s" % (i, asn) print '''\ @@ -60,6 +60,6 @@ print '''\ ''' -for i, prefix in enumerate(prefix for handle, prefix in rpki.myrpki.csv_reader("prefixes.csv", columns = 2)): +for i, prefix in enumerate(prefix for handle, prefix in rpki.myrpki.csv_reader(sys.argv[3] if len(sys.argv) > 2 else "prefixes.csv", columns = 2)): v = 6 if ":" in prefix else 4 print "IPv%d.%d = %s" % (v, i, prefix) -- cgit v1.2.3